Update taskset.go

lenonqing · Jun 30, 2021 · d6e6ad2 · d6e6ad2
1 parent 83e2ca5
commit d6e6ad2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/taskset.go b/taskset.go
@@ -1,6 +1,7 @@
 package boomer
 
 import (
+	"context"
 	"math/rand"
 	"sync"
 	"time"
@@ -103,5 +104,5 @@ func (ts *WeighingTaskSet) Run() {
 	r := rand.New(rand.NewSource(time.Now().UnixNano()))
 	roll := r.Intn(ts.offset)
 	task := ts.GetTask(roll)
-	task.Fn()
+	task.Fn(context.Background())
 }