ok now for real fixed

1 files changed, 9 insertions, 4 deletions

diff --git a/internal/ior.go b/internal/ior.go
index 5e78645..51ce2d1 100644
--- a/internal/ior.go
+++ b/internal/ior.go
@@ -82,17 +82,20 @@ func Run(flags flags.Flags) {
 	}
 
 	loop := newEventLoop(flags)
-
-	startTime := time.Now()
 	duration := time.Duration(flags.Duration) * time.Second
 	fmt.Println("Probing for", duration)
 	ctx, cancel := context.WithTimeout(context.Background(), duration)
+
 	c := make(chan os.Signal, 1)
 	signal.Notify(c, os.Interrupt, syscall.SIGTERM)
 
 	go func() {
-		defer cancel()
 		<-c
+		cancel()
+	}()
+
+	go func() {
+		<-ctx.Done()
 		fmt.Println(loop.stats())
 		if flags.PprofEnable {
 			fmt.Println("Stoppig profiling, writing ior.cpuprofile and ior.memprofile")
@@ -102,7 +105,9 @@ func Run(flags flags.Flags) {
 		}
 	}()
 
+	startTime := time.Now()
 	loop.run(ctx, ch)
+	totalDuration := time.Since(startTime)
 	<-pprofDone
-	fmt.Println("Good bye... (unloading BPF tracepoints will take a few seconds...) after", time.Since(startTime))
+	fmt.Println("Good bye... (unloading BPF tracepoints will take a few seconds...) after", totalDuration)
 }