update examples with context

examples/complex-http-crawler/pkg/model/task.go

@@ -1,6 +1,7 @@
 package model
 
 import (
+	"context"
 	"net/http"
 	"time"
 )
@@ -18,7 +19,7 @@ func New(url string) *MyTask {
 	}
 }
 
-func (t *MyTask) Do() error {
+func (t *MyTask) Do(_ context.Context) error {
 	transport := &http.Transport{
 		DisableCompression: true,
 	}

examples/metadata/main.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"context"
 	"math/rand"
 	"time"
 
@@ -15,7 +16,7 @@ func New() *MyTask {
 	return &MyTask{}
 }
 
-func (t *MyTask) Do() error {
+func (t *MyTask) Do(_ context.Context) error {
 	time.Sleep(time.Duration(rand.Intn(100)) * time.Millisecond)
 	return nil
 }

examples/nopper/main.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"context"
 	"math/rand"
 	"time"
 
@@ -14,7 +15,7 @@ func New() *MyTask {
 	return &MyTask{}
 }
 
-func (t *MyTask) Do() error {
+func (t *MyTask) Do(_ context.Context) error {
 	time.Sleep(time.Duration(rand.Intn(1000)) * time.Millisecond)
 	return nil
 }

examples/prometheus/main.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"context"
 	"fmt"
 	"net/http"
 
@@ -18,7 +19,7 @@ func New(url string) *MyTask {
 	}
 }
 
-func (t *MyTask) Do() error {
+func (t *MyTask) Do(_ context.Context) error {
 	response, err := http.Get(t.Url)
 	if err != nil {
 		return err

examples/random-error/main.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"context"
 	"errors"
 	"math/rand"
 	"time"
@@ -22,7 +23,7 @@ func New(index int, sleepSeconds int) *MyTask {
 	}
 }
 
-func (t *MyTask) Do() error {
+func (t *MyTask) Do(_ context.Context) error {
 	time.Sleep(time.Duration(t.SleepSeconds) * time.Second)
 	if rand.Float64() < t.ErrorProbability {
 		return errors.New("an error occurred")

examples/result-channel/main.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"context"
 	"encoding/json"
 	"fmt"
 	"log/slog"
@@ -20,7 +21,7 @@ func New(url string) *MyTask {
 	}
 }
 
-func (t *MyTask) Do() error {
+func (t *MyTask) Do(_ context.Context) error {
 	response, err := http.Get(t.Url)
 	if err != nil {
 		return err

examples/simple-http-crawler/main.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"context"
 	"fmt"
 	"net/http"
 
@@ -18,7 +19,7 @@ func New(url string) *MyTask {
 	}
 }
 
-func (t *MyTask) Do() error {
+func (t *MyTask) Do(_ context.Context) error {
 	response, err := http.Get(t.Url)
 	if err != nil {
 		return err

examples/sleeper/main.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"context"
 	"math/rand"
 	"time"
 
@@ -19,7 +20,7 @@ func New(index int, sleepSeconds int) *MyTask {
 	}
 }
 
-func (t *MyTask) Do() error {
+func (t *MyTask) Do(_ context.Context) error {
 	time.Sleep(time.Duration(t.SleepSeconds) * time.Second)
 	return nil
 }

examples/tcp-port-scanner/task.go

@@ -1,6 +1,9 @@
 package main
 
-import "net"
+import (
+	"context"
+	"net"
+)
 
 type MyTask struct {
 	IP     string `json:"ip"`
@@ -16,7 +19,7 @@ func New(ip string, port uint16) *MyTask {
 	}
 }
 
-func (t *MyTask) Do() error {
+func (t *MyTask) Do(_ context.Context) error {
 	conn, err := net.DialTCP("tcp", nil, &net.TCPAddr{
 		IP:   net.ParseIP(t.IP),
 		Port: int(t.Port),

pkg/utils/timeout.go

@@ -5,7 +5,7 @@ import (
 	"time"
 )
 
-func RunWithTimeout(f func() error, timeout time.Duration) error {
+func RunWithTimeout(f func(ctx context.Context) error, timeout time.Duration) error {
 	ctx, cancel := context.WithTimeout(context.Background(), timeout)
 	defer cancel()
 
@@ -14,7 +14,7 @@ func RunWithTimeout(f func() error, timeout time.Duration) error {
 	go func() {
 		defer close(done)
 
-		done <- f()
+		done <- f(ctx)
 	}()
 
 	select {

pkg/utils/timeout_test.go

@@ -1,14 +1,15 @@
 package utils_test
 
 import (
+	"context"
 	"testing"
 	"time"
 
 	"github.com/WangYihang/gojob/pkg/utils"
 )
 
 func TestRunWithTimeout(t *testing.T) {
-	task := func() error {
+	task := func(_ context.Context) error {
 		time.Sleep(2 * time.Second)
 		return nil
 	}

scheduler_test.go

@@ -1,6 +1,7 @@
 package gojob_test
 
 import (
+	"context"
 	"fmt"
 	"reflect"
 	"sort"
@@ -46,7 +47,7 @@ func newTask(i int, writer *safeWriter) *schedulerTestTask {
 	}
 }
 
-func (t *schedulerTestTask) Do() error {
+func (t *schedulerTestTask) Do(_ context.Context) error {
 	t.writer.WriteString(fmt.Sprintf("%d\n", t.I))
 	return nil
 }

task.go

@@ -1,6 +1,8 @@
 package gojob
 
 import (
+	"context"
+
 	"github.com/google/uuid"
 )
 
@@ -9,7 +11,7 @@ type Task interface {
 	// Do starts the task, returns error if failed
 	// If an error is returned, the task will be retried until MaxRetries
 	// You can set MaxRetries by calling SetMaxRetries on the scheduler
-	Do() error
+	Do(context.Context) error
 }
 
 type basicTask struct {