feat(xlog): add context support for shutdown timeout control

examples/xlog_example/main.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"context"
 	"os"
 	"os/signal"
 	"syscall"
@@ -89,9 +90,10 @@ func main() {
 		// Wait for interrupt signal
 		<-sigChan
 		xlog.Info("Shutting down application")
-
+		ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+		defer cancel()
 		// Perform shutdown tasks
-		if err := xlog.Shutdown(); err != nil {
+		if err := xlog.Shutdown(ctx); err != nil {
 			xlog.Error("Failed to shutdown logger", "error", err)
 		}
 

xlog/xlog.go

@@ -694,27 +694,42 @@ type ShutdownHandler interface {
 }
 
 // Shutdown shuts down all handlers that implement the ShutdownHandler interface.
-func Shutdown() error {
+// If ctx is nil, it will use context.Background().
+func Shutdown(ctx context.Context) error {
+	if ctx == nil {
+		ctx = context.Background()
+	}
+
 	mu.Lock()
 	defer mu.Unlock()
 
-	var errs []error
-	for _, handler := range handlers {
-		if closer, ok := handler.(ShutdownHandler); ok {
-			if err := closer.Shutdown(); err != nil {
-				errs = append(errs, fmt.Errorf("failed to shutdown handler: %w", err))
+	done := make(chan error, 1)
+	go func() {
+		var errs []error
+		for _, handler := range handlers {
+			if closer, ok := handler.(ShutdownHandler); ok {
+				if err := closer.Shutdown(); err != nil {
+					errs = append(errs, fmt.Errorf("failed to shutdown handler: %w", err))
+				}
 			}
-		}
-		if closer, ok := handler.(io.Closer); ok {
-			if err := closer.Close(); err != nil {
-				errs = append(errs, fmt.Errorf("failed to close handler: %w", err))
+			if closer, ok := handler.(io.Closer); ok {
+				if err := closer.Close(); err != nil {
+					errs = append(errs, fmt.Errorf("failed to close handler: %w", err))
+				}
 			}
 		}
+		if len(errs) > 0 {
+			done <- xerror.Errorf("failed to shutdown one or more handlers: %v", errs)
+		}
+		done <- nil
+	}()
+
+	select {
+	case err := <-done:
+		return err
+	case <-ctx.Done():
+		return xerror.Errorf("shutdown timeout: %v", ctx.Err())
 	}
-	if len(errs) > 0 {
-		return xerror.Errorf("failed to shutdown one or more handlers: %v", errs)
-	}
-	return nil
 }
 
 // SetLogger sets the default logger