Feat: added Shutdown method to logger

log/logger.go

@@ -104,10 +104,11 @@ type Logger interface {
 	// ParentLogger returns the parent of this Logger (or nil if it is the root).
 	ParentLogger() Logger
 
-	// UnsubscribeFromParentLogger unsubscribes the logger from its parent logger (e.g. updates about the log level).
-	// It is important to call this method whenever we remove all references to the logger, otherwise the logger will
-	// not be garbage collected.
-	UnsubscribeFromParentLogger()
+	// Shutdown shuts down the logger by either unsubscribing from its parent logger or shutting down the root logger.
+	//
+	// Note: It is important to call this method whenever we remove all references to a child logger, otherwise the
+	// logger will not be garbage collected until the root logger is garbage collected.
+	Shutdown()
 }
 
 // NewLogger creates a new logger with the given options.

log/logger_impl.go

@@ -295,9 +295,18 @@ func (l *logger) ParentLogger() Logger {
 // UnsubscribeFromParentLogger unsubscribes the logger from its parent logger (e.g. updates about the log level).
 // It is important to call this method whenever we remove all references to the logger, otherwise the logger will
 // not be garbage collected.
-func (l *logger) UnsubscribeFromParentLogger() {
-	if l != nil && l.unsubscribeFromParent != nil {
+func (l *logger) Shutdown() {
+	if l == nil {
+		return
+	}
+
+	switch isChildLogger := l.parentLogger != nil; isChildLogger {
+	case true:
 		l.unsubscribeFromParent()
+	case false:
+		if asyncTextHandler, isAsyncTextHandler := l.rootLogger.Handler().(*textHandler); isAsyncTextHandler {
+			asyncTextHandler.ioWorker.Shutdown().PendingTasksCounter.WaitIsZero()
+		}
 	}
 }
 

log/logger_test.go

@@ -17,12 +17,12 @@ func TestLogger(t *testing.T) {
 	logger.LogTrace("created chain")
 
 	networkLogger := logger.NewChildLogger("network")
-	defer networkLogger.UnsubscribeFromParentLogger()
+	defer networkLogger.Shutdown()
 	networkLogger.SetLogLevel(log.LevelInfo)
 	networkLogger.LogInfo("instantiated chain (invisible)", "id", 1)
 
 	chainLogger := logger.NewChildLogger("chain1")
-	defer chainLogger.UnsubscribeFromParentLogger()
+	defer chainLogger.Shutdown()
 	chainLogger.SetLogLevel(log.LevelDebug)
 	chainLogger.LogDebug("attested weight updated (visible)", "oldWeight", 7, "newWeight", 10)
 	logger.LogTrace("shutdown")

log/text_handler.go

@@ -19,7 +19,7 @@ func NewTextHandler(options *Options) slog.Handler {
 	t := &textHandler{
 		output:     options.Output,
 		timeFormat: options.TimeFormat,
-		ioWorker:   workerpool.New("log.TextHandler", workerpool.WithWorkerCount(1)),
+		ioWorker:   workerpool.New("log.TextHandler", workerpool.WithWorkerCount(1)).Start(),
 	}
 
 	formatString := "%s\t%-7s\t%s\t%s %s\n"