fregate 0.17.1

Framework for services creation
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

use axum::body::boxed;
use axum::Router;
use hyper::{Body, Request, Response};
use sealed::sealed;
use std::convert::Infallible;
use tonic::body::BoxBody;
use tonic::transport::NamedService;
use tower::{Service, ServiceBuilder};
use tower_http::ServiceBuilderExt;

/// Takes Tonic [`Service`] and converts it into [`Router`]
#[sealed]
pub trait RouterTonicExt {
    /// Takes Tonic [`Service`] and converts it into [`Router`]
    fn from_tonic_service<S>(service: S) -> Self
    where
        Self: Sized,
        S: Service<Request<Body>, Response = Response<BoxBody>, Error = Infallible>
            + NamedService
            + Clone
            + Send
            + 'static,
        S::Future: Send + 'static;
}

#[sealed]
impl RouterTonicExt for Router {
    fn from_tonic_service<S>(service: S) -> Self
    where
        Self: Sized,
        S: Service<Request<Body>, Response = Response<BoxBody>, Error = Infallible>
            + NamedService
            + Clone
            + Send
            + 'static,
        S::Future: Send + 'static,
    {
        // this piece of code is taken from:
        // https://github.com/EmbarkStudios/server-framework/blob/83ff44b0ad19e4fcbc163bc652f00e04f4143365/src/server.rs#L679-L685
        let svc = ServiceBuilder::new()
            .map_err(|err: Infallible| match err {})
            .map_response_body(boxed)
            .service(service);

        Router::new().route_service(&format!("/{}/*rest", S::NAME), svc)
    }
}