Introduce GrpcWebLayer implementing tower::Layer. Fix hyperium#1117

Signed-off-by: slinkydeveloper <[email protected]>
slinkydeveloper · Oct 24, 2022 · fb2fe58 · fb2fe58
1 parent 60c0c54
commit fb2fe58
Show file tree

Hide file tree

Showing 5 changed files with 47 additions and 18 deletions.
diff --git a/tonic-web/src/config.rs b/tonic-web/src/config.rs
@@ -4,7 +4,6 @@ use std::time::Duration;
 
 use http::{header::HeaderName, HeaderValue};
 use tonic::body::BoxBody;
-use tower_layer::Layer;
 use tower_service::Service;
 
 use crate::service::GrpcWeb;
@@ -165,17 +164,3 @@ impl Default for Config {
         Config::new()
     }
 }
-
-impl<S> Layer<S> for Config
-where
-    S: Service<http::Request<hyper::Body>, Response = http::Response<BoxBody>>,
-    S: Clone + Send + 'static,
-    S::Future: Send + 'static,
-    S::Error: Into<BoxError> + Send,
-{
-    type Service = GrpcWeb<S>;
-
-    fn layer(&self, inner: S) -> Self::Service {
-        self.enable(inner)
-    }
-}
diff --git a/tonic-web/src/layer.rs b/tonic-web/src/layer.rs
@@ -0,0 +1,31 @@
+use super::{BoxBody, BoxError, Config, GrpcWeb};
+
+use tower_layer::Layer;
+use tower_service::Service;
+
+/// Layer implementing the grpc-web protocol.
+#[derive(Debug)]
+pub struct GrpcWebLayer {
+    _priv: (),
+}
+
+impl GrpcWebLayer {
+    /// Create a new grpc-web layer.
+    pub fn new() -> GrpcWebLayer {
+        Self { _priv: () }
+    }
+}
+
+impl<S> Layer<S> for GrpcWebLayer
+where
+    S: Service<http::Request<hyper::Body>, Response = http::Response<BoxBody>>,
+    S: Clone + Send + 'static,
+    S::Future: Send + 'static,
+    S::Error: Into<BoxError> + Send,
+{
+    type Service = GrpcWeb<S>;
+
+    fn layer(&self, inner: S) -> Self::Service {
+        Config::default().enable(inner)
+    }
+}
diff --git a/tonic-web/src/lib.rs b/tonic-web/src/lib.rs
@@ -92,6 +92,7 @@ pub use config::Config;
 mod call;
 mod config;
 mod cors;
+mod layer;
 mod service;
 
 use crate::service::GrpcWeb;
@@ -101,6 +102,8 @@ use tonic::body::BoxBody;
 use tonic::transport::NamedService;
 use tower_service::Service;
 
+pub use layer::GrpcWebLayer;
+
 /// enable a tonic service to handle grpc-web requests with the default configuration.
 ///
 /// Shortcut for `tonic_web::config().enable(service)`

diff --git a/tonic-web/src/service.rs b/tonic-web/src/service.rs
@@ -266,6 +266,7 @@ mod tests {
     mod grpc_web {
         use super::*;
         use http::HeaderValue;
+        use tower_layer::Layer;
 
         fn request() -> Request<Body> {
             Request::builder()
@@ -284,6 +285,14 @@ mod tests {
             assert_eq!(res.status(), StatusCode::OK);
         }
 
+        #[tokio::test]
+        async fn web_layer() {
+            let mut svc = crate::GrpcWebLayer::new().layer(Svc);
+            let res = svc.call(request()).await.unwrap();
+
+            assert_eq!(res.status(), StatusCode::OK);
+        }
+
         #[tokio::test]
         async fn without_origin() {
             let mut svc = crate::enable(Svc);

diff --git a/tonic-web/tests/integration/tests/grpc_web.rs b/tonic-web/tests/integration/tests/grpc_web.rs
@@ -74,13 +74,14 @@ async fn spawn(allowed_origin: &str) -> String {
     let url = format!("http://{}", listener.local_addr().unwrap());
     let listener_stream = TcpListenerStream::new(listener);
 
-    let tonic_web_config = tonic_web::config().allow_origins(vec![allowed_origin]);
+    let svc = tonic_web::config()
+        .allow_origins(vec![allowed_origin])
+        .enable(TestServer::new(Svc));
 
     let _ = tokio::spawn(async move {
         Server::builder()
             .accept_http1(true)
-            .layer(tonic_web_config)
-            .add_service(TestServer::new(Svc))
+            .add_service(svc)
             .serve_with_incoming(listener_stream)
             .await
             .unwrap()