mirror of
https://github.com/openjdk/jdk.git
synced 2025-08-28 15:24:43 +02:00
8278268: (ch) InputStream returned by Channels.newInputStream should have fast path for FileChannel targets
Reviewed-by: bpb
This commit is contained in:
parent
9b9b5a7a5c
commit
40aea04680
6 changed files with 619 additions and 264 deletions
|
@ -1,5 +1,5 @@
|
||||||
/*
|
/*
|
||||||
* Copyright (c) 2001, 2022, Oracle and/or its affiliates. All rights reserved.
|
* Copyright (c) 2001, 2023, Oracle and/or its affiliates. All rights reserved.
|
||||||
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
|
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
|
||||||
*
|
*
|
||||||
* This code is free software; you can redistribute it and/or modify it
|
* This code is free software; you can redistribute it and/or modify it
|
||||||
|
@ -251,6 +251,20 @@ class ChannelInputStream extends InputStream {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (out instanceof ChannelOutputStream cos && cos.channel() instanceof FileChannel fc) {
|
||||||
|
ReadableByteChannel rbc = ch;
|
||||||
|
|
||||||
|
if (rbc instanceof SelectableChannel sc) {
|
||||||
|
synchronized (sc.blockingLock()) {
|
||||||
|
if (!sc.isBlocking())
|
||||||
|
throw new IllegalBlockingModeException();
|
||||||
|
return transfer(rbc, fc);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return transfer(rbc, fc);
|
||||||
|
}
|
||||||
|
|
||||||
return super.transferTo(out);
|
return super.transferTo(out);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -274,6 +288,25 @@ class ChannelInputStream extends InputStream {
|
||||||
return pos - initialPos;
|
return pos - initialPos;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Transfers all bytes from a readable byte channel to a target channel's file.
|
||||||
|
* If the readable byte channel is a selectable channel then it must be in
|
||||||
|
* blocking mode.
|
||||||
|
*/
|
||||||
|
private static long transfer(ReadableByteChannel src, FileChannel dst) throws IOException {
|
||||||
|
long initialPos = dst.position();
|
||||||
|
long pos = initialPos;
|
||||||
|
try {
|
||||||
|
long n;
|
||||||
|
while ((n = dst.transferFrom(src, pos, Long.MAX_VALUE)) > 0) {
|
||||||
|
pos += n;
|
||||||
|
}
|
||||||
|
} finally {
|
||||||
|
dst.position(pos);
|
||||||
|
}
|
||||||
|
return pos - initialPos;
|
||||||
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void close() throws IOException {
|
public void close() throws IOException {
|
||||||
ch.close();
|
ch.close();
|
||||||
|
|
|
@ -1,5 +1,5 @@
|
||||||
/*
|
/*
|
||||||
* Copyright (c) 2021, Oracle and/or its affiliates. All rights reserved.
|
* Copyright (c) 2021, 2023, Oracle and/or its affiliates. All rights reserved.
|
||||||
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
|
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
|
||||||
*
|
*
|
||||||
* This code is free software; you can redistribute it and/or modify it
|
* This code is free software; you can redistribute it and/or modify it
|
||||||
|
@ -21,12 +21,10 @@
|
||||||
* questions.
|
* questions.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
import java.io.ByteArrayInputStream;
|
|
||||||
import java.io.ByteArrayOutputStream;
|
import java.io.ByteArrayOutputStream;
|
||||||
import java.io.InputStream;
|
import java.io.InputStream;
|
||||||
import java.io.IOException;
|
import java.io.IOException;
|
||||||
import java.io.OutputStream;
|
import java.io.OutputStream;
|
||||||
import java.nio.ByteBuffer;
|
|
||||||
import java.nio.channels.Channels;
|
import java.nio.channels.Channels;
|
||||||
import java.nio.channels.FileChannel;
|
import java.nio.channels.FileChannel;
|
||||||
import java.nio.channels.IllegalBlockingModeException;
|
import java.nio.channels.IllegalBlockingModeException;
|
||||||
|
@ -36,27 +34,14 @@ import java.nio.channels.SelectableChannel;
|
||||||
import java.nio.channels.WritableByteChannel;
|
import java.nio.channels.WritableByteChannel;
|
||||||
import java.nio.file.Files;
|
import java.nio.file.Files;
|
||||||
import java.nio.file.Path;
|
import java.nio.file.Path;
|
||||||
import java.nio.file.StandardOpenOption;
|
|
||||||
import java.util.Arrays;
|
|
||||||
import java.util.Random;
|
|
||||||
import java.util.concurrent.CompletableFuture;
|
import java.util.concurrent.CompletableFuture;
|
||||||
import java.util.concurrent.ExecutionException;
|
import java.util.concurrent.ExecutionException;
|
||||||
import java.util.concurrent.Future;
|
import java.util.concurrent.Future;
|
||||||
import java.util.concurrent.atomic.AtomicReference;
|
|
||||||
import java.util.function.Consumer;
|
|
||||||
import java.util.function.Supplier;
|
|
||||||
|
|
||||||
import org.testng.annotations.DataProvider;
|
import org.testng.annotations.DataProvider;
|
||||||
import org.testng.annotations.Test;
|
import org.testng.annotations.Test;
|
||||||
|
|
||||||
import jdk.test.lib.RandomFactory;
|
|
||||||
|
|
||||||
import static java.lang.String.format;
|
|
||||||
import static java.nio.file.StandardOpenOption.*;
|
|
||||||
|
|
||||||
import static org.testng.Assert.assertEquals;
|
|
||||||
import static org.testng.Assert.assertThrows;
|
import static org.testng.Assert.assertThrows;
|
||||||
import static org.testng.Assert.assertTrue;
|
|
||||||
|
|
||||||
/*
|
/*
|
||||||
* @test
|
* @test
|
||||||
|
@ -68,26 +53,14 @@ import static org.testng.Assert.assertTrue;
|
||||||
* InputStream.transferTo specification
|
* InputStream.transferTo specification
|
||||||
* @key randomness
|
* @key randomness
|
||||||
*/
|
*/
|
||||||
public class TransferTo {
|
public class TransferTo extends TransferToBase {
|
||||||
private static final int MIN_SIZE = 10_000;
|
|
||||||
private static final int MAX_SIZE_INCR = 100_000_000 - MIN_SIZE;
|
|
||||||
|
|
||||||
private static final int ITERATIONS = 10;
|
|
||||||
|
|
||||||
private static final int NUM_WRITES = 3*1024;
|
|
||||||
private static final int BYTES_PER_WRITE = 1024*1024;
|
|
||||||
private static final long BYTES_WRITTEN = (long) NUM_WRITES*BYTES_PER_WRITE;
|
|
||||||
|
|
||||||
private static final Random RND = RandomFactory.getRandom();
|
|
||||||
|
|
||||||
private static final Path CWD = Path.of(".");
|
|
||||||
|
|
||||||
/*
|
/*
|
||||||
* Provides test scenarios, i.e., combinations of input and output streams
|
* Provides test scenarios, i.e., combinations of input and output streams
|
||||||
* to be tested.
|
* to be tested.
|
||||||
*/
|
*/
|
||||||
@DataProvider
|
@DataProvider
|
||||||
public static Object[][] streamCombinations() throws Exception {
|
public static Object[][] streamCombinations() {
|
||||||
return new Object[][] {
|
return new Object[][] {
|
||||||
// tests FileChannel.transferTo(FileChannel) optimized case
|
// tests FileChannel.transferTo(FileChannel) optimized case
|
||||||
{fileChannelInput(), fileChannelOutput()},
|
{fileChannelInput(), fileChannelOutput()},
|
||||||
|
@ -96,7 +69,7 @@ public class TransferTo {
|
||||||
// optimized case
|
// optimized case
|
||||||
{fileChannelInput(), selectableChannelOutput()},
|
{fileChannelInput(), selectableChannelOutput()},
|
||||||
|
|
||||||
// tests FileChannel.transferTo(WritableChannelOutput)
|
// tests FileChannel.transferTo(WritableByteChannelOutput)
|
||||||
// optimized case
|
// optimized case
|
||||||
{fileChannelInput(), writableByteChannelOutput()},
|
{fileChannelInput(), writableByteChannelOutput()},
|
||||||
|
|
||||||
|
@ -105,21 +78,24 @@ public class TransferTo {
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Input streams to be tested.
|
||||||
|
*/
|
||||||
|
@DataProvider
|
||||||
|
public static Object[][] inputStreamProviders() {
|
||||||
|
return new Object[][] {
|
||||||
|
{fileChannelInput()},
|
||||||
|
{readableByteChannelInput()}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
/*
|
/*
|
||||||
* Testing API compliance: input stream must throw NullPointerException
|
* Testing API compliance: input stream must throw NullPointerException
|
||||||
* when parameter "out" is null.
|
* when parameter "out" is null.
|
||||||
*/
|
*/
|
||||||
@Test(dataProvider = "streamCombinations")
|
@Test(dataProvider = "inputStreamProviders")
|
||||||
public void testNullPointerException(InputStreamProvider inputStreamProvider,
|
public void testNullPointerException(InputStreamProvider inputStreamProvider) {
|
||||||
OutputStreamProvider outputStreamProvider) throws Exception {
|
assertNullPointerException(inputStreamProvider);
|
||||||
// tests empty input stream
|
|
||||||
assertThrows(NullPointerException.class, () -> inputStreamProvider.input().transferTo(null));
|
|
||||||
|
|
||||||
// tests single-byte input stream
|
|
||||||
assertThrows(NullPointerException.class, () -> inputStreamProvider.input((byte) 1).transferTo(null));
|
|
||||||
|
|
||||||
// tests dual-byte input stream
|
|
||||||
assertThrows(NullPointerException.class, () -> inputStreamProvider.input((byte) 1, (byte) 2).transferTo(null));
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/*
|
/*
|
||||||
|
@ -129,100 +105,11 @@ public class TransferTo {
|
||||||
@Test(dataProvider = "streamCombinations")
|
@Test(dataProvider = "streamCombinations")
|
||||||
public void testStreamContents(InputStreamProvider inputStreamProvider,
|
public void testStreamContents(InputStreamProvider inputStreamProvider,
|
||||||
OutputStreamProvider outputStreamProvider) throws Exception {
|
OutputStreamProvider outputStreamProvider) throws Exception {
|
||||||
// tests empty input stream
|
assertStreamContents(inputStreamProvider, outputStreamProvider);
|
||||||
checkTransferredContents(inputStreamProvider, outputStreamProvider, new byte[0]);
|
|
||||||
|
|
||||||
// tests input stream with a length between 1k and 4k
|
|
||||||
checkTransferredContents(inputStreamProvider, outputStreamProvider, createRandomBytes(1024, 4096));
|
|
||||||
|
|
||||||
// tests input stream with several data chunks, as 16k is more than a
|
|
||||||
// single chunk can hold
|
|
||||||
checkTransferredContents(inputStreamProvider, outputStreamProvider, createRandomBytes(16384, 16384));
|
|
||||||
|
|
||||||
// tests randomly chosen starting positions within source and
|
|
||||||
// target stream
|
|
||||||
for (int i = 0; i < ITERATIONS; i++) {
|
|
||||||
byte[] inBytes = createRandomBytes(MIN_SIZE, MAX_SIZE_INCR);
|
|
||||||
int posIn = RND.nextInt(inBytes.length);
|
|
||||||
int posOut = RND.nextInt(MIN_SIZE);
|
|
||||||
checkTransferredContents(inputStreamProvider, outputStreamProvider, inBytes, posIn, posOut);
|
|
||||||
}
|
|
||||||
|
|
||||||
// tests reading beyond source EOF (must not transfer any bytes)
|
|
||||||
checkTransferredContents(inputStreamProvider, outputStreamProvider, createRandomBytes(4096, 0), 4096, 0);
|
|
||||||
|
|
||||||
// tests writing beyond target EOF (must extend output stream)
|
|
||||||
checkTransferredContents(inputStreamProvider, outputStreamProvider, createRandomBytes(4096, 0), 0, 4096);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/*
|
/*
|
||||||
* Special test for file-to-file transfer of more than 2 GB. This test
|
* Special test whether selectable channel based transfer throws blocking mode exception.
|
||||||
* covers multiple iterations of FileChannel.transerTo(FileChannel),
|
|
||||||
* which ChannelInputStream.transferTo() only applies in this particular
|
|
||||||
* case, and cannot get tested using a single byte[] due to size limitation
|
|
||||||
* of arrays.
|
|
||||||
*/
|
|
||||||
@Test
|
|
||||||
public void testMoreThanTwoGB() throws IOException {
|
|
||||||
// prepare two temporary files to be compared at the end of the test
|
|
||||||
// set the source file name
|
|
||||||
String sourceName = String.format("test3GBSource%s.tmp",
|
|
||||||
String.valueOf(RND.nextInt(Integer.MAX_VALUE)));
|
|
||||||
Path sourceFile = CWD.resolve(sourceName);
|
|
||||||
|
|
||||||
try {
|
|
||||||
// set the target file name
|
|
||||||
String targetName = String.format("test3GBTarget%s.tmp",
|
|
||||||
String.valueOf(RND.nextInt(Integer.MAX_VALUE)));
|
|
||||||
Path targetFile = CWD.resolve(targetName);
|
|
||||||
|
|
||||||
try {
|
|
||||||
// calculate initial position to be just short of 2GB
|
|
||||||
final long initPos = 2047*BYTES_PER_WRITE;
|
|
||||||
|
|
||||||
// create the source file with a hint to be sparse
|
|
||||||
try (FileChannel fc = FileChannel.open(sourceFile, CREATE_NEW, SPARSE, WRITE, APPEND);) {
|
|
||||||
// set initial position to avoid writing nearly 2GB
|
|
||||||
fc.position(initPos);
|
|
||||||
|
|
||||||
// fill the remainder of the file with random bytes
|
|
||||||
int nw = (int)(NUM_WRITES - initPos/BYTES_PER_WRITE);
|
|
||||||
for (int i = 0; i < nw; i++) {
|
|
||||||
byte[] rndBytes = createRandomBytes(BYTES_PER_WRITE, 0);
|
|
||||||
ByteBuffer src = ByteBuffer.wrap(rndBytes);
|
|
||||||
fc.write(src);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// create the target file with a hint to be sparse
|
|
||||||
try (FileChannel fc = FileChannel.open(targetFile, CREATE_NEW, WRITE, SPARSE);) {
|
|
||||||
}
|
|
||||||
|
|
||||||
// perform actual transfer, effectively by multiple invocations
|
|
||||||
// of Filechannel.transferTo(FileChannel)
|
|
||||||
try (InputStream inputStream = Channels.newInputStream(FileChannel.open(sourceFile));
|
|
||||||
OutputStream outputStream = Channels.newOutputStream(FileChannel.open(targetFile, WRITE))) {
|
|
||||||
long count = inputStream.transferTo(outputStream);
|
|
||||||
|
|
||||||
// compare reported transferred bytes, must be 3 GB
|
|
||||||
// less the value of the initial position
|
|
||||||
assertEquals(count, BYTES_WRITTEN - initPos);
|
|
||||||
}
|
|
||||||
|
|
||||||
// compare content of both files, failing if different
|
|
||||||
assertEquals(Files.mismatch(sourceFile, targetFile), -1);
|
|
||||||
|
|
||||||
} finally {
|
|
||||||
Files.delete(targetFile);
|
|
||||||
}
|
|
||||||
} finally {
|
|
||||||
Files.delete(sourceFile);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
|
||||||
* Special test of whether selectable channel based transfer throws blocking
|
|
||||||
* mode exception.
|
|
||||||
*/
|
*/
|
||||||
@Test
|
@Test
|
||||||
public void testIllegalBlockingMode() throws IOException {
|
public void testIllegalBlockingMode() throws IOException {
|
||||||
|
@ -256,70 +143,14 @@ public class TransferTo {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/*
|
|
||||||
* Asserts that the transferred content is correct, i.e., compares the bytes
|
|
||||||
* actually transferred to those expected. The position of the input and
|
|
||||||
* output streams before the transfer are zero (BOF).
|
|
||||||
*/
|
|
||||||
private static void checkTransferredContents(InputStreamProvider inputStreamProvider,
|
|
||||||
OutputStreamProvider outputStreamProvider, byte[] inBytes) throws Exception {
|
|
||||||
checkTransferredContents(inputStreamProvider, outputStreamProvider, inBytes, 0, 0);
|
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
|
||||||
* Asserts that the transferred content is correct, i. e. compares the bytes
|
|
||||||
* actually transferred to those expected. The positions of the input and
|
|
||||||
* output streams before the transfer are provided by the caller.
|
|
||||||
*/
|
|
||||||
private static void checkTransferredContents(InputStreamProvider inputStreamProvider,
|
|
||||||
OutputStreamProvider outputStreamProvider, byte[] inBytes, int posIn, int posOut) throws Exception {
|
|
||||||
AtomicReference<Supplier<byte[]>> recorder = new AtomicReference<>();
|
|
||||||
try (InputStream in = inputStreamProvider.input(inBytes);
|
|
||||||
OutputStream out = outputStreamProvider.output(recorder::set)) {
|
|
||||||
// skip bytes until starting position
|
|
||||||
in.skipNBytes(posIn);
|
|
||||||
out.write(new byte[posOut]);
|
|
||||||
|
|
||||||
long reported = in.transferTo(out);
|
|
||||||
int count = inBytes.length - posIn;
|
|
||||||
|
|
||||||
assertEquals(reported, count, format("reported %d bytes but should report %d", reported, count));
|
|
||||||
|
|
||||||
byte[] outBytes = recorder.get().get();
|
|
||||||
assertTrue(Arrays.equals(inBytes, posIn, posIn + count, outBytes, posOut, posOut + count),
|
|
||||||
format("inBytes.length=%d, outBytes.length=%d", count, outBytes.length));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
|
||||||
* Creates an array of random size (between min and min + maxRandomAdditive)
|
|
||||||
* filled with random bytes
|
|
||||||
*/
|
|
||||||
private static byte[] createRandomBytes(int min, int maxRandomAdditive) {
|
|
||||||
byte[] bytes = new byte[min + (maxRandomAdditive == 0 ? 0 : RND.nextInt(maxRandomAdditive))];
|
|
||||||
RND.nextBytes(bytes);
|
|
||||||
return bytes;
|
|
||||||
}
|
|
||||||
|
|
||||||
private interface InputStreamProvider {
|
|
||||||
InputStream input(byte... bytes) throws Exception;
|
|
||||||
}
|
|
||||||
|
|
||||||
private interface OutputStreamProvider {
|
|
||||||
OutputStream output(Consumer<Supplier<byte[]>> spy) throws Exception;
|
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
/*
|
||||||
* Creates a provider for an output stream which does not wrap a channel
|
* Creates a provider for an output stream which does not wrap a channel
|
||||||
*/
|
*/
|
||||||
private static OutputStreamProvider defaultOutput() {
|
private static OutputStreamProvider defaultOutput() {
|
||||||
return new OutputStreamProvider() {
|
return spy -> {
|
||||||
@Override
|
|
||||||
public OutputStream output(Consumer<Supplier<byte[]>> spy) {
|
|
||||||
ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
|
ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
|
||||||
spy.accept(outputStream::toByteArray);
|
spy.accept(outputStream::toByteArray);
|
||||||
return outputStream;
|
return outputStream;
|
||||||
}
|
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -327,53 +158,19 @@ public class TransferTo {
|
||||||
* Creates a provider for an input stream which wraps a file channel
|
* Creates a provider for an input stream which wraps a file channel
|
||||||
*/
|
*/
|
||||||
private static InputStreamProvider fileChannelInput() {
|
private static InputStreamProvider fileChannelInput() {
|
||||||
return new InputStreamProvider() {
|
return bytes -> {
|
||||||
@Override
|
|
||||||
public InputStream input(byte... bytes) throws Exception {
|
|
||||||
Path path = Files.createTempFile(CWD, "fileChannelInput", null);
|
Path path = Files.createTempFile(CWD, "fileChannelInput", null);
|
||||||
Files.write(path, bytes);
|
Files.write(path, bytes);
|
||||||
FileChannel fileChannel = FileChannel.open(path);
|
FileChannel fileChannel = FileChannel.open(path);
|
||||||
return Channels.newInputStream(fileChannel);
|
return Channels.newInputStream(fileChannel);
|
||||||
}
|
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
/*
|
/*
|
||||||
* Creates a provider for an input stream which wraps a readable byte
|
* Creates a provider for an output stream which wraps a selectable channel
|
||||||
* channel but is not a file channel
|
|
||||||
*/
|
*/
|
||||||
private static InputStreamProvider readableByteChannelInput() {
|
private static OutputStreamProvider selectableChannelOutput() {
|
||||||
return new InputStreamProvider() {
|
return spy -> {
|
||||||
@Override
|
|
||||||
public InputStream input(byte... bytes) throws Exception {
|
|
||||||
return Channels.newInputStream(Channels.newChannel(new ByteArrayInputStream(bytes)));
|
|
||||||
}
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
|
||||||
* Creates a provider for an output stream which wraps a file channel
|
|
||||||
*/
|
|
||||||
private static OutputStreamProvider fileChannelOutput() {
|
|
||||||
return new OutputStreamProvider() {
|
|
||||||
public OutputStream output(Consumer<Supplier<byte[]>> spy) throws Exception {
|
|
||||||
Path path = Files.createTempFile(CWD, "fileChannelOutput", null);
|
|
||||||
FileChannel fileChannel = FileChannel.open(path, WRITE);
|
|
||||||
spy.accept(() -> {
|
|
||||||
try {
|
|
||||||
return Files.readAllBytes(path);
|
|
||||||
} catch (IOException e) {
|
|
||||||
throw new AssertionError("Failed to verify output file", e);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
return Channels.newOutputStream(fileChannel);
|
|
||||||
}
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
private static OutputStreamProvider selectableChannelOutput() throws IOException {
|
|
||||||
return new OutputStreamProvider() {
|
|
||||||
public OutputStream output(Consumer<Supplier<byte[]>> spy) throws Exception {
|
|
||||||
Pipe pipe = Pipe.open();
|
Pipe pipe = Pipe.open();
|
||||||
Future<byte[]> bytes = CompletableFuture.supplyAsync(() -> {
|
Future<byte[]> bytes = CompletableFuture.supplyAsync(() -> {
|
||||||
try {
|
try {
|
||||||
|
@ -393,17 +190,18 @@ public class TransferTo {
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
return os;
|
return os;
|
||||||
}
|
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Creates a provider for an output stream which wraps a writable byte channel but is not a file channel
|
||||||
|
*/
|
||||||
private static OutputStreamProvider writableByteChannelOutput() {
|
private static OutputStreamProvider writableByteChannelOutput() {
|
||||||
return new OutputStreamProvider() {
|
return spy -> {
|
||||||
public OutputStream output(Consumer<Supplier<byte[]>> spy) throws Exception {
|
|
||||||
ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
|
ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
|
||||||
spy.accept(outputStream::toByteArray);
|
spy.accept(outputStream::toByteArray);
|
||||||
return Channels.newOutputStream(Channels.newChannel(outputStream));
|
return Channels.newOutputStream(Channels.newChannel(outputStream));
|
||||||
}
|
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
107
test/jdk/java/nio/channels/Channels/TransferTo2.java
Normal file
107
test/jdk/java/nio/channels/Channels/TransferTo2.java
Normal file
|
@ -0,0 +1,107 @@
|
||||||
|
/*
|
||||||
|
* Copyright (c) 2021, 2023, 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.
|
||||||
|
*
|
||||||
|
* 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.
|
||||||
|
*/
|
||||||
|
|
||||||
|
import java.io.OutputStream;
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.nio.channels.Channels;
|
||||||
|
import java.nio.channels.Pipe;
|
||||||
|
|
||||||
|
import org.testng.annotations.DataProvider;
|
||||||
|
import org.testng.annotations.Test;
|
||||||
|
|
||||||
|
import static java.lang.String.format;
|
||||||
|
|
||||||
|
/*
|
||||||
|
* @test
|
||||||
|
* @library /test/lib
|
||||||
|
* @build jdk.test.lib.RandomFactory
|
||||||
|
* @run testng/othervm/timeout=180 TransferTo2
|
||||||
|
* @bug 8278268
|
||||||
|
* @summary Tests FileChannel.transferFrom() optimized case
|
||||||
|
* @key randomness
|
||||||
|
*/
|
||||||
|
public class TransferTo2 extends TransferToBase {
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Provides test scenarios, i.e., combinations of input and output streams
|
||||||
|
* to be tested.
|
||||||
|
*/
|
||||||
|
@DataProvider
|
||||||
|
public static Object[][] streamCombinations() {
|
||||||
|
return new Object[][] {
|
||||||
|
// tests FileChannel.transferFrom(SelectableChannelOutput) optimized case
|
||||||
|
{selectableChannelInput(), fileChannelOutput()},
|
||||||
|
|
||||||
|
// tests FileChannel.transferFrom(ReadableByteChannelInput) optimized case
|
||||||
|
{readableByteChannelInput(), fileChannelOutput()},
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Input streams to be tested.
|
||||||
|
*/
|
||||||
|
@DataProvider
|
||||||
|
public static Object[][] inputStreamProviders() {
|
||||||
|
return new Object[][] {
|
||||||
|
{selectableChannelInput()},
|
||||||
|
{readableByteChannelInput()}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Testing API compliance: input stream must throw NullPointerException
|
||||||
|
* when parameter "out" is null.
|
||||||
|
*/
|
||||||
|
@Test(dataProvider = "inputStreamProviders")
|
||||||
|
public void testNullPointerException(InputStreamProvider inputStreamProvider) {
|
||||||
|
assertNullPointerException(inputStreamProvider);
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Testing API compliance: complete content of input stream must be
|
||||||
|
* transferred to output stream.
|
||||||
|
*/
|
||||||
|
@Test(dataProvider = "streamCombinations")
|
||||||
|
public void testStreamContents(InputStreamProvider inputStreamProvider,
|
||||||
|
OutputStreamProvider outputStreamProvider) throws Exception {
|
||||||
|
assertStreamContents(inputStreamProvider, outputStreamProvider);
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Creates a provider for an input stream which wraps a selectable channel
|
||||||
|
*/
|
||||||
|
private static InputStreamProvider selectableChannelInput() {
|
||||||
|
return bytes -> {
|
||||||
|
Pipe pipe = Pipe.open();
|
||||||
|
new Thread(() -> {
|
||||||
|
try (OutputStream os = Channels.newOutputStream(pipe.sink())) {
|
||||||
|
os.write(bytes);
|
||||||
|
} catch (IOException e) {
|
||||||
|
throw new RuntimeException(e);
|
||||||
|
}
|
||||||
|
}).start();
|
||||||
|
return Channels.newInputStream(pipe.source());
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
255
test/jdk/java/nio/channels/Channels/TransferToBase.java
Normal file
255
test/jdk/java/nio/channels/Channels/TransferToBase.java
Normal file
|
@ -0,0 +1,255 @@
|
||||||
|
/*
|
||||||
|
* Copyright (c) 2021, 2023, 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.
|
||||||
|
*
|
||||||
|
* 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.
|
||||||
|
*/
|
||||||
|
|
||||||
|
import java.io.ByteArrayInputStream;
|
||||||
|
import java.io.InputStream;
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.io.OutputStream;
|
||||||
|
import java.nio.ByteBuffer;
|
||||||
|
import java.nio.channels.Channels;
|
||||||
|
import java.nio.channels.FileChannel;
|
||||||
|
import java.nio.file.Files;
|
||||||
|
import java.nio.file.Path;
|
||||||
|
import java.util.Arrays;
|
||||||
|
import java.util.Random;
|
||||||
|
import java.util.concurrent.atomic.AtomicReference;
|
||||||
|
import java.util.function.BiFunction;
|
||||||
|
import java.util.function.Consumer;
|
||||||
|
import java.util.function.Supplier;
|
||||||
|
import java.util.function.ToLongBiFunction;
|
||||||
|
|
||||||
|
import jdk.test.lib.RandomFactory;
|
||||||
|
|
||||||
|
import static java.lang.String.format;
|
||||||
|
import static java.nio.file.StandardOpenOption.*;
|
||||||
|
|
||||||
|
import static org.testng.Assert.assertEquals;
|
||||||
|
import static org.testng.Assert.assertThrows;
|
||||||
|
import static org.testng.Assert.assertTrue;
|
||||||
|
|
||||||
|
class TransferToBase {
|
||||||
|
static final int MIN_SIZE = 10_000;
|
||||||
|
static final int MAX_SIZE_INCR = 100_000_000 - MIN_SIZE;
|
||||||
|
|
||||||
|
static final int ITERATIONS = 10;
|
||||||
|
|
||||||
|
static final int NUM_WRITES = 3*1024;
|
||||||
|
static final int BYTES_PER_WRITE = 1024*1024;
|
||||||
|
static final long BYTES_WRITTEN = (long) NUM_WRITES*BYTES_PER_WRITE;
|
||||||
|
|
||||||
|
static final Random RND = RandomFactory.getRandom();
|
||||||
|
|
||||||
|
static final Path CWD = Path.of(".");
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Asserts that the transferred content is correct, i.e., compares the bytes
|
||||||
|
* actually transferred to those expected. The position of the input and
|
||||||
|
* output streams before the transfer are zero (BOF).
|
||||||
|
*/
|
||||||
|
static void checkTransferredContents(InputStreamProvider inputStreamProvider,
|
||||||
|
OutputStreamProvider outputStreamProvider, byte[] inBytes) throws Exception {
|
||||||
|
checkTransferredContents(inputStreamProvider, outputStreamProvider, inBytes, 0, 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Asserts that the transferred content is correct, i.e. compares the bytes
|
||||||
|
* actually transferred to those expected. The positions of the input and
|
||||||
|
* output streams before the transfer are provided by the caller.
|
||||||
|
*/
|
||||||
|
static void checkTransferredContents(InputStreamProvider inputStreamProvider,
|
||||||
|
OutputStreamProvider outputStreamProvider, byte[] inBytes, int posIn, int posOut) throws Exception {
|
||||||
|
AtomicReference<Supplier<byte[]>> recorder = new AtomicReference<>();
|
||||||
|
try (InputStream in = inputStreamProvider.input(inBytes);
|
||||||
|
OutputStream out = outputStreamProvider.output(recorder::set)) {
|
||||||
|
// skip bytes until starting position
|
||||||
|
in.skipNBytes(posIn);
|
||||||
|
out.write(new byte[posOut]);
|
||||||
|
|
||||||
|
long reported = in.transferTo(out);
|
||||||
|
int count = inBytes.length - posIn;
|
||||||
|
|
||||||
|
assertEquals(reported, count, format("reported %d bytes but should report %d", reported, count));
|
||||||
|
|
||||||
|
byte[] outBytes = recorder.get().get();
|
||||||
|
assertTrue(Arrays.equals(inBytes, posIn, posIn + count, outBytes, posOut, posOut + count),
|
||||||
|
format("inBytes.length=%d, outBytes.length=%d", count, outBytes.length));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Creates an array of random size (between min and min + maxRandomAdditive)
|
||||||
|
* filled with random bytes
|
||||||
|
*/
|
||||||
|
static byte[] createRandomBytes(int min, int maxRandomAdditive) {
|
||||||
|
byte[] bytes = new byte[min + (maxRandomAdditive == 0 ? 0 : RND.nextInt(maxRandomAdditive))];
|
||||||
|
RND.nextBytes(bytes);
|
||||||
|
return bytes;
|
||||||
|
}
|
||||||
|
|
||||||
|
interface InputStreamProvider {
|
||||||
|
InputStream input(byte... bytes) throws Exception;
|
||||||
|
}
|
||||||
|
|
||||||
|
interface OutputStreamProvider {
|
||||||
|
OutputStream output(Consumer<Supplier<byte[]>> spy) throws Exception;
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Creates a provider for an input stream which wraps a readable byte
|
||||||
|
* channel but is not a file channel
|
||||||
|
*/
|
||||||
|
static InputStreamProvider readableByteChannelInput() {
|
||||||
|
return bytes -> Channels.newInputStream(Channels.newChannel(new ByteArrayInputStream(bytes)));
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Creates a provider for an output stream which wraps a file channel
|
||||||
|
*/
|
||||||
|
static OutputStreamProvider fileChannelOutput() {
|
||||||
|
return spy -> {
|
||||||
|
Path path = Files.createTempFile(CWD, "fileChannelOutput", null);
|
||||||
|
FileChannel fileChannel = FileChannel.open(path, WRITE);
|
||||||
|
spy.accept(() -> {
|
||||||
|
try {
|
||||||
|
return Files.readAllBytes(path);
|
||||||
|
} catch (IOException e) {
|
||||||
|
throw new AssertionError("Failed to verify output file", e);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
return Channels.newOutputStream(fileChannel);
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Testing API compliance: input stream must throw NullPointerException
|
||||||
|
* when parameter "out" is null.
|
||||||
|
*/
|
||||||
|
static void assertNullPointerException(InputStreamProvider inputStreamProvider) {
|
||||||
|
// tests empty input stream
|
||||||
|
assertThrows(NullPointerException.class, () -> inputStreamProvider.input().transferTo(null));
|
||||||
|
|
||||||
|
// tests single-byte input stream
|
||||||
|
assertThrows(NullPointerException.class, () -> inputStreamProvider.input((byte) 1).transferTo(null));
|
||||||
|
|
||||||
|
// tests dual-byte input stream
|
||||||
|
assertThrows(NullPointerException.class, () -> inputStreamProvider.input((byte) 1, (byte) 2).transferTo(null));
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Testing API compliance: complete content of input stream must be
|
||||||
|
* transferred to output stream.
|
||||||
|
*/
|
||||||
|
static void assertStreamContents(InputStreamProvider inputStreamProvider,
|
||||||
|
OutputStreamProvider outputStreamProvider) throws Exception {
|
||||||
|
// tests empty input stream
|
||||||
|
checkTransferredContents(inputStreamProvider, outputStreamProvider, new byte[0]);
|
||||||
|
|
||||||
|
// tests input stream with a length between 1k and 4k
|
||||||
|
checkTransferredContents(inputStreamProvider, outputStreamProvider, createRandomBytes(1024, 4096));
|
||||||
|
|
||||||
|
// tests input stream with several data chunks, as 16k is more than a
|
||||||
|
// single chunk can hold
|
||||||
|
checkTransferredContents(inputStreamProvider, outputStreamProvider, createRandomBytes(16384, 16384));
|
||||||
|
|
||||||
|
// tests randomly chosen starting positions within source and
|
||||||
|
// target stream
|
||||||
|
for (int i = 0; i < ITERATIONS; i++) {
|
||||||
|
byte[] inBytes = createRandomBytes(MIN_SIZE, MAX_SIZE_INCR);
|
||||||
|
int posIn = RND.nextInt(inBytes.length);
|
||||||
|
int posOut = RND.nextInt(MIN_SIZE);
|
||||||
|
checkTransferredContents(inputStreamProvider, outputStreamProvider, inBytes, posIn, posOut);
|
||||||
|
}
|
||||||
|
|
||||||
|
// tests reading beyond source EOF (must not transfer any bytes)
|
||||||
|
checkTransferredContents(inputStreamProvider, outputStreamProvider, createRandomBytes(4096, 0), 4096, 0);
|
||||||
|
|
||||||
|
// tests writing beyond target EOF (must extend output stream)
|
||||||
|
checkTransferredContents(inputStreamProvider, outputStreamProvider, createRandomBytes(4096, 0), 0, 4096);
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Special test for stream-to-file / file-to-stream transfer of more than 2 GB.
|
||||||
|
*/
|
||||||
|
static void testMoreThanTwoGB(String direction, BiFunction<Path, Path, InputStream> inputStreamProvider,
|
||||||
|
BiFunction<Path, Path, OutputStream> outputStreamProvider,
|
||||||
|
ToLongBiFunction<InputStream, OutputStream> transfer) throws IOException {
|
||||||
|
// prepare two temporary files to be compared at the end of the test
|
||||||
|
// set the source file name
|
||||||
|
String sourceName = String.format("test3GBSource_transfer%s%s.tmp", direction,
|
||||||
|
RND.nextInt(Integer.MAX_VALUE));
|
||||||
|
Path sourceFile = CWD.resolve(sourceName);
|
||||||
|
|
||||||
|
try {
|
||||||
|
// set the target file name
|
||||||
|
String targetName = String.format("test3GBTarget_transfer%s%s.tmp", direction,
|
||||||
|
RND.nextInt(Integer.MAX_VALUE));
|
||||||
|
Path targetFile = CWD.resolve(targetName);
|
||||||
|
|
||||||
|
try {
|
||||||
|
// calculate initial position to be just short of 2GB
|
||||||
|
final long initPos = 2047*BYTES_PER_WRITE;
|
||||||
|
|
||||||
|
// create the source file with a hint to be sparse
|
||||||
|
try (FileChannel fc = FileChannel.open(sourceFile, CREATE_NEW, SPARSE, WRITE, APPEND)) {
|
||||||
|
// set initial position to avoid writing nearly 2GB
|
||||||
|
fc.position(initPos);
|
||||||
|
|
||||||
|
// Add random bytes to the remainder of the file
|
||||||
|
long pos = initPos;
|
||||||
|
while (pos < BYTES_WRITTEN) {
|
||||||
|
int len = Math.toIntExact(BYTES_WRITTEN - pos);
|
||||||
|
if (len > BYTES_PER_WRITE)
|
||||||
|
len = BYTES_PER_WRITE;
|
||||||
|
byte[] rndBytes = createRandomBytes(len, 0);
|
||||||
|
ByteBuffer src = ByteBuffer.wrap(rndBytes);
|
||||||
|
pos += fc.write(src);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// create the target file with a hint to be sparse
|
||||||
|
try (FileChannel fc = FileChannel.open(targetFile, CREATE_NEW, WRITE, SPARSE)) {
|
||||||
|
}
|
||||||
|
|
||||||
|
// performing actual transfer, effectively by multiple invocations of
|
||||||
|
// FileChannel.transferFrom(ReadableByteChannel) / FileChannel.transferTo(WritableByteChannel)
|
||||||
|
try (InputStream inputStream = inputStreamProvider.apply(sourceFile, targetFile);
|
||||||
|
OutputStream outputStream = outputStreamProvider.apply(sourceFile, targetFile)) {
|
||||||
|
long count = transfer.applyAsLong(inputStream, outputStream);
|
||||||
|
|
||||||
|
// compare reported transferred bytes, must be 3 GB
|
||||||
|
// less the value of the initial position
|
||||||
|
assertEquals(count, BYTES_WRITTEN - initPos);
|
||||||
|
}
|
||||||
|
|
||||||
|
// compare content of both files, failing if different
|
||||||
|
assertEquals(Files.mismatch(sourceFile, targetFile), -1);
|
||||||
|
|
||||||
|
} finally {
|
||||||
|
Files.delete(targetFile);
|
||||||
|
}
|
||||||
|
} finally {
|
||||||
|
Files.delete(sourceFile);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
|
@ -0,0 +1,82 @@
|
||||||
|
/*
|
||||||
|
* Copyright (c) 2021, 2023, 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.
|
||||||
|
*
|
||||||
|
* 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.
|
||||||
|
*/
|
||||||
|
|
||||||
|
import java.io.BufferedInputStream;
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.io.UncheckedIOException;
|
||||||
|
import java.nio.channels.Channels;
|
||||||
|
import java.nio.channels.FileChannel;
|
||||||
|
import java.nio.file.Files;
|
||||||
|
|
||||||
|
import static java.nio.file.StandardOpenOption.*;
|
||||||
|
|
||||||
|
import org.testng.annotations.Test;
|
||||||
|
|
||||||
|
/*
|
||||||
|
* @test
|
||||||
|
* @library /test/lib
|
||||||
|
* @build jdk.test.lib.RandomFactory
|
||||||
|
* @run testng/othervm/timeout=180 TransferTo_2GB_transferFrom
|
||||||
|
* @bug 8278268
|
||||||
|
* @summary Tests if ChannelInputStream.transferFrom correctly
|
||||||
|
* transfers 2GB+ using FileChannel.transferFrom(ReadableByteChannel).
|
||||||
|
* @key randomness
|
||||||
|
*/
|
||||||
|
public class TransferTo_2GB_transferFrom extends TransferToBase {
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Special test for stream-to-file transfer of more than 2 GB. This test
|
||||||
|
* covers multiple iterations of FileChannel.transferFrom(ReadableByteChannel),
|
||||||
|
* which ChannelInputStream.transferFrom() only applies in this particular
|
||||||
|
* case, and cannot get tested using a single byte[] due to size limitation
|
||||||
|
* of arrays.
|
||||||
|
*/
|
||||||
|
@Test
|
||||||
|
public void testMoreThanTwoGB() throws IOException {
|
||||||
|
testMoreThanTwoGB("From",
|
||||||
|
(sourceFile, targetFile) -> {
|
||||||
|
try {
|
||||||
|
return Channels.newInputStream(
|
||||||
|
Channels.newChannel(new BufferedInputStream(Files.newInputStream(sourceFile))));
|
||||||
|
} catch (IOException e) {
|
||||||
|
throw new UncheckedIOException(e);
|
||||||
|
}
|
||||||
|
},
|
||||||
|
(sourceFile, targetFile) -> {
|
||||||
|
try {
|
||||||
|
return Channels.newOutputStream(FileChannel.open(targetFile, WRITE));
|
||||||
|
} catch (IOException e) {
|
||||||
|
throw new UncheckedIOException(e);
|
||||||
|
}
|
||||||
|
},
|
||||||
|
(inputStream, outputStream) -> {
|
||||||
|
try {
|
||||||
|
return inputStream.transferTo(outputStream);
|
||||||
|
} catch (IOException e) {
|
||||||
|
throw new UncheckedIOException(e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
|
@ -0,0 +1,80 @@
|
||||||
|
/*
|
||||||
|
* Copyright (c) 2021, 2023, 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.
|
||||||
|
*
|
||||||
|
* 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.
|
||||||
|
*/
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.io.UncheckedIOException;
|
||||||
|
import java.nio.channels.Channels;
|
||||||
|
import java.nio.channels.FileChannel;
|
||||||
|
import java.nio.file.Files;
|
||||||
|
|
||||||
|
import static java.nio.file.StandardOpenOption.*;
|
||||||
|
|
||||||
|
import org.testng.annotations.Test;
|
||||||
|
|
||||||
|
/*
|
||||||
|
* @test
|
||||||
|
* @library /test/lib
|
||||||
|
* @build jdk.test.lib.RandomFactory
|
||||||
|
* @run testng/othervm/timeout=180 TransferTo_2GB_transferTo
|
||||||
|
* @bug 8265891
|
||||||
|
* @summary Tests if ChannelInputStream.transferTo correctly
|
||||||
|
* transfers 2GB+ using FileChannel.transferTo(WritableByteChannel).
|
||||||
|
* @key randomness
|
||||||
|
*/
|
||||||
|
public class TransferTo_2GB_transferTo extends TransferToBase {
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Special test for file-to-stream transfer of more than 2 GB. This test
|
||||||
|
* covers multiple iterations of FileChannel.transferTo(WritableByteChannel),
|
||||||
|
* which ChannelInputStream.transferTo() only applies in this particular
|
||||||
|
* case, and cannot get tested using a single byte[] due to size limitation
|
||||||
|
* of arrays.
|
||||||
|
*/
|
||||||
|
@Test
|
||||||
|
public void testMoreThanTwoGB() throws IOException {
|
||||||
|
testMoreThanTwoGB("To",
|
||||||
|
(sourceFile, targetFile) -> {
|
||||||
|
try {
|
||||||
|
return Channels.newInputStream(FileChannel.open(sourceFile));
|
||||||
|
} catch (IOException e) {
|
||||||
|
throw new UncheckedIOException(e);
|
||||||
|
}
|
||||||
|
},
|
||||||
|
(sourceFile, targetFile) -> {
|
||||||
|
try {
|
||||||
|
return Channels.newOutputStream(FileChannel.open(targetFile, WRITE));
|
||||||
|
} catch (IOException e) {
|
||||||
|
throw new UncheckedIOException(e);
|
||||||
|
}
|
||||||
|
},
|
||||||
|
(inputStream, outputStream) -> {
|
||||||
|
try {
|
||||||
|
return inputStream.transferTo(outputStream);
|
||||||
|
} catch (IOException e) {
|
||||||
|
throw new UncheckedIOException(e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
Loading…
Add table
Add a link
Reference in a new issue