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

1 files changed, 7 insertions, 6 deletions

diff --git a/internal/ior.go b/internal/ior.go
index 3afb09a..61be397 100644
--- a/internal/ior.go
+++ b/internal/ior.go
@@ -379,12 +379,13 @@ func newEventLoopConfig(cfg flags.Config) eventLoopConfig {
 	fields := make([]string, len(cfg.CollapsedFields))
 	copy(fields, cfg.CollapsedFields)
 	return eventLoopConfig{
-		pidFilter:       cfg.PidFilter,
-		filter:          traceFilterFromConfig(cfg),
-		collapsedFields: fields,
-		countField:      cfg.CountField,
-		pprofEnable:     cfg.PprofEnable,
-		plainMode:       cfg.PlainMode,
+		pidFilter:             cfg.PidFilter,
+		filter:                traceFilterFromConfig(cfg),
+		collapsedFields:       fields,
+		countField:            cfg.CountField,
+		pprofEnable:           cfg.PprofEnable,
+		plainMode:             cfg.PlainMode,
+		aggregateOnlyTraceIDs: buildAggregateOnlyTraceIDs(cfg),
 	}
 }