feat:add DoBefore

workpool/workpool.go

@@ -39,6 +39,21 @@ func (p *WorkPool) Do(fn TaskHandler) { // 添加到工作池，并立即返回
 	// p.task <- fn
 }
 
+// Do Add the task to the workpool and return immediately with a deadline, if the deadline is exceeded, the task will be aborted
+func (p *WorkPool) DoBefore(fn TaskHandler, time time.Duration) {
+	ctx, cancel := context.WithTimeout(context.Background(), time)
+	p.Do(func() error {
+		select {
+		case <-ctx.Done():
+			cancel()
+			return ctx.Err()
+		default:
+			return fn()
+		}
+	},
+	)
+}
+
 // DoWait Add to the workpool and wait for execution to complete before returning
 func (p *WorkPool) DoWait(task TaskHandler) { // 添加到工作池，并等待执行完成之后再返回
 	if p.IsClosed() { // closed

workpool/workpool_test.go

@@ -2,9 +2,11 @@ package workpool
 
 import (
 	"fmt"
+	"sync/atomic"
 	"testing"
 	"time"
 
+	"github.com/stretchr/testify/assert"
 	"github.com/xxjwxc/public/errors"
 )
 
@@ -104,3 +106,23 @@ func TestWorkerPoolIsDone(t *testing.T) {
 	fmt.Println(wp.IsDone())
 	fmt.Println("down")
 }
+
+func TestDoBefore(t *testing.T) {
+	wp := New(1) // Set the maximum number of threads
+
+	var res int32 = 0
+
+	wp.Do(func() error {
+		time.Sleep(5 * time.Second)
+		atomic.AddInt32(&res, 1)
+		return nil
+	})
+
+	wp.DoBefore(func() error {
+		atomic.AddInt32(&res, 1)
+		return nil
+	}, 5*time.Second)
+
+	wp.Wait()
+	assert.Equal(t, int32(1), res)
+}