feat: MAMORU-813:Collect Daemons statistics

added statistics
Mamoru-Foundation · Aug 24, 2023 · 314d8cb · 314d8cb
1 parent c8a7f2a
commit 314d8cb
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/crates/mamoru-sui-sniffer/src/lib.rs b/crates/mamoru-sui-sniffer/src/lib.rs
@@ -69,7 +69,7 @@ impl SuiSniffer {
 
             let before_ms = Utc::now().timestamp_millis();
 
-            register_events(ctx_builder.data_mut(), seq, events);
+            register_events(ctx_builder.data_mut(), seq, events.clone());
 
             let after_ms = Utc::now().timestamp_millis();
 
@@ -80,7 +80,7 @@ impl SuiSniffer {
 
             let before_ms = Utc::now().timestamp_millis();
 
-            register_call_traces(ctx_builder.data_mut(), seq, call_traces);
+            register_call_traces(ctx_builder.data_mut(), seq, call_traces.clone());
 
             let after_ms = Utc::now().timestamp_millis();
 
@@ -89,6 +89,15 @@ impl SuiSniffer {
                 after_ms - before_ms,
             );
 
+            // Set statistics
+            ctx_builder.set_statistics(0, 1, events.len() as u64, call_traces.len() as u64);
+
+            info!(
+                "sniffer collected {} events and {} call traces.",
+                events.len(),
+                call_traces.len(),
+            );
+
             let ctx = ctx_builder.build()?;
 
             Result::<_, SuiSnifferError>::Ok(ctx)