Feature ctxtask

pool.go

@@ -270,6 +270,12 @@ type contextTask struct {
 	task Task
 }
 
+// UpdateTask wraps task with ctx so the task can observe cancellation and
+// deadlines while it is waiting in or executing from the pool.
+func UpdateTask(ctx context.Context, task Task) *contextTask {
+	return &contextTask{ctx: ctx, task: task}
+}
+
 func (t *contextTask) Process() {
 	if t.ctx.Err() != nil {
 		return

test/count.go

@@ -6,7 +6,7 @@ import (
 	"runtime"
 	"time"
 
-	agilepool "github.com/Yiming1997/agilePool"
+	agilepool "github.com/Yiming1997/agilePool/v2"
 	"github.com/shirou/gopsutil/v3/cpu"
 )
 

test/flag.go

@@ -13,7 +13,7 @@ import (
 	"sync/atomic"
 	"time"
 
-	agilepool "github.com/Yiming1997/agilePool"
+	agilepool "github.com/Yiming1997/agilePool/v2"
 )
 
 // TaskType controls how simulated task durations are generated.

test/go.mod

@@ -3,7 +3,7 @@ module agilepool-test
 go 1.23.4
 
 require (
-	github.com/Yiming1997/agilePool v0.0.0
+	github.com/Yiming1997/agilePool/v2 v2.0.0
 	github.com/shirou/gopsutil/v3 v3.24.5
 )
 
@@ -18,4 +18,4 @@ require (
 	golang.org/x/sys v0.20.0 // indirect
 )
 
-replace github.com/Yiming1997/agilePool => ../
+replace github.com/Yiming1997/agilePool/v2 => ../

test/main.go

@@ -8,7 +8,7 @@ import (
 	"runtime/pprof"
 	"time"
 
-	agilepool "github.com/Yiming1997/agilePool"
+	agilepool "github.com/Yiming1997/agilePool/v2"
 )
 
 func main() {