InterruptCopy.java revision 3471:236e3f2d0a6b
1/* 2 * Copyright (c) 2008, 2010, 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. 8 * 9 * This code is distributed in the hope that it will be useful, but WITHOUT 10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 12 * version 2 for more details (a copy is included in the LICENSE file that 13 * accompanied this code). 14 * 15 * You should have received a copy of the GNU General Public License version 16 * 2 along with this work; if not, write to the Free Software Foundation, 17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 18 * 19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 20 * or visit www.oracle.com if you need additional information or have any 21 * questions. 22 */ 23 24/* @test 25 * @bug 4313887 6993267 26 * @summary Unit test for Sun-specific ExtendedCopyOption.INTERRUPTIBLE option 27 * @library .. 28 * @run main/othervm -XX:-UseVMInterruptibleIO InterruptCopy 29 */ 30 31import java.nio.file.*; 32import java.io.*; 33import java.util.concurrent.*; 34import com.sun.nio.file.ExtendedCopyOption; 35 36public class InterruptCopy { 37 38 private static final long FILE_SIZE_TO_COPY = 512L * 1024L * 1024L; 39 private static final int DELAY_IN_MS = 500; 40 private static final int DURATION_MAX_IN_MS = 5000; 41 42 public static void main(String[] args) throws Exception { 43 Path dir = TestUtil.createTemporaryDirectory(); 44 try { 45 FileStore store = Files.getFileStore(dir); 46 System.out.format("Checking space (%s)\n", store); 47 long usableSpace = store.getUsableSpace(); 48 if (usableSpace < 2*FILE_SIZE_TO_COPY) { 49 System.out.println("Insufficient disk space to run test."); 50 return; 51 } 52 doTest(dir); 53 } finally { 54 TestUtil.removeAll(dir); 55 } 56 } 57 58 static void doTest(Path dir) throws Exception { 59 final Path source = dir.resolve("foo"); 60 final Path target = dir.resolve("bar"); 61 62 // create source file (don't create it as sparse file because we 63 // require the copy to take a long time) 64 System.out.println("Creating source file..."); 65 byte[] buf = new byte[32*1024]; 66 long total = 0; 67 try (OutputStream out = Files.newOutputStream(source)) { 68 do { 69 out.write(buf); 70 total += buf.length; 71 } while (total < FILE_SIZE_TO_COPY); 72 } 73 System.out.println("Source file created."); 74 75 ScheduledExecutorService pool = 76 Executors.newSingleThreadScheduledExecutor(); 77 try { 78 // copy source to target in main thread, interrupting it after a delay 79 final Thread me = Thread.currentThread(); 80 Future<?> wakeup = pool.schedule(new Runnable() { 81 public void run() { 82 me.interrupt(); 83 }}, DELAY_IN_MS, TimeUnit.MILLISECONDS); 84 System.out.println("Copying file..."); 85 try { 86 long start = System.currentTimeMillis(); 87 Files.copy(source, target, ExtendedCopyOption.INTERRUPTIBLE); 88 long duration = System.currentTimeMillis() - start; 89 if (duration > DURATION_MAX_IN_MS) 90 throw new RuntimeException("Copy was not interrupted"); 91 } catch (IOException e) { 92 boolean interrupted = Thread.interrupted(); 93 if (!interrupted) 94 throw new RuntimeException("Interrupt status was not set"); 95 System.out.println("Copy failed (this is expected)"); 96 } 97 try { 98 wakeup.get(); 99 } catch (InterruptedException ignore) { } 100 Thread.interrupted(); 101 102 // copy source to target via task in thread pool, interrupting it after 103 // a delay using cancel(true) 104 Future<Void> result = pool.submit(new Callable<Void>() { 105 public Void call() throws IOException { 106 System.out.println("Copying file..."); 107 Files.copy(source, target, ExtendedCopyOption.INTERRUPTIBLE, 108 StandardCopyOption.REPLACE_EXISTING); 109 return null; 110 } 111 }); 112 Thread.sleep(DELAY_IN_MS); 113 boolean cancelled = result.cancel(true); 114 if (!cancelled) 115 result.get(); 116 System.out.println("Copy cancelled."); 117 } finally { 118 pool.shutdown(); 119 } 120 } 121} 122