third-party/atomix/storage/src/main/java/io/atomix/storage/buffer/PooledAllocator.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.ReferencePool;
  19
  20 /**
  21  * Pooled buffer allocator.
  22  *
  23  * @author <a href="http://github.com/kuujo">Jordan Halterman</a>
  24  */
  25 public abstract class PooledAllocator implements BufferAllocator {
  26   private final ReferencePool<AbstractBuffer> pool;
  27
  28   protected PooledAllocator(ReferencePool<AbstractBuffer> pool) {
  29     this.pool = pool;
  30   }
  31
  32   /**
  33    * Returns the maximum buffer capacity.
  34    *
  35    * @return The maximum buffer capacity.
  36    */
  37   protected abstract int maxCapacity();
  38
  39   @Override
  40   public Buffer allocate() {
  41     return allocate(4096, maxCapacity());
  42   }
  43
  44   @Override
  45   public Buffer allocate(int capacity) {
  46     return allocate(capacity, maxCapacity());
  47   }
  48
  49   @Override
  50   public Buffer allocate(int initialCapacity, int maxCapacity) {
  51     return pool.acquire().reset(0, initialCapacity, maxCapacity).clear();
  52   }
  53
  54 }