feat: print reponse

src/app.rs

@@ -1,8 +1,10 @@
 use async_trait::async_trait;
 use base64::Engine;
 use base64::engine::general_purpose::STANDARD;
+use bytes::Bytes;
 use http::HeaderName;
 use pingora::{Error, ErrorType};
+use pingora::http::ResponseHeader;
 use pingora::prelude::{HttpPeer, ProxyHttp, Result, Session};
 
 use super::service::HostConfig;
@@ -83,4 +85,34 @@ impl ProxyHttp for ProxyApp {
         log::info!("Request:\n{}\n\n{}", req, body.unwrap_or_else(|| "<None>".into()));
         Ok(false)
     }
+
+    async fn response_filter(
+        &self,
+        _session: &mut Session,
+        upstream_response: &mut ResponseHeader,
+        _ctx: &mut Self::CTX,
+    ) -> Result<()>
+        where Self::CTX: Send + Sync,
+    {
+        let mut resp = String::new();
+        resp.push_str(&format!("version: {}\n", upstream_response.status));
+        resp.push_str(&format!("headers: {:#?}", upstream_response.headers));
+
+        log::info!("Response: {}", resp);
+        Ok(())
+    }
+
+    fn upstream_response_body_filter(
+        &self,
+        _session: &mut Session,
+        body: &Option<Bytes>,
+        end_of_stream: bool,
+        _ctx: &mut Self::CTX,
+    ) {
+        log::info!("Body {}: [[[{}]]] ",end_of_stream,
+            body.as_ref().map(|bytes|
+                String::from_utf8_lossy(bytes.iter().as_slice())
+            ).unwrap_or_else(|| "<none>".into())
+        );
+    }
 }