From 9031d895f4e97945ebd090587ba1b1fbec3c045e Mon Sep 17 00:00:00 2001
From: rfyiamcool <rfyiamcool@163.com>
Date: Sun, 8 Oct 2023 14:28:19 +0800
Subject: [PATCH] feat: add taskcount() for gopool

Signed-off-by: rfyiamcool <rfyiamcool@163.com>
---
 util/gopool/pool.go | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/util/gopool/pool.go b/util/gopool/pool.go
index 276cc3f5..3fdcdd1f 100644
--- 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)
 }