frege 0.1.0

A lightweight, async HTTP router for Rust with middleware support
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77

use async_trait::async_trait;
use hyper::{Body, Request, Response};
use super::router::Router;
use std::sync::Arc;

/// Трейт для CRUD-ресурсов
#[async_trait]
pub trait Resource: Send + Sync + 'static {
    async fn get(&self, id: i32) -> Response<Body>;
    async fn create(&self, data: String) -> Response<Body>;
    async fn update(&self, id: i32, data: String) -> Response<Body>;
    async fn delete(&self, id: i32) -> Response<Body>;
}

impl Router {
    /// Регистрирует CRUD-ресурс
    pub fn resource<R: Resource>(&mut self, path: &str, resource: R) -> &mut Self {
        let resource = Arc::new(resource);

        // GET /path/:id
        self.get(&format!("{}/:id", path), {
            let resource = resource.clone();
            move |req| {
                let resource = resource.clone();
                Box::pin(async move {
                    let id = req.uri().path().split('/').next_back().unwrap().parse().unwrap();
                    resource.get(id).await
                })
            }
        });

        // POST /path
        self.post(path, {
            let resource = resource.clone();
            move |req| {
                let resource = resource.clone();
                Box::pin(async move {
                    let data = extract_body(req).await.unwrap_or_default();
                    resource.create(data).await
                })
            }
        });

        // PUT /path/:id
        self.put(&format!("{}/:id", path), {
            let resource = resource.clone();
            move |req| {
                let resource = resource.clone();
                Box::pin(async move {
                    let id = req.uri().path().split('/').next_back().unwrap().parse().unwrap();
                    let data = extract_body(req).await.unwrap_or_default();
                    resource.update(id, data).await
                })
            }
        });

        // DELETE /path/:id
        self.delete(&format!("{}/:id", path), {
            let resource = resource.clone();
            move |req| {
                let resource = resource.clone();
                Box::pin(async move {
                    let id = req.uri().path().split('/').next_back().unwrap().parse().unwrap();
                    resource.delete(id).await
                })
            }
        });

        self
    }
}

/// Вспомогательная функция для извлечения тела запроса
async fn extract_body(req: Request<Body>) -> Result<String, hyper::Error> {
    let bytes = hyper::body::to_bytes(req.into_body()).await?;
    Ok(String::from_utf8_lossy(&bytes).to_string())
}