]>
Commit | Line | Data |
---|---|---|
fded1f31 DM |
1 | use failure::*; |
2 | use futures::*; | |
3 | ||
4 | // Simple H2 server to test H2 speed with h2client.rs | |
5 | ||
6 | use tokio::net::TcpListener; | |
7 | ||
8 | use proxmox_backup::client::pipe_to_stream::*; | |
9 | ||
10 | pub fn main() -> Result<(), Error> { | |
11 | ||
12 | start_h2_server()?; | |
13 | ||
14 | Ok(()) | |
15 | } | |
16 | ||
17 | pub fn start_h2_server() -> Result<(), Error> { | |
18 | ||
19 | let listener = TcpListener::bind(&"127.0.0.1:8008".parse().unwrap()).unwrap(); | |
20 | ||
21 | println!("listening on {:?}", listener.local_addr()); | |
22 | ||
23 | let server = listener.incoming().for_each(move |socket| { | |
24 | ||
25 | let connection = h2::server::handshake(socket) | |
26 | .map_err(Error::from) | |
27 | .and_then(|conn| { | |
28 | println!("H2 connection bound"); | |
29 | ||
30 | conn | |
31 | .map_err(Error::from) | |
32 | .for_each(|(request, mut respond)| { | |
33 | println!("GOT request: {:?}", request); | |
34 | ||
35 | let response = http::Response::builder().status(http::StatusCode::OK).body(()).unwrap(); | |
36 | ||
37 | let send = respond.send_response(response, false).unwrap(); | |
38 | let data = vec![65u8; 1024*1024]; | |
39 | PipeToSendStream::new(bytes::Bytes::from(data), send) | |
40 | .and_then(|_| { | |
41 | println!("DATA SENT"); | |
42 | Ok(()) | |
43 | }) | |
44 | }) | |
45 | }) | |
46 | .and_then(|_| { | |
47 | println!("H2 connection CLOSE !"); | |
48 | Ok(()) | |
49 | }) | |
50 | .then(|res| { | |
51 | if let Err(e) = res { | |
52 | println!(" -> err={:?}", e); | |
53 | } | |
54 | Ok(()) | |
55 | }); | |
56 | ||
57 | tokio::spawn(Box::new(connection)); | |
58 | Ok(()) | |
59 | }) | |
60 | .map_err(|e| eprintln!("accept error: {}", e)); | |
61 | ||
62 | tokio::run(server); | |
63 | ||
64 | Ok(()) | |
65 | } |