Merge pull request #170 from actix/read_body-example

Add middleware example for reading the Request body

middleware/Cargo.toml

@@ -10,3 +10,4 @@ actix-service = "0.4.1"
 actix-web = "1.0.0"
 futures = "0.1.25"
 env_logger = "0.6"
+bytes = "0.4"

middleware/README.md

@@ -8,4 +8,3 @@ cargo run
 
 Look in `src/main.rs` and comment the different middlewares in/out to see how
 they function.
-

middleware/src/main.rs

@@ -5,6 +5,8 @@ use futures::future::Future;
 #[allow(dead_code)]
 mod redirect;
 #[allow(dead_code)]
+mod read_body;
+#[allow(dead_code)]
 mod simple;
 
 fn main() -> std::io::Result<()> {
@@ -14,6 +16,7 @@ fn main() -> std::io::Result<()> {
     HttpServer::new(|| {
         App::new()
             .wrap(redirect::CheckLogin)
+            .wrap(read_body::Logging)
             .wrap(simple::SayHi)
             .wrap_fn(|req, srv| {
                 println!("Hi from start. You requested: {}", req.path());

middleware/src/read_body.rs

@@ -0,0 +1,70 @@
+use actix_service::{Service, Transform};
+use actix_web::error::PayloadError;
+use actix_web::{dev::ServiceRequest, dev::ServiceResponse, Error, HttpMessage};
+use bytes::BytesMut;
+use futures::future::{ok, FutureResult};
+use futures::stream::Stream;
+use futures::{Future, Poll};
+use std::cell::RefCell;
+use std::rc::Rc;
+
+pub struct Logging;
+
+impl<S: 'static, B> Transform<S> for Logging
+where
+    S: Service<Request = ServiceRequest, Response = ServiceResponse<B>, Error = Error>,
+    S::Future: 'static,
+    B: 'static,
+{
+    type Request = ServiceRequest;
+    type Response = ServiceResponse<B>;
+    type Error = Error;
+    type InitError = ();
+    type Transform = LoggingMiddleware<S>;
+    type Future = FutureResult<Self::Transform, Self::InitError>;
+
+    fn new_transform(&self, service: S) -> Self::Future {
+        ok(LoggingMiddleware {
+            service: Rc::new(RefCell::new(service)),
+        })
+    }
+}
+
+pub struct LoggingMiddleware<S> {
+    // This is special: We need this to avoid lifetime issues.
+    service: Rc<RefCell<S>>,
+}
+
+impl<S, B> Service for LoggingMiddleware<S>
+where
+    S: Service<Request = ServiceRequest, Response = ServiceResponse<B>, Error = Error>
+        + 'static,
+    S::Future: 'static,
+    B: 'static,
+{
+    type Request = ServiceRequest;
+    type Response = ServiceResponse<B>;
+    type Error = Error;
+    type Future = Box<dyn Future<Item = Self::Response, Error = Self::Error>>;
+
+    fn poll_ready(&mut self) -> Poll<(), Self::Error> {
+        self.service.poll_ready()
+    }
+
+    fn call(&mut self, mut req: ServiceRequest) -> Self::Future {
+        let mut svc = self.service.clone();
+
+        Box::new(
+            req.take_payload()
+                .fold(BytesMut::new(), move |mut body, chunk| {
+                    body.extend_from_slice(&chunk);
+                    Ok::<_, PayloadError>(body)
+                })
+                .map_err(|e| e.into())
+                .and_then(move |bytes| {
+                    println!("request body: {:?}", bytes);
+                    svc.call(req).and_then(|res| Ok(res))
+                }),
+        )
+    }
+}