fix(task-17): prevent aggregate double-count and flush on shutdown

1 files changed, 10 insertions, 0 deletions

diff --git a/internal/syscall_aggregate_consumer.go b/internal/syscall_aggregate_consumer.go
index 108bbeb..dadd38e 100644
--- a/internal/syscall_aggregate_consumer.go
+++ b/internal/syscall_aggregate_consumer.go
@@ -127,3 +127,13 @@ func buildSyscallSamplingRates(cfg flags.Config) map[types.TraceId]uint32 {
 	}
 	return rates
 }
+
+func buildAggregateOnlyTraceIDs(cfg flags.Config) map[types.TraceId]struct{} {
+	ids := make(map[types.TraceId]struct{})
+	for traceID, rate := range buildSyscallSamplingRates(cfg) {
+		if rate == 0 {
+			ids[traceID] = struct{}{}
+		}
+	}
+	return ids
+}