1 /*

     2  * Copyright (c) 2015, 2018, 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.internal.net.http.common;

    26 

    27 import java.nio.ByteBuffer;

    28 import java.util.Objects;

    29 import java.util.function.Supplier;

    30 

    31 public class ByteBufferReference  implements Supplier<ByteBuffer> {

    32 

    33     private ByteBuffer buffer;

    34     private final ByteBufferPool pool;

    35 

    36     public static ByteBufferReference of(ByteBuffer buffer) {

    37         return of(buffer, null);

    38     }

    39 

    40     public static ByteBufferReference of(ByteBuffer buffer, ByteBufferPool pool) {

    41         Objects.requireNonNull(buffer);

    42         return new ByteBufferReference(buffer, pool);

    43     }

    44 

    45     public static ByteBuffer[] toBuffers(ByteBufferReference... refs) {

    46         ByteBuffer[] bufs = new ByteBuffer[refs.length];

    47         for (int i = 0; i < refs.length; i++) {

    48             bufs[i] = refs[i].get();

    49         }

    50         return bufs;

    51     }

    52 

    53     public static ByteBufferReference[] toReferences(ByteBuffer... buffers) {

    54         ByteBufferReference[] refs = new ByteBufferReference[buffers.length];

    55         for (int i = 0; i < buffers.length; i++) {

    56             refs[i] = of(buffers[i]);

    57         }

    58         return refs;

    59     }

    60 

    61 

    62     public static void clear(ByteBufferReference[] refs) {

    63         for(ByteBufferReference ref : refs) {

    64             ref.clear();

    65         }

    66     }

    67 

    68     private ByteBufferReference(ByteBuffer buffer, ByteBufferPool pool) {

    69         this.buffer = buffer;

    70         this.pool = pool;

    71     }

    72 

    73     @Override

    74     public ByteBuffer get() {

    75         ByteBuffer buf = this.buffer;

    76         assert buf!=null : "getting ByteBuffer after clearance";

    77         return buf;

    78     }

    79 

    80     public void clear() {

    81         ByteBuffer buf = this.buffer;

    82         assert buf!=null : "double ByteBuffer clearance";

    83         this.buffer = null;

    84         if (pool != null) {

    85             pool.release(buf);

    86         }

    87     }

    88 }