mirror of
https://github.com/tokio-rs/axum.git
synced 2024-11-26 00:56:27 +01:00
89 lines
2.2 KiB
Rust
89 lines
2.2 KiB
Rust
use bytes::Bytes;
|
|
use http::{Request, StatusCode};
|
|
use hyper::Server;
|
|
use serde::Deserialize;
|
|
use std::{
|
|
collections::HashMap,
|
|
net::SocketAddr,
|
|
sync::{Arc, Mutex},
|
|
time::Duration,
|
|
};
|
|
use tower::{make::Shared, ServiceBuilder};
|
|
use tower_http::{
|
|
add_extension::AddExtensionLayer, compression::CompressionLayer, trace::TraceLayer,
|
|
};
|
|
use tower_web::{body::Body, extract, response, Error};
|
|
|
|
#[tokio::main]
|
|
async fn main() {
|
|
tracing_subscriber::fmt::init();
|
|
|
|
// build our application with some routes
|
|
let app = tower_web::app()
|
|
.at("/get")
|
|
.get(get)
|
|
.at("/set")
|
|
.post(set)
|
|
// convert it into a `Service`
|
|
.into_service();
|
|
|
|
// add some middleware
|
|
let app = ServiceBuilder::new()
|
|
.timeout(Duration::from_secs(10))
|
|
.layer(TraceLayer::new_for_http())
|
|
.layer(CompressionLayer::new())
|
|
.layer(AddExtensionLayer::new(SharedState::default()))
|
|
.service(app);
|
|
|
|
// run it with hyper
|
|
let addr = SocketAddr::from(([127, 0, 0, 1], 3000));
|
|
tracing::debug!("listening on {}", addr);
|
|
let server = Server::bind(&addr).serve(Shared::new(app));
|
|
server.await.unwrap();
|
|
}
|
|
|
|
type SharedState = Arc<Mutex<State>>;
|
|
|
|
#[derive(Default)]
|
|
struct State {
|
|
db: HashMap<String, Bytes>,
|
|
}
|
|
|
|
#[derive(Deserialize)]
|
|
struct GetSetQueryString {
|
|
key: String,
|
|
}
|
|
|
|
async fn get(
|
|
_req: Request<Body>,
|
|
query: extract::Query<GetSetQueryString>,
|
|
state: extract::Extension<SharedState>,
|
|
) -> Result<Bytes, Error> {
|
|
let state = state.into_inner();
|
|
let db = &state.lock().unwrap().db;
|
|
|
|
let key = query.into_inner().key;
|
|
|
|
if let Some(value) = db.get(&key) {
|
|
Ok(value.clone())
|
|
} else {
|
|
Err(Error::WithStatus(StatusCode::NOT_FOUND))
|
|
}
|
|
}
|
|
|
|
async fn set(
|
|
_req: Request<Body>,
|
|
query: extract::Query<GetSetQueryString>,
|
|
value: extract::BytesMaxLength<{ 1024 * 5_000 }>, // ~5mb
|
|
state: extract::Extension<SharedState>,
|
|
) -> Result<response::Empty, Error> {
|
|
let state = state.into_inner();
|
|
let db = &mut state.lock().unwrap().db;
|
|
|
|
let key = query.into_inner().key;
|
|
let value = value.into_inner();
|
|
|
|
db.insert(key, value);
|
|
|
|
Ok(response::Empty)
|
|
}
|