blob: ed412fa6f5ed07969d0568466b98531a96b42bcf (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
/*
* Copyright (C) 2009-2010, Google Inc. and others
*
* This program and the accompanying materials are made available under the
* terms of the Eclipse Distribution License v. 1.0 which is available at
* https://www.eclipse.org/org/documents/edl-v10.php.
*
* SPDX-License-Identifier: BSD-3-Clause
*/
package org.eclipse.jgit.util.io;
import java.io.IOException;
import java.io.InputStream;
import java.io.InterruptedIOException;
import java.io.OutputStream;
/**
* Thread to copy from an input stream to an output stream.
*/
public class StreamCopyThread extends Thread {
private static final int BUFFER_SIZE = 1024;
private final InputStream src;
private final OutputStream dst;
private volatile boolean done;
/** Lock held by flush to avoid interrupting a write. */
private final Object writeLock;
/**
* Create a thread to copy data from an input stream to an output stream.
*
* @param i
* stream to copy from. The thread terminates when this stream
* reaches EOF. The thread closes this stream before it exits.
* @param o
* stream to copy into. The destination stream is automatically
* closed when the thread terminates.
*/
public StreamCopyThread(InputStream i, OutputStream o) {
setName(Thread.currentThread().getName() + "-StreamCopy"); //$NON-NLS-1$
src = i;
dst = o;
writeLock = new Object();
}
/**
* Request that the thread terminate, and wait for it.
* <p>
* This method signals to the copy thread that it should stop as soon as
* there is no more IO occurring.
*
* @throws java.lang.InterruptedException
* the calling thread was interrupted.
*/
public void halt() throws InterruptedException {
for (;;) {
join(250 /* milliseconds */);
if (isAlive()) {
done = true;
interrupt();
} else
break;
}
}
@Override
public void run() {
try {
final byte[] buf = new byte[BUFFER_SIZE];
boolean readInterrupted = false;
for (;;) {
try {
if (readInterrupted) {
synchronized (writeLock) {
boolean interruptedAgain = Thread.interrupted();
dst.flush();
if (interruptedAgain) {
interrupt();
}
}
readInterrupted = false;
}
if (done)
break;
final int n;
try {
n = src.read(buf);
} catch (InterruptedIOException wakey) {
readInterrupted = true;
continue;
}
if (n < 0)
break;
synchronized (writeLock) {
boolean writeInterrupted = Thread.interrupted();
dst.write(buf, 0, n);
if (writeInterrupted) {
interrupt();
}
}
} catch (IOException e) {
break;
}
}
} finally {
try {
src.close();
} catch (IOException e) {
// Ignore IO errors on close
}
try {
dst.close();
} catch (IOException e) {
// Ignore IO errors on close
}
}
}
}
|
