From bb141abe22671474f4912075e2f969c249684289 Mon Sep 17 00:00:00 2001
From: "cong.xie" <cong.xie@datadoghq.com>
Date: Mon, 9 Feb 2026 15:38:35 -0500
Subject: [PATCH] feat(aggregation): add keys() accessor to
 IntermediateAggregationResults

---
 src/aggregation/intermediate_agg_result.rs | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/aggregation/intermediate_agg_result.rs b/src/aggregation/intermediate_agg_result.rs
index 5a3cddd1c..bdc7a0cb0 100644
--- a/src/aggregation/intermediate_agg_result.rs
+++ b/src/aggregation/intermediate_agg_result.rs
@@ -128,6 +128,11 @@ impl IntermediateAggregationResults {
         self.aggs_res.remove(key)
     }
 
+    /// Returns an iterator over the keys in the intermediate aggregation results.
+    pub fn keys(&self) -> impl Iterator<Item = &String> {
+        self.aggs_res.keys()
+    }
+
     /// Add a result
     pub fn push(&mut self, key: String, value: IntermediateAggregationResult) -> crate::Result<()> {
         let entry = self.aggs_res.entry(key);