feat(携程池[routinePool]): 携程池错误处理

Ccheers · Jul 1, 2022 · 365505a · 365505a
1 parent c1631f3
commit 365505a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/sync/routinepool/pool.go b/sync/routinepool/pool.go
@@ -2,6 +2,7 @@ package routinepool
 
 import (
 	"context"
+	"log"
 	"sync"
 	"sync/atomic"
 )
@@ -83,6 +84,9 @@ func NewPool(name string, cap int32, config *Config) Pool {
 		cap:    cap,
 		config: config,
 	}
+	p.SetPanicHandler(func(ctx context.Context, err error) {
+		log.Println(err.Error())
+	})
 	return p
 }
 

diff --git a/sync/routinepool/worker.go b/sync/routinepool/worker.go
@@ -44,7 +44,9 @@ func (w *worker) run() {
 			// check context before doing task
 			select {
 			case <-t.ctx.Done():
-				w.pool.panicHandler(t.ctx, fmt.Errorf("[routinepool] task cancel: %s error: %w", w.pool.name, t.ctx.Err()))
+				if w.pool.panicHandler != nil {
+					w.pool.panicHandler(t.ctx, fmt.Errorf("[routinepool] task cancel: %s error: %w", w.pool.name, t.ctx.Err()))
+				}
 				t.Recycle()
 				continue
 			default: