@@ -8,7 +8,8 @@ extern crate tokio;
8
8
9
9
use std:: net:: SocketAddr ;
10
10
11
- use futures:: { Future , Stream } ;
11
+ use futures:: { FutureExt , StreamExt } ;
12
+ use futures:: executor:: block_on;
12
13
use tokio:: runtime:: Runtime ;
13
14
use tokio:: net:: TcpListener ;
14
15
@@ -28,13 +29,13 @@ fn get_one_at_a_time(b: &mut test::Bencher) {
28
29
29
30
b. bytes = 160 * 2 + PHRASE . len ( ) as u64 ;
30
31
b. iter ( move || {
31
- client. get ( url. clone ( ) )
32
+ block_on ( client. get ( url. clone ( ) )
32
33
. and_then ( |res| {
33
34
res. into_body ( ) . into_stream ( ) . for_each ( |_chunk| {
34
35
Ok ( ( ) )
35
- } )
36
+ } ) . map ( |_| ( ) )
36
37
} )
37
- . wait ( ) . expect ( "client wait" ) ;
38
+ ) . expect ( "client wait" ) ;
38
39
} ) ;
39
40
}
40
41
@@ -54,11 +55,11 @@ fn post_one_at_a_time(b: &mut test::Bencher) {
54
55
let mut req = Request :: new ( post. into ( ) ) ;
55
56
* req. method_mut ( ) = Method :: POST ;
56
57
* req. uri_mut ( ) = url. clone ( ) ;
57
- client. request ( req) . and_then ( |res| {
58
+ block_on ( client. request ( req) . and_then ( |res| {
58
59
res. into_body ( ) . into_stream ( ) . for_each ( |_chunk| {
59
60
Ok ( ( ) )
60
- } )
61
- } ) . wait ( ) . expect ( "client wait" ) ;
61
+ } ) . map ( |_| ( ) )
62
+ } ) ) . expect ( "client wait" ) ;
62
63
63
64
} ) ;
64
65
}
@@ -76,21 +77,22 @@ fn spawn_hello(rt: &mut Runtime) -> SocketAddr {
76
77
let service = const_service ( service_fn ( |req : Request < Body > | {
77
78
req. into_body ( )
78
79
. into_stream ( )
79
- . concat2 ( )
80
+ . concat ( )
80
81
. map ( |_| {
81
82
Response :: new ( Body :: from ( PHRASE ) )
82
83
} )
83
84
} ) ) ;
84
85
85
86
let srv = listener. incoming ( )
86
- . into_future ( )
87
+ . next ( )
87
88
. map_err ( |( e, _inc) | panic ! ( "accept error: {}" , e) )
88
89
. and_then ( move |( accepted, _inc) | {
89
90
let socket = accepted. expect ( "accepted socket" ) ;
90
91
http. serve_connection ( socket, service. new_service ( ) . expect ( "new_service" ) )
91
92
. map ( |_| ( ) )
92
93
. map_err ( |_| ( ) )
93
- } ) ;
94
- rt. spawn ( srv) ;
94
+ } )
95
+ . map_err ( |_| panic ! ( "server error" ) ) ;
96
+ rt. spawn2 ( srv) ;
95
97
return addr
96
98
}
0 commit comments