--- /dev/null 2018-06-17 23:18:20.806999507 +0800 +++ new/src/share/classes/jdk/jfr/internal/ChunkInputStream.java 2019-01-29 10:59:24.380003771 +0800 @@ -0,0 +1,116 @@ +/* + * Copyright (c) 2001, 2019, Oracle and/or its affiliates. All rights reserved. + * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. + * + * This code is free software; you can redistribute it and/or modify it + * under the terms of the GNU General Public License version 2 only, as + * published by the Free Software Foundation. Oracle designates this + * particular file as subject to the "Classpath" exception as provided + * by Oracle in the LICENSE file that accompanied this code. + * + * This code is distributed in the hope that it will be useful, but WITHOUT + * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or + * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License + * version 2 for more details (a copy is included in the LICENSE file that + * accompanied this code). + * + * You should have received a copy of the GNU General Public License version + * 2 along with this work; if not, write to the Free Software Foundation, + * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. + * + * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA + * or visit www.oracle.com if you need additional information or have any + * questions. + */ + +package jdk.jfr.internal; + +import java.io.BufferedInputStream; +import java.io.IOException; +import java.io.InputStream; +import java.util.ArrayList; +import java.util.Iterator; +import java.util.List; + +final class ChunkInputStream extends InputStream { + private final Iterator chunks; + private RepositoryChunk currentChunk; + private InputStream stream; + + ChunkInputStream(List chunks) throws IOException { + List l = new ArrayList<>(chunks.size()); + for (RepositoryChunk c : chunks) { + c.use(); // keep alive while we're reading. + l.add(c); + } + + this.chunks = l.iterator(); + nextStream(); + } + + @Override + public int available() throws IOException { + if (stream != null) { + return stream.available(); + } + return 0; + } + + private boolean nextStream() throws IOException { + if (!nextChunk()) { + return false; + } + + stream = new BufferedInputStream(SecuritySupport.newFileInputStream(currentChunk.getFile())); + return true; + } + + private boolean nextChunk() { + if (!chunks.hasNext()) { + return false; + } + currentChunk = chunks.next(); + return true; + } + + @Override + public int read() throws IOException { + while (true) { + if (stream != null) { + int r = stream.read(); + if (r != -1) { + return r; + } + stream.close(); + currentChunk.release(); + stream = null; + currentChunk = null; + } + if (!nextStream()) { + return -1; + } + } + } + + @Override + public void close() throws IOException { + if (stream != null) { + stream.close(); + stream = null; + } + while (currentChunk != null) { + currentChunk.release(); + currentChunk = null; + if (!nextChunk()) { + return; + } + } + } + + @Override + @SuppressWarnings("deprecation") + protected void finalize() throws Throwable { + super.finalize(); + close(); + } +}