1 /*
2  * Copyright (c) 2016, 2020, 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.jfr.internal.consumer;
26 
27 import java.io.BufferedInputStream;
28 import java.io.IOException;
29 import java.io.InputStream;
30 
31 import jdk.jfr.internal.management.EventByteStream;
32 
33 public final class FinishedStream extends EventByteStream {
34     private final BufferedInputStream inputStream;
35     private final byte[] buffer;
36 
FinishedStream(InputStream is, int blockSize)37     public FinishedStream(InputStream is, int blockSize) {
38         super();
39         this.inputStream = new BufferedInputStream(is, 50000);
40         this.buffer = new byte[blockSize];
41     }
42 
43     @Override
read()44     public synchronized byte[] read() throws IOException {
45         // OK to reuse buffer since this
46         // is only used for serialization
47         touch();
48         int read = inputStream.read(buffer);
49         if (read == -1) {
50             // null indicate no more data
51             return null;
52         }
53         if (read != buffer.length) {
54             byte[] smallerBuffer = new byte[read];
55             System.arraycopy(buffer, 0, smallerBuffer, 0, read);
56             return smallerBuffer;
57         }
58 
59         return buffer;
60     }
61 
62     @Override
close()63     public synchronized void close() throws IOException {
64         inputStream.close();
65     }
66 
67 }
68