fix(state-machine): remove claims from license state log (#6795)

Co-authored-by: Renée <renee.kooi@apollographql.com>
apollographql · Feb 18, 2025 · 550d5d7 · 550d5d7
1 parent e8b7a06
commit 550d5d7
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/apollo-router/src/state_machine.rs b/apollo-router/src/state_machine.rs
@@ -525,7 +525,13 @@ where
 
         // Process all the events in turn until we get to error state or we run out of events.
         while let Some(event) = messages.next().await {
-            let event_name = format!("{event:?}");
+            let event_name = match &event {
+                Event::UpdateLicense(license_state) => {
+                    format!("UpdateLicense({})", license_state.get_name())
+                }
+                event => format!("{event:?}"),
+            };
+
             let previous_state = format!("{state:?}");
 
             state = match event {

diff --git a/apollo-router/src/uplink/license_enforcement.rs b/apollo-router/src/uplink/license_enforcement.rs
@@ -659,6 +659,15 @@ impl LicenseState {
             _ => None,
         }
     }
+
+    pub(crate) fn get_name(&self) -> &'static str {
+        match self {
+            Self::Licensed { limits: _ } => "Licensed",
+            Self::LicensedWarn { limits: _ } => "LicensedWarn",
+            Self::LicensedHalt { limits: _ } => "LicensedHalt",
+            Self::Unlicensed => "Unlicensed",
+        }
+    }
 }
 
 impl Display for License {