third-party/atomix/storage/src/main/java/io/atomix/storage/buffer/ByteBufferBuffer.java

   1 /*
   2  * Copyright 2015-present Open Networking Foundation
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  * http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16 package io.atomix.storage.buffer;
  17
  18 import io.atomix.utils.concurrent.ReferenceManager;
  19
  20 /**
  21  * {@link java.nio.ByteBuffer} based buffer.
  22  */
  23 public abstract class ByteBufferBuffer extends AbstractBuffer {
  24   protected final ByteBufferBytes bytes;
  25
  26   public ByteBufferBuffer(ByteBufferBytes bytes, ReferenceManager<Buffer> referenceManager) {
  27     super(bytes, referenceManager);
  28     this.bytes = bytes;
  29   }
  30
  31   public ByteBufferBuffer(ByteBufferBytes bytes, int offset, int initialCapacity, int maxCapacity, ReferenceManager<Buffer> referenceManager) {
  32     super(bytes, offset, initialCapacity, maxCapacity, referenceManager);
  33     this.bytes = bytes;
  34   }
  35
  36   @Override
  37   public byte[] array() {
  38     return bytes.array();
  39   }
  40
  41   @Override
  42   protected void compact(int from, int to, int length) {
  43     byte[] bytes = new byte[1024];
  44     int position = from;
  45     while (position < from + length) {
  46       int size = Math.min((from + length) - position, 1024);
  47       this.bytes.read(position, bytes, 0, size);
  48       this.bytes.write(0, bytes, 0, size);
  49       position += size;
  50     }
  51   }
  52
  53 }