From 52cc7dc6ccdf2d97ea49d361299701cf6b76e898 Mon Sep 17 00:00:00 2001
From: Paul Buetow <paul@buetow.org>
Date: Sun, 1 Mar 2026 23:51:33 +0200
Subject: Use print callback for all event loop output modes

---
 internal/eventloop.go | 32 ++++++++++++++++++++++----------
 1 file changed, 22 insertions(+), 10 deletions(-)

(limited to 'internal/eventloop.go')

diff --git a/internal/eventloop.go b/internal/eventloop.go
index 536fe0a..a33ebbb 100644
--- a/internal/eventloop.go
+++ b/internal/eventloop.go
@@ -210,6 +210,7 @@ func newEventLoop(cfg eventLoopConfig) *eventLoop {
 	if cfg.liveFlamegraph {
 		el.liveTrie = flamegraph.NewLiveTrie(cfg.collapsedFields, cfg.countField)
 	}
+	el.configureOutputCallback()
 	el.seedTrackedPidComm()
 	return el
 }
@@ -238,6 +239,23 @@ func (e *eventLoop) commState() *commResolver {
 	return e.commResolver
 }
 
+func (e *eventLoop) configureOutputCallback() {
+	switch {
+	case e.cfg.flamegraphEnable:
+		e.printCb = func(ep *event.Pair) {
+			e.flamegraph.Ch <- ep
+		}
+	case e.liveTrie != nil:
+		e.printCb = func(ep *event.Pair) {
+			e.liveTrie.Ingest(ep)
+		}
+	case e.cfg.pprofEnable:
+		e.printCb = func(ep *event.Pair) {
+			ep.Recycle()
+		}
+	}
+}
+
 func (e *eventLoop) stats() string {
 	fmt.Println("Waiting for stats to be ready")
 	<-e.done
@@ -282,17 +300,11 @@ func (e *eventLoop) run(ctx context.Context, rawCh <-chan []byte) {
 	}
 
 	e.startTime = time.Now()
+	if e.printCb == nil {
+		e.printCb = func(ep *event.Pair) { ep.Recycle() }
+	}
 	for ep := range e.events(ctx, rawCh) {
-		switch {
-		case e.cfg.flamegraphEnable:
-			e.flamegraph.Ch <- ep
-		case e.liveTrie != nil:
-			e.liveTrie.Ingest(ep)
-		case e.cfg.pprofEnable:
-			ep.Recycle()
-		default:
-			e.printCb(ep)
-		}
+		e.printCb(ep)
 		e.numSyscallsAfterFilter++
 	}
 
-- 
cgit v1.2.3