java 缓冲池 BufferPool

/** * 缓冲池 */ public class BufferPool { private static final int BUFFER_SIZE = 2 * 1024 * 1024; //缓冲包2M private static final int BUFFER_MAX_NUM = 20; //缓冲池保留最大数量 private static final ConcurrentLinkedQueue<byte[]> buffers = new ConcurrentLinkedQueue<byte[]>(); private static AtomicInteger countCreated = new AtomicInteger(0); //已创建的缓冲包个数 /** 分配 */ public static byte[] allocate() { byte[] result = buffers.poll(); //创建新缓冲包 if(result == null) { result = new byte[BUFFER_SIZE]; //记录创建个数 int count = countCreated.incrementAndGet(); //日志 if(count <= BUFFER_MAX_NUM) { LogCore.CORE.info("创建新的BufferPool缓冲池，已创建总数量：count={}", count); } else { LogCore.CORE.warn("创建新的BufferPool缓冲池，已创建总数量：count={}", count, new Throwable()); } } return result; } /** * 回收 * @param buff */ public static void deallocate(byte[] buff) { //缓冲池已达上限 if(buffers.size() >= BUFFER_MAX_NUM) return; //回收的缓冲大小必须正确 if(buff.length != BUFFER_SIZE) return; //加回到池中 buffers.add(buff); } }