Refactor hammer TUI log writer

Author: roger2hk

internal/hammer/hammer.go

@@ -189,7 +189,6 @@ func main() {
 
 	if *showUI {
 		c := loadtest.NewController(hammer, ha)
-		slog.SetDefault(slog.New(slog.NewTextHandler(c.LogWriter(), &slog.HandlerOptions{Level: slog.Level(*slogLevel)})))
 		c.Run(ctx)
 	} else {
 		<-ctx.Done()

internal/hammer/loadtest/tui.go

@@ -17,7 +17,6 @@ package loadtest
 import (
 	"context"
 	"fmt"
-	"io"
 	"strings"
 	"time"
 
@@ -69,6 +68,9 @@ func NewController(h *Hammer, a *HammerAnalyser) *tuiController {
 }
 
 func (c *tuiController) Run(ctx context.Context) {
+	// Redirect logs to the view.
+	slog.SetDefault(slog.New(slog.NewTextHandler(c.logView, &slog.HandlerOptions{Level: slog.LevelInfo})))
+
 	go c.updateStatsLoop(ctx, 500*time.Millisecond)
 
 	c.app.SetInputCapture(func(event *tcell.EventKey) *tcell.EventKey {
@@ -144,8 +146,3 @@ func (c *tuiController) updateStatsLoop(ctx context.Context, interval time.Durat
 		}
 	}
 }
-
-// LogWriter returns a writer that writes to the log TextView.
-func (c *tuiController) LogWriter() io.Writer {
-	return c.logView
-}

internal/mirror/hammer/hammer.go

@@ -230,7 +230,6 @@ func main() {
 
 	if *showUI {
 		c := loadtest.NewController(hammer, ha)
-		slog.SetDefault(slog.New(slog.NewTextHandler(c.LogWriter(), &slog.HandlerOptions{Level: slog.Level(*slogLevel)})))
 		c.Run(ctx)
 	} else {
 		<-ctx.Done()

internal/mirror/hammer/loadtest/tui.go

@@ -17,7 +17,6 @@ package loadtest
 import (
 	"context"
 	"fmt"
-	"io"
 	"strings"
 	"time"
 
@@ -69,6 +68,9 @@ func NewController(h *Hammer, a *HammerAnalyser) *tuiController {
 }
 
 func (c *tuiController) Run(ctx context.Context) {
+	// Redirect logs to the view.
+	slog.SetDefault(slog.New(slog.NewTextHandler(c.logView, &slog.HandlerOptions{Level: slog.LevelInfo})))
+
 	go c.updateStatsLoop(ctx, 500*time.Millisecond)
 
 	c.app.SetInputCapture(func(event *tcell.EventKey) *tcell.EventKey {
@@ -144,8 +146,3 @@ func (c *tuiController) updateStatsLoop(ctx context.Context, interval time.Durat
 		}
 	}
 }
-
-// LogWriter returns a writer that writes to the log TextView.
-func (c *tuiController) LogWriter() io.Writer {
-	return c.logView
-}