blob: afd798a1a28f76acd656b3667e6070f3a6faad44 (
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
126
127
128
129
130
131
132
133
134
135
136
|
/*
* Copyright (C) 2009, 2013 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.InterruptedIOException;
import java.io.OutputStream;
import java.text.MessageFormat;
import org.eclipse.jgit.internal.JGitText;
/**
* OutputStream with a configurable timeout.
*/
public class TimeoutOutputStream extends OutputStream {
private final OutputStream dst;
private final InterruptTimer myTimer;
private int timeout;
/**
* Wrap an output stream with a timeout on all write operations.
*
* @param destination
* base input stream (to write to). The stream must be
* interruptible (most socket streams are).
* @param timer
* timer to manage the timeouts during writes.
*/
public TimeoutOutputStream(final OutputStream destination,
final InterruptTimer timer) {
dst = destination;
myTimer = timer;
}
/**
* Get number of milliseconds before aborting a write.
*
* @return number of milliseconds before aborting a write.
*/
public int getTimeout() {
return timeout;
}
/**
* Set number of milliseconds before aborting a write.
*
* @param millis
* number of milliseconds before aborting a write. Must be >
* 0.
*/
public void setTimeout(int millis) {
if (millis < 0)
throw new IllegalArgumentException(MessageFormat.format(
JGitText.get().invalidTimeout, Integer.valueOf(millis)));
timeout = millis;
}
@Override
public void write(int b) throws IOException {
try {
beginWrite();
dst.write(b);
} catch (InterruptedIOException e) {
throw writeTimedOut(e);
} finally {
endWrite();
}
}
@Override
public void write(byte[] buf) throws IOException {
write(buf, 0, buf.length);
}
@Override
public void write(byte[] buf, int off, int len) throws IOException {
try {
beginWrite();
dst.write(buf, off, len);
} catch (InterruptedIOException e) {
throw writeTimedOut(e);
} finally {
endWrite();
}
}
@Override
public void flush() throws IOException {
try {
beginWrite();
dst.flush();
} catch (InterruptedIOException e) {
throw writeTimedOut(e);
} finally {
endWrite();
}
}
@Override
public void close() throws IOException {
try {
beginWrite();
dst.close();
} catch (InterruptedIOException e) {
throw writeTimedOut(e);
} finally {
endWrite();
}
}
private void beginWrite() {
myTimer.begin(timeout);
}
private void endWrite() {
myTimer.end();
}
private InterruptedIOException writeTimedOut(InterruptedIOException cause) {
InterruptedIOException e = new InterruptedIOException(
MessageFormat.format(JGitText.get().writeTimedOut,
Integer.valueOf(timeout)));
e.initCause(cause);
return e;
}
}
|
