router-service 0.2.0

A simple HTTP router to be used as a tower service
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

use std::{
    collections::HashMap,
    future::Future,
    pin::Pin,
    task::{Context, Poll},
};

use futures_util::{future::BoxFuture, FutureExt};
use http::{Request, Response, StatusCode};
use tower::Service;

use crate::{RouteContext, Router};

impl<Body, Data, Error> Service<Request<Body>> for Router<Body, Data, Error>
where
    Body: Default,
    Data: Clone,
{
    type Response = Response<Body>;

    type Error = Error;

    type Future = ResponseFuture<Body, Error>;

    fn poll_ready(&mut self, _cx: &mut Context<'_>) -> Poll<Result<(), Self::Error>> {
        Poll::Ready(Ok(()))
    }

    fn call(&mut self, req: Request<Body>) -> Self::Future {
        let uri = req.uri();

        let inner = self.inner.read().unwrap();

        if let Ok(node) = inner.at(uri.path()) {
            let route = node.value;
            let ctx = RouteContext {
                params: {
                    let mut params = HashMap::with_capacity(node.params.len());
                    for (name, value) in node.params.iter() {
                        params.insert(name.into(), value.into());
                    }
                    params
                },
                data: self.data.clone(),
            };

            if let Some(handler) = route.handlers.get(req.method()) {
                return ResponseFuture((handler.0)(req, ctx));
            }

            if let Some(handler) = &route.catchall {
                return ResponseFuture((handler.0)(req, ctx));
            }
        }

        ResponseFuture(Box::pin(async move {
            Ok(Response::builder()
                .status(StatusCode::NOT_FOUND)
                .body(Body::default())
                .unwrap())
        }))
    }
}

/// A [`Future`] that resolves to a [`Response`](http::Response).
pub struct ResponseFuture<Body, Error>(BoxFuture<'static, Result<Response<Body>, Error>>);

impl<Body, Error> Future for ResponseFuture<Body, Error> {
    type Output = Result<Response<Body>, Error>;

    fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
        self.0.poll_unpin(cx)
    }
}