feat: add taskcount() for gopool

Signed-off-by: rfyiamcool <[email protected]>
bytedance · Oct 8, 2023 · 9031d89 · 9031d89
1 parent 614d0af
commit 9031d89
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/util/gopool/pool.go b/util/gopool/pool.go
@@ -143,10 +143,16 @@ func (p *pool) SetPanicHandler(f func(context.Context, interface{})) {
 	p.panicHandler = f
 }
 
+// WorkerCount get worker count
 func (p *pool) WorkerCount() int32 {
 	return atomic.LoadInt32(&p.workerCount)
 }
 
+// TaskCount get task count
+func (p *pool) TaskCount() int32 {
+	return atomic.LoadInt32(&p.taskCount)
+}
+
 func (p *pool) incWorkerCount() {
 	atomic.AddInt32(&p.workerCount, 1)
 }