flamegraph: replace worker busy-wait with blocking select

1 files changed, 6 insertions, 10 deletions

diff --git a/internal/flamegraph/worker.go b/internal/flamegraph/worker.go
index 534bfc3..0f49568 100644
--- a/internal/flamegraph/worker.go
+++ b/internal/flamegraph/worker.go
@@ -4,7 +4,6 @@ import (
 	"context"
 	"ior/internal/event"
 	"sync"
-	"time"
 )
 
 type worker struct {
@@ -21,17 +20,14 @@ func (w worker) run(ctx context.Context, wg *sync.WaitGroup, ch <-chan *event.Pa
 
 	for {
 		select {
-		case ev := <-ch:
-			w.iod.addEventPair(ev)
-			ev.Recycle()
-
-		default:
-			select {
-			case <-ctx.Done():
+		case ev, ok := <-ch:
+			if !ok {
 				return
-			default:
-				time.Sleep(time.Millisecond * 10)
 			}
+			w.iod.addEventPair(ev)
+			ev.Recycle()
+		case <-ctx.Done():
+			return
 		}
 	}
 }