opt: set up the PanicHandler and Logger for pkg/pool/goroutine

panjf2000 · May 18, 2023 · 222a395 · 222a395
1 parent 453e26b
commit 222a395
Showing 1 changed file with 19 additions and 1 deletion.
diff --git a/pkg/pool/goroutine/goroutine.go b/pkg/pool/goroutine/goroutine.go
@@ -18,6 +18,8 @@ import (
 	"time"
 
 	"github.com/panjf2000/ants/v2"
+
+	"github.com/panjf2000/gnet/v2/pkg/logging"
 )
 
 const (
@@ -40,9 +42,25 @@ func init() {
 // Pool is the alias of ants.Pool.
 type Pool = ants.Pool
 
+type antsLogger struct {
+	logging.Logger
+}
+
+// Printf implements the ants.Logger interface.
+func (l antsLogger) Printf(format string, args ...interface{}) {
+	l.Infof(format, args...)
+}
+
 // Default instantiates a non-blocking *WorkerPool with the capacity of DefaultAntsPoolSize.
 func Default() *Pool {
-	options := ants.Options{ExpiryDuration: ExpiryDuration, Nonblocking: Nonblocking}
+	options := ants.Options{
+		ExpiryDuration: ExpiryDuration,
+		Nonblocking:    Nonblocking,
+		Logger:         &antsLogger{logging.GetDefaultLogger()},
+		PanicHandler: func(i interface{}) {
+			logging.Errorf("goroutine pool panic: %v", i)
+		},
+	}
 	defaultAntsPool, _ := ants.NewPool(DefaultAntsPoolSize, ants.WithOptions(options))
 	return defaultAntsPool
 }