OLD | NEW |
1 // Copyright (c) 2012, the Dart project authors. Please see the AUTHORS file | 1 // Copyright (c) 2012, the Dart project authors. Please see the AUTHORS file |
2 // for details. All rights reserved. Use of this source code is governed by a | 2 // for details. All rights reserved. Use of this source code is governed by a |
3 // BSD-style license that can be found in the LICENSE file. | 3 // BSD-style license that can be found in the LICENSE file. |
4 | 4 |
5 library io_client; | 5 library io_client; |
6 | 6 |
7 import 'dart:async'; | 7 import 'dart:async'; |
8 import 'dart:io'; | 8 import 'dart:io'; |
9 | 9 |
10 import 'base_client.dart'; | 10 import 'base_client.dart'; |
11 import 'base_request.dart'; | 11 import 'base_request.dart'; |
12 import 'streamed_response.dart'; | 12 import 'streamed_response.dart'; |
13 import 'utils.dart'; | 13 import 'utils.dart'; |
14 | 14 |
15 /// A `dart:io`-based HTTP client. This is the default client. | 15 /// A `dart:io`-based HTTP client. This is the default client. |
16 class IOClient extends BaseClient { | 16 class IOClient extends BaseClient { |
17 /// The underlying `dart:io` HTTP client. | 17 /// The underlying `dart:io` HTTP client. |
18 HttpClient _inner; | 18 HttpClient _inner; |
19 | 19 |
20 /// Creates a new HTTP client. | 20 /// Creates a new HTTP client. |
21 IOClient() : _inner = new HttpClient(); | 21 IOClient() : _inner = new HttpClient(); |
22 | 22 |
23 /// Sends an HTTP request and asynchronously returns the response. | 23 /// Sends an HTTP request and asynchronously returns the response. |
24 Future<StreamedResponse> send(BaseRequest request) { | 24 Future<StreamedResponse> send(BaseRequest request) { |
25 var stream = request.finalize(); | 25 var stream = request.finalize(); |
26 | 26 |
27 var completer = new Completer<StreamedResponse>(); | 27 return _inner.openUrl(request.method, request.url).then((ioRequest) { |
28 var connection = _inner.openUrl(request.method, request.url); | 28 ioRequest.followRedirects = request.followRedirects; |
29 bool completed = false; | 29 ioRequest.maxRedirects = request.maxRedirects; |
30 connection.followRedirects = request.followRedirects; | 30 ioRequest.contentLength = request.contentLength; |
31 connection.maxRedirects = request.maxRedirects; | 31 ioRequest.persistentConnection = request.persistentConnection; |
32 connection.onError = (e) { | 32 request.headers.forEach((name, value) { |
33 async.then((_) { | 33 ioRequest.headers.set(name, value); |
34 // TODO(nweiz): issue 4974 means that any errors that appear in the | |
35 // onRequest or onResponse callbacks get passed to onError. If the | |
36 // completer has already fired, we want to re-throw those exceptions | |
37 // to the top level so that they aren't silently ignored. | |
38 if (completed) throw e; | |
39 | |
40 completed = true; | |
41 completer.completeError(e); | |
42 }); | 34 }); |
43 }; | 35 return Future.wait([stream.pipe(ioRequest), ioRequest.response]) |
44 | 36 .then((list) => list[1]); |
45 var pipeCompleter = new Completer(); | 37 }).then((response) { |
46 connection.onRequest = (underlyingRequest) { | |
47 underlyingRequest.contentLength = request.contentLength; | |
48 underlyingRequest.persistentConnection = request.persistentConnection; | |
49 request.headers.forEach((name, value) { | |
50 underlyingRequest.headers.set(name, value); | |
51 }); | |
52 | |
53 chainToCompleter( | |
54 stream.pipe(wrapOutputStream(underlyingRequest.outputStream)), | |
55 pipeCompleter); | |
56 }; | |
57 | |
58 connection.onResponse = (response) { | |
59 var headers = {}; | 38 var headers = {}; |
60 response.headers.forEach((key, values) { | 39 response.headers.forEach((key, values) { |
61 headers[key] = values.join(','); | 40 headers[key] = values.join(','); |
62 }); | 41 }); |
63 | 42 |
64 if (completed) return; | 43 return new StreamedResponse( |
65 | 44 response, |
66 completed = true; | |
67 completer.complete(new StreamedResponse( | |
68 wrapInputStream(response.inputStream), | |
69 response.statusCode, | 45 response.statusCode, |
70 response.contentLength, | 46 response.contentLength, |
71 request: request, | 47 request: request, |
72 headers: headers, | 48 headers: headers, |
73 isRedirect: response.isRedirect, | 49 isRedirect: response.isRedirect, |
74 persistentConnection: response.persistentConnection, | 50 persistentConnection: response.persistentConnection, |
75 reasonPhrase: response.reasonPhrase)); | 51 reasonPhrase: response.reasonPhrase); |
76 }; | 52 }); |
77 | |
78 return Future.wait([ | |
79 completer.future, | |
80 pipeCompleter.future | |
81 ]).then((values) => values.first); | |
82 } | 53 } |
83 | 54 |
84 /// Closes the client. This terminates all active connections. If a client | 55 /// Closes the client. This terminates all active connections. If a client |
85 /// remains unclosed, the Dart process may not terminate. | 56 /// remains unclosed, the Dart process may not terminate. |
86 void close() { | 57 void close() { |
87 if (_inner != null) _inner.shutdown(force: true); | 58 if (_inner != null) _inner.close(force: true); |
88 _inner = null; | 59 _inner = null; |
89 } | 60 } |
90 } | 61 } |
OLD | NEW |