Merge pull request #600 from viccuad/feat-request-origin

feat: Add new `request_origin` to metrics baggage

Author: viccuad

src/communication.rs

@@ -1,4 +1,5 @@
 use anyhow::Result;
+use core::fmt;
 use policy_evaluator::admission_response::AdmissionResponse;
 use policy_evaluator::policy_evaluator::ValidateRequest;
 use std::collections::HashMap;
@@ -13,6 +14,15 @@ pub(crate) enum RequestOrigin {
     Audit,
 }
 
+impl fmt::Display for RequestOrigin {
+    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
+        match self {
+            RequestOrigin::Validate => write!(f, "validate"),
+            RequestOrigin::Audit => write!(f, "audit"),
+        }
+    }
+}
+
 #[derive(Debug)]
 pub(crate) struct EvalRequest {
     pub policy_id: String,

src/metrics/mod.rs

@@ -32,6 +32,7 @@ pub struct PolicyEvaluation {
     pub(crate) resource_request_operation: String,
     pub(crate) accepted: bool,
     pub(crate) mutated: bool,
+    pub(crate) request_origin: String,
     pub(crate) error_code: Option<u16>,
 }
 
@@ -50,6 +51,7 @@ impl Into<Vec<KeyValue>> for &PolicyEvaluation {
             ),
             KeyValue::new("accepted", self.accepted),
             KeyValue::new("mutated", self.mutated),
+            KeyValue::new("request_origin", self.request_origin.clone()),
         ];
         if let Some(resource_namespace) = &self.resource_namespace {
             baggage.append(&mut vec![KeyValue::new(

src/worker.rs

@@ -227,6 +227,7 @@ impl Worker {
                     resource_request_operation: adm_req.clone().operation,
                     accepted,
                     mutated,
+                    request_origin: req.request_origin.to_string(),
                     error_code,
                 };
                 metrics::record_policy_latency(policy_evaluation_duration, &policy_evaluation);