lxy 0.1.1

A convenient async http and RPC framework in Rust
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

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

use http::Request;
use tower::Service;

use crate::app::{context::run_with_app, layer::AppService};

type GrpcRequest = Request<tonic::body::Body>;

impl<S> Service<GrpcRequest> for AppService<S>
where
  S: Service<GrpcRequest> + Clone + Send + 'static,
  S::Future: Send,
{
  type Response = S::Response;
  type Error = S::Error;
  type Future = Pin<Box<dyn Future<Output = Result<Self::Response, Self::Error>> + Send>>;

  fn poll_ready(&mut self, cx: &mut Context<'_>) -> Poll<Result<(), Self::Error>> {
    self.inner.poll_ready(cx)
  }

  fn call(&mut self, req: GrpcRequest) -> Self::Future {
    let inner = self.inner.clone();
    let mut inner = std::mem::replace(&mut self.inner, inner);
    let app = self.app.clone();

    Box::pin(async move { run_with_app(app, inner.call(req)).await })
  }
}

#[cfg(test)]
mod tests {
  use super::*;
  use crate::{
    app::{App, get_app, layer::AppLayer},
    container::Container,
  };
  use tower::{Layer, ServiceExt};

  #[tokio::test]
  async fn test_grpc_app_layer() {
    #[derive(Clone)]
    struct TestService;

    let mut app = App::new();
    app.container.bind(TestService);

    let layer = AppLayer(app.clone());
    let service = layer.layer(tower::service_fn(|_req: GrpcRequest| async {
      let app = get_app().expect("get_app should work within app layer");
      let _service = app.get::<TestService>().unwrap();
      Ok::<_, std::convert::Infallible>(http::Response::new(tonic::body::Body::empty()))
    }));

    let req = Request::builder()
      .uri("/test.Service/Method")
      .method("POST")
      .body(tonic::body::Body::empty())
      .unwrap();

    let response: http::Response<tonic::body::Body> = service.oneshot(req).await.unwrap();
    assert_eq!(response.status(), 200);
  }
}