1 use futures_util::{FutureExt, StreamExt, SinkExt};
4 // GET /hello/warp => 200 OK with body "Hello, warp!"
5 async fn hello(name: String, agent: String) -> Result<impl warp::Reply, warp::Rejection> {
6 Ok(format!("Hello, {} from {}!", name, agent))
9 // websocat ws://127.0.0.1:3030/ws-echo
10 async fn ws_echo_connected(websocket: warp::ws::WebSocket) {
11 // echo all messages back
12 let (tx, rx) = websocket.split();
13 rx.forward(tx).map(|result| {
14 if let Err(e) = result {
15 log::warn!("websocket error: {:?}", e);
20 // websocat ws://127.0.0.1:3030/ws-rev
21 async fn ws_rev_connected(websocket: warp::ws::WebSocket) {
22 // echo all messages back
23 tokio::task::spawn(async {
24 let (mut tx, mut rx) = websocket.split();
25 while let Some(message) = rx.next().await {
26 let msg = match message {
29 log::error!("websocket error: {}", e);
33 log::info!("ws_rev_connected: got message: {:?}", msg);
39 let text = msg.to_str().unwrap();
40 let rev = text.chars().rev().collect::<String>();
41 if let Err(e) = tx.send(warp::ws::Message::text(rev)).await {
43 log::info!("peer disconnected: {}", e);
48 let mut rev = msg.into_bytes();
50 if let Err(e) = tx.send(warp::ws::Message::binary(rev)).await {
52 log::info!("peer disconnected: {}", e);
57 log::info!("ws_rev_connected ended");
61 pub fn api() -> impl Filter<Extract = impl warp::Reply, Error = warp::Rejection> + Clone {
62 let hello = warp::path!("hello" / String)
63 .and(warp::header::<String>("user-agent"))
66 let ws_echo = warp::path("ws-echo")
68 .map(|ws: warp::ws::Ws| { ws.on_upgrade(ws_echo_connected) });
70 let ws_rev = warp::path("ws-rev")
72 .map(|ws: warp::ws::Ws| { ws.on_upgrade(ws_rev_connected) });
77 .with(warp::log("warp-server"))
82 env_logger::Builder::from_env(env_logger::Env::default().default_filter_or("info")).init();
85 .run(([127, 0, 0, 1], 3030))
92 async fn test_hello() {
93 let res = warp::test::request()
95 .header("user-agent", "TestBrowser 0.1")
98 assert_eq!(res.status(), 200);
99 assert_eq!(res.body(), "Hello, rust from TestBrowser 0.1!");