Add select wrapper (#158)

Better handling of job delay
---------
Co-authored-by: samyfodil <[email protected]>

Author: mortum5

protocols/monkey/jobs/run.go

@@ -2,13 +2,15 @@ package jobs
 
 import (
 	"context"
+	"errors"
 	"time"
 
 	"github.com/ipfs/go-log/v2"
 )
 
 var (
-	logger = log.Logger("tau.monkey.jobs.client")
+	logger               = log.Logger("tau.monkey.jobs.client")
+	ErrorContextCanceled = errors.New("context cancel")
 )
 
 func (c *Context) Run(ctx context.Context, ctxC context.CancelFunc) (err error) {
@@ -17,7 +19,11 @@ func (c *Context) Run(ctx context.Context, ctxC context.CancelFunc) (err error)
 	defer ctxC()
 
 	if c.Job.Delay != nil {
-		<-time.After(time.Duration(c.Job.Delay.Time) * time.Second)
+		select {
+		case <-time.After(time.Duration(c.Job.Delay.Time) * time.Second):
+		case <-c.ctx.Done():
+			return ErrorContextCanceled
+		}
 	}
 
 	if err = c.cloneAndSet(); err != nil {

protocols/monkey/jobs/run_test.go

@@ -0,0 +1,24 @@
+package jobs
+
+import (
+	"context"
+	"testing"
+	"time"
+
+	"github.com/taubyte/go-interfaces/services/patrick"
+	"gotest.tools/v3/assert"
+)
+
+func TestRunDelay(t *testing.T) {
+	c := &Context{
+		Job: &patrick.Job{
+			Delay: &patrick.DelayConfig{
+				Time: 300,
+			},
+		},
+	}
+
+	ctx, ctxC := context.WithTimeout(context.Background(), 1*time.Second)
+	err := c.Run(ctx, ctxC)
+	assert.Equal(t, err, ErrorContextCanceled)
+}