author | iignatyev |
Tue, 25 Sep 2018 18:26:46 -0700 | |
changeset 51876 | d2b381ea8477 |
parent 51462 | 9d7d74c6f2cb |
child 52283 | ef0fed0a3953 |
child 56868 | 67c7659ecda5 |
permissions | -rw-r--r-- |
42460
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
1 |
/* |
49765 | 2 |
* Copyright (c) 2015, 2018, Oracle and/or its affiliates. All rights reserved. |
42460
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
3 |
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
4 |
* |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
5 |
* This code is free software; you can redistribute it and/or modify it |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
6 |
* under the terms of the GNU General Public License version 2 only, as |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
7 |
* published by the Free Software Foundation. Oracle designates this |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
8 |
* particular file as subject to the "Classpath" exception as provided |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
9 |
* by Oracle in the LICENSE file that accompanied this code. |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
10 |
* |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
11 |
* This code is distributed in the hope that it will be useful, but WITHOUT |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
12 |
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
13 |
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
14 |
* version 2 for more details (a copy is included in the LICENSE file that |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
15 |
* accompanied this code). |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
16 |
* |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
17 |
* You should have received a copy of the GNU General Public License version |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
18 |
* 2 along with this work; if not, write to the Free Software Foundation, |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
19 |
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
20 |
* |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
21 |
* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
22 |
* or visit www.oracle.com if you need additional information or have any |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
23 |
* questions. |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
24 |
*/ |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
25 |
|
49765 | 26 |
package jdk.internal.net.http; |
42460
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
27 |
|
48083 | 28 |
import java.io.EOFException; |
29 |
import java.lang.System.Logger.Level; |
|
42460
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
30 |
import java.nio.ByteBuffer; |
49765 | 31 |
import java.util.List; |
42460
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
32 |
import java.util.concurrent.CompletableFuture; |
48083 | 33 |
import java.util.concurrent.CompletionStage; |
42460
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
34 |
import java.util.concurrent.Executor; |
49765 | 35 |
import java.util.concurrent.Flow; |
36 |
import java.util.concurrent.atomic.AtomicBoolean; |
|
37 |
import java.util.concurrent.atomic.AtomicLong; |
|
48083 | 38 |
import java.util.function.Consumer; |
39 |
import java.util.function.Function; |
|
49765 | 40 |
import java.net.http.HttpHeaders; |
41 |
import java.net.http.HttpResponse; |
|
42 |
import jdk.internal.net.http.ResponseContent.BodyParser; |
|
51462
9d7d74c6f2cb
8207966: HttpClient response without content-length does not return body
michaelm
parents:
50681
diff
changeset
|
43 |
import jdk.internal.net.http.ResponseContent.UnknownLengthBodyParser; |
49765 | 44 |
import jdk.internal.net.http.common.Log; |
45 |
import jdk.internal.net.http.common.Logger; |
|
46 |
import jdk.internal.net.http.common.MinimalFuture; |
|
47 |
import jdk.internal.net.http.common.Utils; |
|
48 |
import static java.net.http.HttpClient.Version.HTTP_1_1; |
|
49 |
import static java.net.http.HttpResponse.BodySubscribers.discarding; |
|
50681 | 50 |
import static jdk.internal.net.http.common.Utils.wrapWithExtraDetail; |
51 |
import static jdk.internal.net.http.RedirectFilter.HTTP_NOT_MODIFIED; |
|
42460
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
52 |
|
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
53 |
/** |
48083 | 54 |
* Handles a HTTP/1.1 response (headers + body). |
55 |
* There can be more than one of these per Http exchange. |
|
42460
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
56 |
*/ |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
57 |
class Http1Response<T> { |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
58 |
|
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
59 |
private volatile ResponseContent content; |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
60 |
private final HttpRequestImpl request; |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
61 |
private Response response; |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
62 |
private final HttpConnection connection; |
48083 | 63 |
private HttpHeaders headers; |
42460
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
64 |
private int responseCode; |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
65 |
private final Http1Exchange<T> exchange; |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
66 |
private boolean return2Cache; // return connection to cache when finished |
48083 | 67 |
private final HeadersReader headersReader; // used to read the headers |
68 |
private final BodyReader bodyReader; // used to read the body |
|
69 |
private final Http1AsyncReceiver asyncReceiver; |
|
70 |
private volatile EOFException eof; |
|
51462
9d7d74c6f2cb
8207966: HttpClient response without content-length does not return body
michaelm
parents:
50681
diff
changeset
|
71 |
private volatile BodyParser bodyParser; |
48083 | 72 |
// max number of bytes of (fixed length) body to ignore on redirect |
73 |
private final static int MAX_IGNORE = 1024; |
|
42460
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
74 |
|
48083 | 75 |
// Revisit: can we get rid of this? |
76 |
static enum State {INITIAL, READING_HEADERS, READING_BODY, DONE} |
|
77 |
private volatile State readProgress = State.INITIAL; |
|
78 |
||
49765 | 79 |
final Logger debug = Utils.getDebugLogger(this::dbgString, Utils.DEBUG); |
80 |
final static AtomicLong responseCount = new AtomicLong(); |
|
81 |
final long id = responseCount.incrementAndGet(); |
|
50681 | 82 |
private Http1HeaderParser hd; |
48083 | 83 |
|
84 |
Http1Response(HttpConnection conn, |
|
85 |
Http1Exchange<T> exchange, |
|
86 |
Http1AsyncReceiver asyncReceiver) { |
|
87 |
this.readProgress = State.INITIAL; |
|
42460
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
88 |
this.request = exchange.request(); |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
89 |
this.exchange = exchange; |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
90 |
this.connection = conn; |
48083 | 91 |
this.asyncReceiver = asyncReceiver; |
92 |
headersReader = new HeadersReader(this::advance); |
|
93 |
bodyReader = new BodyReader(this::advance); |
|
50681 | 94 |
|
95 |
hd = new Http1HeaderParser(); |
|
96 |
readProgress = State.READING_HEADERS; |
|
97 |
headersReader.start(hd); |
|
98 |
asyncReceiver.subscribe(headersReader); |
|
42460
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
99 |
} |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
100 |
|
49765 | 101 |
String dbgTag; |
102 |
private String dbgString() { |
|
103 |
String dbg = dbgTag; |
|
104 |
if (dbg == null) { |
|
105 |
String cdbg = connection.dbgTag; |
|
106 |
if (cdbg != null) { |
|
107 |
dbgTag = dbg = "Http1Response(id=" + id + ", " + cdbg + ")"; |
|
108 |
} else { |
|
109 |
dbg = "Http1Response(id=" + id + ")"; |
|
110 |
} |
|
111 |
} |
|
112 |
return dbg; |
|
113 |
} |
|
114 |
||
115 |
// The ClientRefCountTracker is used to track the state |
|
116 |
// of a pending operation. Altough there usually is a single |
|
117 |
// point where the operation starts, it may terminate at |
|
118 |
// different places. |
|
119 |
private final class ClientRefCountTracker { |
|
120 |
final HttpClientImpl client = connection.client(); |
|
121 |
// state & 0x01 != 0 => acquire called |
|
122 |
// state & 0x02 != 0 => tryRelease called |
|
123 |
byte state; |
|
124 |
||
125 |
public synchronized void acquire() { |
|
126 |
if (state == 0) { |
|
127 |
// increment the reference count on the HttpClientImpl |
|
128 |
// to prevent the SelectorManager thread from exiting |
|
129 |
// until our operation is complete. |
|
130 |
if (debug.on()) |
|
131 |
debug.log("Operation started: incrementing ref count for %s", client); |
|
132 |
client.reference(); |
|
133 |
state = 0x01; |
|
134 |
} else { |
|
135 |
if (debug.on()) |
|
136 |
debug.log("Operation ref count for %s is already %s", |
|
137 |
client, ((state & 0x2) == 0x2) ? "released." : "incremented!" ); |
|
138 |
assert (state & 0x01) == 0 : "reference count already incremented"; |
|
139 |
} |
|
140 |
} |
|
141 |
||
142 |
public synchronized void tryRelease() { |
|
143 |
if (state == 0x01) { |
|
144 |
// decrement the reference count on the HttpClientImpl |
|
145 |
// to allow the SelectorManager thread to exit if no |
|
146 |
// other operation is pending and the facade is no |
|
147 |
// longer referenced. |
|
148 |
if (debug.on()) |
|
149 |
debug.log("Operation finished: decrementing ref count for %s", client); |
|
150 |
client.unreference(); |
|
151 |
} else if (state == 0) { |
|
152 |
if (debug.on()) |
|
153 |
debug.log("Operation finished: releasing ref count for %s", client); |
|
154 |
} else if ((state & 0x02) == 0x02) { |
|
155 |
if (debug.on()) |
|
156 |
debug.log("ref count for %s already released", client); |
|
157 |
} |
|
158 |
state |= 0x02; |
|
159 |
} |
|
160 |
} |
|
161 |
||
50681 | 162 |
private volatile boolean firstTimeAround = true; |
163 |
||
49765 | 164 |
public CompletableFuture<Response> readHeadersAsync(Executor executor) { |
165 |
if (debug.on()) |
|
166 |
debug.log("Reading Headers: (remaining: " |
|
167 |
+ asyncReceiver.remaining() +") " + readProgress); |
|
50681 | 168 |
|
169 |
if (firstTimeAround) { |
|
170 |
if (debug.on()) debug.log("First time around"); |
|
171 |
firstTimeAround = false; |
|
172 |
} else { |
|
173 |
// with expect continue we will resume reading headers + body. |
|
174 |
asyncReceiver.unsubscribe(bodyReader); |
|
175 |
bodyReader.reset(); |
|
176 |
||
177 |
hd = new Http1HeaderParser(); |
|
178 |
readProgress = State.READING_HEADERS; |
|
179 |
headersReader.reset(); |
|
180 |
headersReader.start(hd); |
|
181 |
asyncReceiver.subscribe(headersReader); |
|
182 |
} |
|
183 |
||
48083 | 184 |
CompletableFuture<State> cf = headersReader.completion(); |
185 |
assert cf != null : "parsing not started"; |
|
50681 | 186 |
if (debug.on()) { |
187 |
debug.log("headersReader is %s", |
|
188 |
cf == null ? "not yet started" |
|
189 |
: cf.isDone() ? "already completed" |
|
190 |
: "not yet completed"); |
|
191 |
} |
|
48083 | 192 |
|
193 |
Function<State, Response> lambda = (State completed) -> { |
|
194 |
assert completed == State.READING_HEADERS; |
|
49765 | 195 |
if (debug.on()) |
196 |
debug.log("Reading Headers: creating Response object;" |
|
197 |
+ " state is now " + readProgress); |
|
48083 | 198 |
asyncReceiver.unsubscribe(headersReader); |
199 |
responseCode = hd.responseCode(); |
|
200 |
headers = hd.headers(); |
|
201 |
||
202 |
response = new Response(request, |
|
203 |
exchange.getExchange(), |
|
204 |
headers, |
|
49765 | 205 |
connection, |
48083 | 206 |
responseCode, |
207 |
HTTP_1_1); |
|
49765 | 208 |
|
209 |
if (Log.headers()) { |
|
210 |
StringBuilder sb = new StringBuilder("RESPONSE HEADERS:\n"); |
|
211 |
Log.dumpHeaders(sb, " ", headers); |
|
212 |
Log.logHeaders(sb.toString()); |
|
213 |
} |
|
214 |
||
48083 | 215 |
return response; |
216 |
}; |
|
217 |
||
218 |
if (executor != null) { |
|
219 |
return cf.thenApplyAsync(lambda, executor); |
|
220 |
} else { |
|
221 |
return cf.thenApply(lambda); |
|
42460
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
222 |
} |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
223 |
} |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
224 |
|
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
225 |
private boolean finished; |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
226 |
|
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
227 |
synchronized void completed() { |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
228 |
finished = true; |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
229 |
} |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
230 |
|
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
231 |
synchronized boolean finished() { |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
232 |
return finished; |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
233 |
} |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
234 |
|
51462
9d7d74c6f2cb
8207966: HttpClient response without content-length does not return body
michaelm
parents:
50681
diff
changeset
|
235 |
/** |
9d7d74c6f2cb
8207966: HttpClient response without content-length does not return body
michaelm
parents:
50681
diff
changeset
|
236 |
* Return known fixed content length or -1 if chunked, or -2 if no content-length |
9d7d74c6f2cb
8207966: HttpClient response without content-length does not return body
michaelm
parents:
50681
diff
changeset
|
237 |
* information in which case, connection termination delimits the response body |
9d7d74c6f2cb
8207966: HttpClient response without content-length does not return body
michaelm
parents:
50681
diff
changeset
|
238 |
*/ |
42460
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
239 |
int fixupContentLen(int clen) { |
50681 | 240 |
if (request.method().equalsIgnoreCase("HEAD") || responseCode == HTTP_NOT_MODIFIED) { |
42460
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
241 |
return 0; |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
242 |
} |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
243 |
if (clen == -1) { |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
244 |
if (headers.firstValue("Transfer-encoding").orElse("") |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
245 |
.equalsIgnoreCase("chunked")) { |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
246 |
return -1; |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
247 |
} |
51462
9d7d74c6f2cb
8207966: HttpClient response without content-length does not return body
michaelm
parents:
50681
diff
changeset
|
248 |
if (responseCode == 101) { |
9d7d74c6f2cb
8207966: HttpClient response without content-length does not return body
michaelm
parents:
50681
diff
changeset
|
249 |
// this is a h2c or websocket upgrade, contentlength must be zero |
9d7d74c6f2cb
8207966: HttpClient response without content-length does not return body
michaelm
parents:
50681
diff
changeset
|
250 |
return 0; |
9d7d74c6f2cb
8207966: HttpClient response without content-length does not return body
michaelm
parents:
50681
diff
changeset
|
251 |
} |
9d7d74c6f2cb
8207966: HttpClient response without content-length does not return body
michaelm
parents:
50681
diff
changeset
|
252 |
return -2; |
42460
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
253 |
} |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
254 |
return clen; |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
255 |
} |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
256 |
|
48083 | 257 |
/** |
258 |
* Read up to MAX_IGNORE bytes discarding |
|
259 |
*/ |
|
260 |
public CompletableFuture<Void> ignoreBody(Executor executor) { |
|
261 |
int clen = (int)headers.firstValueAsLong("Content-Length").orElse(-1); |
|
262 |
if (clen == -1 || clen > MAX_IGNORE) { |
|
263 |
connection.close(); |
|
264 |
return MinimalFuture.completedFuture(null); // not treating as error |
|
265 |
} else { |
|
49765 | 266 |
return readBody(discarding(), true, executor); |
267 |
} |
|
268 |
} |
|
269 |
||
270 |
static final Flow.Subscription NOP = new Flow.Subscription() { |
|
271 |
@Override |
|
272 |
public void request(long n) { } |
|
273 |
public void cancel() { } |
|
274 |
}; |
|
275 |
||
276 |
/** |
|
277 |
* The Http1AsyncReceiver ensures that all calls to |
|
278 |
* the subscriber, including onSubscribe, occur sequentially. |
|
279 |
* There could however be some race conditions that could happen |
|
280 |
* in case of unexpected errors thrown at unexpected places, which |
|
281 |
* may cause onError to be called multiple times. |
|
282 |
* The Http1BodySubscriber will ensure that the user subscriber |
|
283 |
* is actually completed only once - and only after it is |
|
284 |
* subscribed. |
|
285 |
* @param <U> The type of response. |
|
286 |
*/ |
|
287 |
final static class Http1BodySubscriber<U> implements HttpResponse.BodySubscriber<U> { |
|
288 |
final HttpResponse.BodySubscriber<U> userSubscriber; |
|
289 |
final AtomicBoolean completed = new AtomicBoolean(); |
|
290 |
volatile Throwable withError; |
|
291 |
volatile boolean subscribed; |
|
292 |
Http1BodySubscriber(HttpResponse.BodySubscriber<U> userSubscriber) { |
|
293 |
this.userSubscriber = userSubscriber; |
|
294 |
} |
|
295 |
||
296 |
// propagate the error to the user subscriber, even if not |
|
297 |
// subscribed yet. |
|
298 |
private void propagateError(Throwable t) { |
|
299 |
assert t != null; |
|
300 |
try { |
|
301 |
// if unsubscribed at this point, it will not |
|
302 |
// get subscribed later - so do it now and |
|
303 |
// propagate the error |
|
304 |
if (subscribed == false) { |
|
305 |
subscribed = true; |
|
306 |
userSubscriber.onSubscribe(NOP); |
|
307 |
} |
|
308 |
} finally { |
|
309 |
// if onError throws then there is nothing to do |
|
310 |
// here: let the caller deal with it by logging |
|
311 |
// and closing the connection. |
|
312 |
userSubscriber.onError(t); |
|
313 |
} |
|
314 |
} |
|
315 |
||
316 |
// complete the subscriber, either normally or exceptionally |
|
317 |
// ensure that the subscriber is completed only once. |
|
318 |
private void complete(Throwable t) { |
|
319 |
if (completed.compareAndSet(false, true)) { |
|
320 |
t = withError = Utils.getCompletionCause(t); |
|
321 |
if (t == null) { |
|
322 |
assert subscribed; |
|
323 |
try { |
|
324 |
userSubscriber.onComplete(); |
|
325 |
} catch (Throwable x) { |
|
50681 | 326 |
// Simply propagate the error by calling |
327 |
// onError on the user subscriber, and let the |
|
328 |
// connection be reused since we should have received |
|
329 |
// and parsed all the bytes when we reach here. |
|
330 |
// If onError throws in turn, then we will simply |
|
331 |
// let that new exception flow up to the caller |
|
332 |
// and let it deal with it. |
|
49765 | 333 |
// (i.e: log and close the connection) |
50681 | 334 |
// Note that rethrowing here could introduce a |
335 |
// race that might cause the next send() operation to |
|
336 |
// fail as the connection has already been put back |
|
337 |
// into the cache when we reach here. |
|
338 |
propagateError(t = withError = Utils.getCompletionCause(x)); |
|
49765 | 339 |
} |
340 |
} else { |
|
341 |
propagateError(t); |
|
342 |
} |
|
343 |
} |
|
344 |
} |
|
345 |
||
346 |
@Override |
|
347 |
public CompletionStage<U> getBody() { |
|
348 |
return userSubscriber.getBody(); |
|
349 |
} |
|
350 |
@Override |
|
351 |
public void onSubscribe(Flow.Subscription subscription) { |
|
352 |
if (!subscribed) { |
|
353 |
subscribed = true; |
|
354 |
userSubscriber.onSubscribe(subscription); |
|
355 |
} else { |
|
356 |
// could be already subscribed and completed |
|
357 |
// if an unexpected error occurred before the actual |
|
358 |
// subscription - though that's not supposed |
|
359 |
// happen. |
|
360 |
assert completed.get(); |
|
361 |
} |
|
362 |
} |
|
363 |
@Override |
|
364 |
public void onNext(List<ByteBuffer> item) { |
|
365 |
assert !completed.get(); |
|
366 |
userSubscriber.onNext(item); |
|
367 |
} |
|
368 |
@Override |
|
369 |
public void onError(Throwable throwable) { |
|
370 |
complete(throwable); |
|
371 |
} |
|
372 |
@Override |
|
373 |
public void onComplete() { |
|
374 |
complete(null); |
|
48083 | 375 |
} |
42460
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
376 |
} |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
377 |
|
48083 | 378 |
public <U> CompletableFuture<U> readBody(HttpResponse.BodySubscriber<U> p, |
379 |
boolean return2Cache, |
|
380 |
Executor executor) { |
|
42460
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
381 |
this.return2Cache = return2Cache; |
49765 | 382 |
final Http1BodySubscriber<U> subscriber = new Http1BodySubscriber<>(p); |
383 |
||
384 |
final CompletableFuture<U> cf = new MinimalFuture<>(); |
|
42460
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
385 |
|
48083 | 386 |
int clen0 = (int)headers.firstValueAsLong("Content-Length").orElse(-1); |
387 |
||
42460
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
388 |
final int clen = fixupContentLen(clen0); |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
389 |
|
48083 | 390 |
// expect-continue reads headers and body twice. |
391 |
// if we reach here, we must reset the headersReader state. |
|
392 |
asyncReceiver.unsubscribe(headersReader); |
|
393 |
headersReader.reset(); |
|
49765 | 394 |
ClientRefCountTracker refCountTracker = new ClientRefCountTracker(); |
48083 | 395 |
|
49765 | 396 |
// We need to keep hold on the client facade until the |
397 |
// tracker has been incremented. |
|
398 |
connection.client().reference(); |
|
42460
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
399 |
executor.execute(() -> { |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
400 |
try { |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
401 |
content = new ResponseContent( |
49765 | 402 |
connection, clen, headers, subscriber, |
48083 | 403 |
this::onFinished |
42460
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
404 |
); |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
405 |
if (cf.isCompletedExceptionally()) { |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
406 |
// if an error occurs during subscription |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
407 |
connection.close(); |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
408 |
return; |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
409 |
} |
48083 | 410 |
// increment the reference count on the HttpClientImpl |
411 |
// to prevent the SelectorManager thread from exiting until |
|
412 |
// the body is fully read. |
|
49765 | 413 |
refCountTracker.acquire(); |
51462
9d7d74c6f2cb
8207966: HttpClient response without content-length does not return body
michaelm
parents:
50681
diff
changeset
|
414 |
bodyParser = content.getBodyParser( |
48083 | 415 |
(t) -> { |
416 |
try { |
|
417 |
if (t != null) { |
|
49944 | 418 |
try { |
419 |
subscriber.onError(t); |
|
420 |
} finally { |
|
421 |
cf.completeExceptionally(t); |
|
422 |
} |
|
48083 | 423 |
} |
424 |
} finally { |
|
425 |
bodyReader.onComplete(t); |
|
49944 | 426 |
if (t != null) { |
427 |
connection.close(); |
|
428 |
} |
|
48083 | 429 |
} |
51462
9d7d74c6f2cb
8207966: HttpClient response without content-length does not return body
michaelm
parents:
50681
diff
changeset
|
430 |
}); |
9d7d74c6f2cb
8207966: HttpClient response without content-length does not return body
michaelm
parents:
50681
diff
changeset
|
431 |
bodyReader.start(bodyParser); |
48083 | 432 |
CompletableFuture<State> bodyReaderCF = bodyReader.completion(); |
433 |
asyncReceiver.subscribe(bodyReader); |
|
434 |
assert bodyReaderCF != null : "parsing not started"; |
|
435 |
// Make sure to keep a reference to asyncReceiver from |
|
436 |
// within this |
|
437 |
CompletableFuture<?> trailingOp = bodyReaderCF.whenComplete((s,t) -> { |
|
438 |
t = Utils.getCompletionCause(t); |
|
439 |
try { |
|
49765 | 440 |
if (t == null) { |
441 |
if (debug.on()) debug.log("Finished reading body: " + s); |
|
48083 | 442 |
assert s == State.READING_BODY; |
443 |
} |
|
49765 | 444 |
if (t != null) { |
445 |
subscriber.onError(t); |
|
48083 | 446 |
cf.completeExceptionally(t); |
447 |
} |
|
448 |
} catch (Throwable x) { |
|
449 |
// not supposed to happen |
|
450 |
asyncReceiver.onReadError(x); |
|
49765 | 451 |
} finally { |
452 |
// we're done: release the ref count for |
|
453 |
// the current operation. |
|
454 |
refCountTracker.tryRelease(); |
|
48083 | 455 |
} |
456 |
}); |
|
457 |
connection.addTrailingOperation(trailingOp); |
|
42460
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
458 |
} catch (Throwable t) { |
49765 | 459 |
if (debug.on()) debug.log("Failed reading body: " + t); |
48083 | 460 |
try { |
49765 | 461 |
subscriber.onError(t); |
462 |
cf.completeExceptionally(t); |
|
48083 | 463 |
} finally { |
464 |
asyncReceiver.onReadError(t); |
|
465 |
} |
|
49765 | 466 |
} finally { |
467 |
connection.client().unreference(); |
|
42460
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
468 |
} |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
469 |
}); |
49765 | 470 |
try { |
471 |
p.getBody().whenComplete((U u, Throwable t) -> { |
|
472 |
if (t == null) |
|
473 |
cf.complete(u); |
|
474 |
else |
|
475 |
cf.completeExceptionally(t); |
|
476 |
}); |
|
477 |
} catch (Throwable t) { |
|
478 |
cf.completeExceptionally(t); |
|
479 |
asyncReceiver.setRetryOnError(false); |
|
480 |
asyncReceiver.onReadError(t); |
|
481 |
} |
|
482 |
||
483 |
return cf.whenComplete((s,t) -> { |
|
484 |
if (t != null) { |
|
485 |
// If an exception occurred, release the |
|
486 |
// ref count for the current operation, as |
|
487 |
// it may never be triggered otherwise |
|
488 |
// (BodySubscriber ofInputStream) |
|
489 |
// If there was no exception then the |
|
490 |
// ref count will be/have been released when |
|
491 |
// the last byte of the response is/was received |
|
492 |
refCountTracker.tryRelease(); |
|
493 |
} |
|
494 |
}); |
|
42460
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
495 |
} |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
496 |
|
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
497 |
|
48083 | 498 |
private void onFinished() { |
499 |
asyncReceiver.clear(); |
|
500 |
if (return2Cache) { |
|
501 |
Log.logTrace("Attempting to return connection to the pool: {0}", connection); |
|
502 |
// TODO: need to do something here? |
|
503 |
// connection.setAsyncCallbacks(null, null, null); |
|
42460
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
504 |
|
48083 | 505 |
// don't return the connection to the cache if EOF happened. |
49765 | 506 |
if (debug.on()) |
507 |
debug.log(connection.getConnectionFlow() + ": return to HTTP/1.1 pool"); |
|
48083 | 508 |
connection.closeOrReturnToCache(eof == null ? headers : null); |
509 |
} |
|
42460
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
510 |
} |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
511 |
|
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
512 |
HttpHeaders responseHeaders() { |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
513 |
return headers; |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
514 |
} |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
515 |
|
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
516 |
int responseCode() { |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
517 |
return responseCode; |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
518 |
} |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
519 |
|
48083 | 520 |
// ================ Support for plugging into Http1Receiver ================= |
521 |
// ============================================================================ |
|
42460
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
522 |
|
48083 | 523 |
// Callback: Error receiver: Consumer of Throwable. |
524 |
void onReadError(Throwable t) { |
|
525 |
Log.logError(t); |
|
526 |
Receiver<?> receiver = receiver(readProgress); |
|
527 |
if (t instanceof EOFException) { |
|
528 |
debug.log(Level.DEBUG, "onReadError: received EOF"); |
|
529 |
eof = (EOFException) t; |
|
530 |
} |
|
531 |
CompletableFuture<?> cf = receiver == null ? null : receiver.completion(); |
|
532 |
debug.log(Level.DEBUG, () -> "onReadError: cf is " |
|
533 |
+ (cf == null ? "null" |
|
534 |
: (cf.isDone() ? "already completed" |
|
535 |
: "not yet completed"))); |
|
49765 | 536 |
if (cf != null) { |
537 |
cf.completeExceptionally(t); |
|
538 |
} else { |
|
539 |
debug.log(Level.DEBUG, "onReadError", t); |
|
540 |
} |
|
48083 | 541 |
debug.log(Level.DEBUG, () -> "closing connection: cause is " + t); |
542 |
connection.close(); |
|
543 |
} |
|
544 |
||
545 |
// ======================================================================== |
|
546 |
||
547 |
private State advance(State previous) { |
|
548 |
assert readProgress == previous; |
|
549 |
switch(previous) { |
|
550 |
case READING_HEADERS: |
|
551 |
asyncReceiver.unsubscribe(headersReader); |
|
552 |
return readProgress = State.READING_BODY; |
|
553 |
case READING_BODY: |
|
554 |
asyncReceiver.unsubscribe(bodyReader); |
|
555 |
return readProgress = State.DONE; |
|
556 |
default: |
|
557 |
throw new InternalError("can't advance from " + previous); |
|
558 |
} |
|
559 |
} |
|
42460
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
560 |
|
48083 | 561 |
Receiver<?> receiver(State state) { |
562 |
switch(state) { |
|
563 |
case READING_HEADERS: return headersReader; |
|
564 |
case READING_BODY: return bodyReader; |
|
565 |
default: return null; |
|
42460
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
566 |
} |
48083 | 567 |
|
568 |
} |
|
569 |
||
570 |
static abstract class Receiver<T> |
|
571 |
implements Http1AsyncReceiver.Http1AsyncDelegate { |
|
572 |
abstract void start(T parser); |
|
573 |
abstract CompletableFuture<State> completion(); |
|
574 |
// accepts a buffer from upstream. |
|
575 |
// this should be implemented as a simple call to |
|
576 |
// accept(ref, parser, cf) |
|
577 |
public abstract boolean tryAsyncReceive(ByteBuffer buffer); |
|
578 |
public abstract void onReadError(Throwable t); |
|
579 |
// handle a byte buffer received from upstream. |
|
580 |
// this method should set the value of Http1Response.buffer |
|
581 |
// to ref.get() before beginning parsing. |
|
582 |
abstract void handle(ByteBuffer buf, T parser, |
|
583 |
CompletableFuture<State> cf); |
|
584 |
// resets this objects state so that it can be reused later on |
|
585 |
// typically puts the reference to parser and completion to null |
|
586 |
abstract void reset(); |
|
587 |
||
588 |
// accepts a byte buffer received from upstream |
|
589 |
// returns true if the buffer is fully parsed and more data can |
|
590 |
// be accepted, false otherwise. |
|
591 |
final boolean accept(ByteBuffer buf, T parser, |
|
592 |
CompletableFuture<State> cf) { |
|
593 |
if (cf == null || parser == null || cf.isDone()) return false; |
|
594 |
handle(buf, parser, cf); |
|
595 |
return !cf.isDone(); |
|
596 |
} |
|
597 |
public abstract void onSubscribe(AbstractSubscription s); |
|
598 |
public abstract AbstractSubscription subscription(); |
|
599 |
||
42460
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
600 |
} |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
601 |
|
48083 | 602 |
// Invoked with each new ByteBuffer when reading headers... |
603 |
final class HeadersReader extends Receiver<Http1HeaderParser> { |
|
604 |
final Consumer<State> onComplete; |
|
605 |
volatile Http1HeaderParser parser; |
|
606 |
volatile CompletableFuture<State> cf; |
|
607 |
volatile long count; // bytes parsed (for debug) |
|
608 |
volatile AbstractSubscription subscription; |
|
609 |
||
610 |
HeadersReader(Consumer<State> onComplete) { |
|
611 |
this.onComplete = onComplete; |
|
612 |
} |
|
613 |
||
614 |
@Override |
|
615 |
public AbstractSubscription subscription() { |
|
616 |
return subscription; |
|
617 |
} |
|
618 |
||
619 |
@Override |
|
620 |
public void onSubscribe(AbstractSubscription s) { |
|
621 |
this.subscription = s; |
|
622 |
s.request(1); |
|
623 |
} |
|
42460
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
624 |
|
48083 | 625 |
@Override |
626 |
void reset() { |
|
627 |
cf = null; |
|
628 |
parser = null; |
|
629 |
count = 0; |
|
630 |
subscription = null; |
|
631 |
} |
|
632 |
||
633 |
// Revisit: do we need to support restarting? |
|
634 |
@Override |
|
635 |
final void start(Http1HeaderParser hp) { |
|
636 |
count = 0; |
|
637 |
cf = new MinimalFuture<>(); |
|
638 |
parser = hp; |
|
639 |
} |
|
640 |
||
641 |
@Override |
|
642 |
CompletableFuture<State> completion() { |
|
643 |
return cf; |
|
644 |
} |
|
645 |
||
646 |
@Override |
|
647 |
public final boolean tryAsyncReceive(ByteBuffer ref) { |
|
648 |
boolean hasDemand = subscription.demand().tryDecrement(); |
|
649 |
assert hasDemand; |
|
650 |
boolean needsMore = accept(ref, parser, cf); |
|
651 |
if (needsMore) subscription.request(1); |
|
652 |
return needsMore; |
|
653 |
} |
|
654 |
||
655 |
@Override |
|
656 |
public final void onReadError(Throwable t) { |
|
50681 | 657 |
t = wrapWithExtraDetail(t, parser::currentStateMessage); |
48083 | 658 |
Http1Response.this.onReadError(t); |
659 |
} |
|
42460
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
660 |
|
48083 | 661 |
@Override |
662 |
final void handle(ByteBuffer b, |
|
663 |
Http1HeaderParser parser, |
|
664 |
CompletableFuture<State> cf) { |
|
665 |
assert cf != null : "parsing not started"; |
|
666 |
assert parser != null : "no parser"; |
|
667 |
try { |
|
668 |
count += b.remaining(); |
|
49765 | 669 |
if (debug.on()) |
670 |
debug.log("Sending " + b.remaining() + "/" + b.capacity() |
|
671 |
+ " bytes to header parser"); |
|
48083 | 672 |
if (parser.parse(b)) { |
673 |
count -= b.remaining(); |
|
49765 | 674 |
if (debug.on()) |
675 |
debug.log("Parsing headers completed. bytes=" + count); |
|
48083 | 676 |
onComplete.accept(State.READING_HEADERS); |
677 |
cf.complete(State.READING_HEADERS); |
|
42460
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
678 |
} |
48083 | 679 |
} catch (Throwable t) { |
49765 | 680 |
if (debug.on()) |
681 |
debug.log("Header parser failed to handle buffer: " + t); |
|
48083 | 682 |
cf.completeExceptionally(t); |
683 |
} |
|
684 |
} |
|
49765 | 685 |
|
686 |
@Override |
|
687 |
public void close(Throwable error) { |
|
688 |
// if there's no error nothing to do: the cf should/will |
|
689 |
// be completed. |
|
690 |
if (error != null) { |
|
691 |
CompletableFuture<State> cf = this.cf; |
|
692 |
if (cf != null) { |
|
693 |
if (debug.on()) |
|
694 |
debug.log("close: completing header parser CF with " + error); |
|
695 |
cf.completeExceptionally(error); |
|
696 |
} |
|
697 |
} |
|
698 |
} |
|
48083 | 699 |
} |
700 |
||
701 |
// Invoked with each new ByteBuffer when reading bodies... |
|
702 |
final class BodyReader extends Receiver<BodyParser> { |
|
703 |
final Consumer<State> onComplete; |
|
704 |
volatile BodyParser parser; |
|
705 |
volatile CompletableFuture<State> cf; |
|
706 |
volatile AbstractSubscription subscription; |
|
707 |
BodyReader(Consumer<State> onComplete) { |
|
708 |
this.onComplete = onComplete; |
|
709 |
} |
|
710 |
||
711 |
@Override |
|
712 |
void reset() { |
|
713 |
parser = null; |
|
714 |
cf = null; |
|
715 |
subscription = null; |
|
716 |
} |
|
717 |
||
718 |
// Revisit: do we need to support restarting? |
|
719 |
@Override |
|
720 |
final void start(BodyParser parser) { |
|
721 |
cf = new MinimalFuture<>(); |
|
722 |
this.parser = parser; |
|
723 |
} |
|
724 |
||
725 |
@Override |
|
726 |
CompletableFuture<State> completion() { |
|
727 |
return cf; |
|
728 |
} |
|
729 |
||
730 |
@Override |
|
731 |
public final boolean tryAsyncReceive(ByteBuffer b) { |
|
732 |
return accept(b, parser, cf); |
|
733 |
} |
|
734 |
||
735 |
@Override |
|
736 |
public final void onReadError(Throwable t) { |
|
51462
9d7d74c6f2cb
8207966: HttpClient response without content-length does not return body
michaelm
parents:
50681
diff
changeset
|
737 |
if (t instanceof EOFException && bodyParser != null && |
9d7d74c6f2cb
8207966: HttpClient response without content-length does not return body
michaelm
parents:
50681
diff
changeset
|
738 |
bodyParser instanceof UnknownLengthBodyParser) { |
9d7d74c6f2cb
8207966: HttpClient response without content-length does not return body
michaelm
parents:
50681
diff
changeset
|
739 |
((UnknownLengthBodyParser)bodyParser).complete(); |
9d7d74c6f2cb
8207966: HttpClient response without content-length does not return body
michaelm
parents:
50681
diff
changeset
|
740 |
return; |
9d7d74c6f2cb
8207966: HttpClient response without content-length does not return body
michaelm
parents:
50681
diff
changeset
|
741 |
} |
50681 | 742 |
t = wrapWithExtraDetail(t, parser::currentStateMessage); |
48083 | 743 |
Http1Response.this.onReadError(t); |
744 |
} |
|
745 |
||
746 |
@Override |
|
747 |
public AbstractSubscription subscription() { |
|
748 |
return subscription; |
|
749 |
} |
|
750 |
||
751 |
@Override |
|
752 |
public void onSubscribe(AbstractSubscription s) { |
|
753 |
this.subscription = s; |
|
49765 | 754 |
try { |
755 |
parser.onSubscribe(s); |
|
756 |
} catch (Throwable t) { |
|
757 |
cf.completeExceptionally(t); |
|
758 |
throw t; |
|
759 |
} |
|
48083 | 760 |
} |
761 |
||
762 |
@Override |
|
763 |
final void handle(ByteBuffer b, |
|
764 |
BodyParser parser, |
|
765 |
CompletableFuture<State> cf) { |
|
766 |
assert cf != null : "parsing not started"; |
|
767 |
assert parser != null : "no parser"; |
|
768 |
try { |
|
49765 | 769 |
if (debug.on()) |
770 |
debug.log("Sending " + b.remaining() + "/" + b.capacity() |
|
771 |
+ " bytes to body parser"); |
|
48083 | 772 |
parser.accept(b); |
773 |
} catch (Throwable t) { |
|
49765 | 774 |
if (debug.on()) |
775 |
debug.log("Body parser failed to handle buffer: " + t); |
|
48083 | 776 |
if (!cf.isDone()) { |
777 |
cf.completeExceptionally(t); |
|
42460
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
778 |
} |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
779 |
} |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
780 |
} |
48083 | 781 |
|
782 |
final void onComplete(Throwable closedExceptionally) { |
|
783 |
if (cf.isDone()) return; |
|
784 |
if (closedExceptionally != null) { |
|
785 |
cf.completeExceptionally(closedExceptionally); |
|
786 |
} else { |
|
787 |
onComplete.accept(State.READING_BODY); |
|
788 |
cf.complete(State.READING_BODY); |
|
789 |
} |
|
790 |
} |
|
791 |
||
792 |
@Override |
|
49765 | 793 |
public final void close(Throwable error) { |
794 |
CompletableFuture<State> cf = this.cf; |
|
795 |
if (cf != null && !cf.isDone()) { |
|
796 |
// we want to make sure dependent actions are triggered |
|
797 |
// in order to make sure the client reference count |
|
798 |
// is decremented |
|
799 |
if (error != null) { |
|
800 |
if (debug.on()) |
|
801 |
debug.log("close: completing body parser CF with " + error); |
|
802 |
cf.completeExceptionally(error); |
|
803 |
} else { |
|
804 |
if (debug.on()) |
|
805 |
debug.log("close: completing body parser CF"); |
|
806 |
cf.complete(State.READING_BODY); |
|
807 |
} |
|
808 |
} |
|
809 |
} |
|
810 |
||
811 |
@Override |
|
48083 | 812 |
public String toString() { |
813 |
return super.toString() + "/parser=" + String.valueOf(parser); |
|
814 |
} |
|
42460
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
815 |
} |
7133f144981a
8170648: Move java.net.http package out of Java SE to incubator namespace
michaelm
parents:
diff
changeset
|
816 |
} |