0c8ee93d7f
This removes the `tcp` feature from hyper's `Cargo.toml`, and the code it enabled: - `HttpConnector` - `GaiResolver` - `AddrStream` And parts of `Client` and `Server` that used those types. Alternatives will be available in the `hyper-util` crate. Closes #2856 Co-authored-by: MrGunflame <mrgunflame@protonmail.com>
114 lines
3.7 KiB
Rust
114 lines
3.7 KiB
Rust
#![deny(warnings)]
|
|
|
|
use std::net::SocketAddr;
|
|
|
|
use bytes::Buf;
|
|
use hyper::server::conn::Http;
|
|
use hyper::service::service_fn;
|
|
use hyper::{header, Body, Method, Request, Response, StatusCode};
|
|
use tokio::net::{TcpListener, TcpStream};
|
|
|
|
type GenericError = Box<dyn std::error::Error + Send + Sync>;
|
|
type Result<T> = std::result::Result<T, GenericError>;
|
|
|
|
static INDEX: &[u8] = b"<a href=\"test.html\">test.html</a>";
|
|
static INTERNAL_SERVER_ERROR: &[u8] = b"Internal Server Error";
|
|
static NOTFOUND: &[u8] = b"Not Found";
|
|
static POST_DATA: &str = r#"{"original": "data"}"#;
|
|
static URL: &str = "http://127.0.0.1:1337/json_api";
|
|
|
|
async fn client_request_response() -> Result<Response<Body>> {
|
|
let req = Request::builder()
|
|
.method(Method::POST)
|
|
.uri(URL)
|
|
.header(header::CONTENT_TYPE, "application/json")
|
|
.body(POST_DATA.into())
|
|
.unwrap();
|
|
|
|
let host = req.uri().host().expect("uri has no host");
|
|
let port = req.uri().port_u16().expect("uri has no port");
|
|
let stream = TcpStream::connect(format!("{}:{}", host, port)).await?;
|
|
|
|
let (mut sender, conn) = hyper::client::conn::handshake(stream).await?;
|
|
|
|
tokio::task::spawn(async move {
|
|
if let Err(err) = conn.await {
|
|
println!("Connection error: {:?}", err);
|
|
}
|
|
});
|
|
|
|
let web_res = sender.send_request(req).await?;
|
|
|
|
let res_body = web_res.into_body();
|
|
|
|
Ok(Response::new(res_body))
|
|
}
|
|
|
|
async fn api_post_response(req: Request<Body>) -> Result<Response<Body>> {
|
|
// Aggregate the body...
|
|
let whole_body = hyper::body::aggregate(req).await?;
|
|
// Decode as JSON...
|
|
let mut data: serde_json::Value = serde_json::from_reader(whole_body.reader())?;
|
|
// Change the JSON...
|
|
data["test"] = serde_json::Value::from("test_value");
|
|
// And respond with the new JSON.
|
|
let json = serde_json::to_string(&data)?;
|
|
let response = Response::builder()
|
|
.status(StatusCode::OK)
|
|
.header(header::CONTENT_TYPE, "application/json")
|
|
.body(Body::from(json))?;
|
|
Ok(response)
|
|
}
|
|
|
|
async fn api_get_response() -> Result<Response<Body>> {
|
|
let data = vec!["foo", "bar"];
|
|
let res = match serde_json::to_string(&data) {
|
|
Ok(json) => Response::builder()
|
|
.header(header::CONTENT_TYPE, "application/json")
|
|
.body(Body::from(json))
|
|
.unwrap(),
|
|
Err(_) => Response::builder()
|
|
.status(StatusCode::INTERNAL_SERVER_ERROR)
|
|
.body(INTERNAL_SERVER_ERROR.into())
|
|
.unwrap(),
|
|
};
|
|
Ok(res)
|
|
}
|
|
|
|
async fn response_examples(req: Request<Body>) -> Result<Response<Body>> {
|
|
match (req.method(), req.uri().path()) {
|
|
(&Method::GET, "/") | (&Method::GET, "/index.html") => Ok(Response::new(INDEX.into())),
|
|
(&Method::GET, "/test.html") => client_request_response().await,
|
|
(&Method::POST, "/json_api") => api_post_response(req).await,
|
|
(&Method::GET, "/json_api") => api_get_response().await,
|
|
_ => {
|
|
// Return 404 not found response.
|
|
Ok(Response::builder()
|
|
.status(StatusCode::NOT_FOUND)
|
|
.body(NOTFOUND.into())
|
|
.unwrap())
|
|
}
|
|
}
|
|
}
|
|
|
|
#[tokio::main]
|
|
async fn main() -> Result<()> {
|
|
pretty_env_logger::init();
|
|
|
|
let addr: SocketAddr = "127.0.0.1:1337".parse().unwrap();
|
|
|
|
let listener = TcpListener::bind(&addr).await?;
|
|
println!("Listening on http://{}", addr);
|
|
loop {
|
|
let (stream, _) = listener.accept().await?;
|
|
|
|
tokio::task::spawn(async move {
|
|
let service = service_fn(move |req| response_examples(req));
|
|
|
|
if let Err(err) = Http::new().serve_connection(stream, service).await {
|
|
println!("Failed to serve connection: {:?}", err);
|
|
}
|
|
});
|
|
}
|
|
}
|