+use futures_util::{FutureExt, StreamExt};
use warp::Filter;
+// GET /hello/warp => 200 OK with body "Hello, warp!"
+async fn hello(name: String, agent: String) -> Result<impl warp::Reply, warp::Rejection> {
+ Ok(format!("Hello, {} from {}!", name, agent))
+}
+
+// websocat ws://127.0.0.1:3030/ws-echo
+async fn ws_echo_connected(websocket: warp::ws::WebSocket) {
+ // echo all messages back
+ let (tx, rx) = websocket.split();
+ rx.forward(tx).map(|result| {
+ if let Err(e) = result {
+ log::warn!("websocket error: {:?}", e);
+ }
+ }).await;
+}
+
#[tokio::main]
async fn main() {
- // GET /hello/warp => 200 OK with body "Hello, warp!"
+ env_logger::Builder::from_env(env_logger::Env::default().default_filter_or("info")).init();
+
let hello = warp::path!("hello" / String)
- .map(|name| format!("Hello, {}!", name));
+ .and(warp::header::<String>("user-agent"))
+ .and_then(hello);
+
+ let ws_echo = warp::path("ws-echo")
+ .and(warp::ws())
+ .map(|ws: warp::ws::Ws| {
+ ws.on_upgrade(|websocket| { ws_echo_connected(websocket) })
+ });
+
+ let api = hello
+ .or(ws_echo)
+ .with(warp::log("warp-server"));
- warp::serve(hello)
+ warp::serve(api)
.run(([127, 0, 0, 1], 3030))
.await;
}