From 9426d5be7bf2d778c6552430656583293dbb640e Mon Sep 17 00:00:00 2001
From: trinity Pointard <trinity.pointard@datadoghq.com>
Date: Fri, 14 Mar 2025 14:39:43 +0100
Subject: [PATCH] fix agg Key PartialEq impl

---
 src/aggregation/mod.rs | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/aggregation/mod.rs b/src/aggregation/mod.rs
index 4bfe9a6d2..173ada3be 100644
--- a/src/aggregation/mod.rs
+++ b/src/aggregation/mod.rs
@@ -366,8 +366,12 @@ impl PartialEq for Key {
     fn eq(&self, other: &Self) -> bool {
         match (self, other) {
             (Self::Str(l), Self::Str(r)) => l == r,
-            (Self::F64(l), Self::F64(r)) => l == r,
-            _ => false,
+            (Self::F64(l), Self::F64(r)) => l.to_bits() == r.to_bits(),
+            (Self::I64(l), Self::I64(r)) => l == r,
+            (Self::U64(l), Self::U64(r)) => l == r,
+            // we list all variant of left operand to make sure this gets updated when we add
+            // variants to the enum
+            (Self::Str(_) | Self::F64(_) | Self::I64(_) | Self::U64(_), _) => false,
         }
     }
 }