Merge pull request #453 from answerbook/mdeltito/LOG-19789

fix(user_trace): log http error responses as captured_data
mezmo · Apr 30, 2024 · ee3fac9 · ee3fac9
2 parents 60f375f + 6870759
commit ee3fac9
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 1 deletion.
diff --git a/src/mezmo/user_trace.rs b/src/mezmo/user_trace.rs
@@ -287,6 +287,10 @@ pub trait UserLoggingResponse {
     fn log_msg(&self) -> Option<Value> {
         None
     }
+
+    fn log_captured_data(&self) -> Option<Value> {
+        None
+    }
 }
 
 pub trait UserLoggingError {
@@ -330,7 +334,14 @@ where
             match &res {
                 Ok(response) => {
                     if let Some(msg) = response.log_msg() {
-                        user_log_error!(ctx, msg);
+                        match response.log_captured_data() {
+                            Some(captured_data) => {
+                                user_log_error!(ctx, msg, captured_data: captured_data);
+                            }
+                            None => {
+                                user_log_error!(ctx, msg);
+                            }
+                        }
                     }
                 }
                 Err(err) => {
@@ -748,6 +759,12 @@ mod tests {
         fn log_msg(&self) -> Option<Value> {
             Some("log_msg(): response".into())
         }
+
+        fn log_captured_data(&self) -> Option<Value> {
+            Some(Value::Object(btreemap! {
+                "response" => r#"{"error": "badness"}"#
+            }))
+        }
     }
 
     struct MockWrapperService {
@@ -828,5 +845,16 @@ mod tests {
 
         let msg = log_res.get(".message").unwrap().as_str().unwrap();
         assert_eq!(msg, "log_msg(): response");
+
+        let captured_data = log_res
+            .get(".meta.mezmo.captured_data")
+            .expect("captured data should exist");
+
+        assert_eq!(
+            captured_data,
+            &Value::Object(btreemap! {
+                "response" => r#"{"error": "badness"}"#
+            })
+        );
     }
 }
diff --git a/src/sinks/util/http.rs b/src/sinks/util/http.rs
@@ -799,6 +799,12 @@ impl UserLoggingResponse for HttpResponse {
             None
         }
     }
+
+    fn log_captured_data(&self) -> Option<Value> {
+        Some(Value::Object(btreemap! {
+            "response" => self.http_response.body().to_owned()
+        }))
+    }
 }
 
 impl UserLoggingError for crate::Error {