1 /* 2 * Copyright (c) 2015, 2016, Oracle and/or its affiliates. All rights reserved. 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 4 * 5 * This code is free software; you can redistribute it and/or modify it 6 * under the terms of the GNU General Public License version 2 only, as 7 * published by the Free Software Foundation. Oracle designates this 8 * particular file as subject to the "Classpath" exception as provided 9 * by Oracle in the LICENSE file that accompanied this code. 10 * 11 * This code is distributed in the hope that it will be useful, but WITHOUT 12 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 13 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 14 * version 2 for more details (a copy is included in the LICENSE file that 15 * accompanied this code). 16 * 17 * You should have received a copy of the GNU General Public License version 18 * 2 along with this work; if not, write to the Free Software Foundation, 19 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 20 * 21 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 22 * or visit www.oracle.com if you need additional information or have any 23 * questions. 24 */ 25 package jdk.incubator.http.internal.common; 26 27 import java.nio.ByteBuffer; 28 import java.util.List; 29 import java.util.Objects; 30 import java.util.function.Supplier; 31 import java.util.stream.Collectors; 32 33 public class ByteBufferReference implements Supplier<ByteBuffer> { 34 35 private ByteBuffer buffer; 36 private final ByteBufferPool pool; 37 38 public static ByteBufferReference of(ByteBuffer buffer) { 39 return of(buffer, null); 40 } 41 42 public static ByteBufferReference of(ByteBuffer buffer, ByteBufferPool pool) { 43 Objects.requireNonNull(buffer); 44 return new ByteBufferReference(buffer, pool); 45 } 46 47 public static ByteBuffer[] toBuffers(ByteBufferReference... refs) { 48 ByteBuffer[] bufs = new ByteBuffer[refs.length]; 49 for (int i = 0; i < refs.length; i++) { 50 bufs[i] = refs[i].get(); 51 } 52 return bufs; 53 } 54 55 public static ByteBufferReference[] toReferences(ByteBuffer... buffers) { 56 ByteBufferReference[] refs = new ByteBufferReference[buffers.length]; 57 for (int i = 0; i < buffers.length; i++) { 58 refs[i] = of(buffers[i]); 59 } 60 return refs; 61 } 62 63 64 public static void clear(ByteBufferReference[] refs) { 65 for(ByteBufferReference ref : refs) { 66 ref.clear(); 67 } 68 } 69 70 private ByteBufferReference(ByteBuffer buffer, ByteBufferPool pool) { 71 this.buffer = buffer; 72 this.pool = pool; 73 } 74 75 @Override 76 public ByteBuffer get() { 77 ByteBuffer buf = this.buffer; 78 assert buf!=null : "getting ByteBuffer after clearance"; 79 return buf; 80 } 81 82 public void clear() { 83 ByteBuffer buf = this.buffer; 84 assert buf!=null : "double ByteBuffer clearance"; 85 this.buffer = null; 86 if (pool != null) { 87 pool.release(buf); 88 } 89 } 90 }