fix: Enable stats collection (#348)

getsentry · Mar 29, 2024 · 6f840d8 · 6f840d8
1 parent 4a2015c
commit 6f840d8
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/rust-arroyo/src/backends/kafka/config.rs b/rust-arroyo/src/backends/kafka/config.rs
@@ -3,6 +3,8 @@ use std::collections::HashMap;
 
 use super::InitialOffset;
 
+const STATS_COLLECTION_FREQ_MS: u32 = 1000;
+
 #[derive(Debug, Clone)]
 pub struct OffsetResetConfig {
     pub auto_offset_reset: InitialOffset,
@@ -54,6 +56,11 @@ impl KafkaConfig {
             max_poll_interval_ms.to_string(),
         );
 
+        config.config_map.insert(
+            "statistics.interval.ms".to_string(),
+            STATS_COLLECTION_FREQ_MS.to_string(),
+        );
+
         // HACK: If the max poll interval is less than 45 seconds, set the session timeout
         // to the same. (its default is 45 seconds and it must be <= to max.poll.interval.ms)
         if max_poll_interval_ms < 45_000 {