fix: populate timeout context to node-collector (#120)

Signed-off-by: chenk <[email protected]>

pkg/jobs/runner.go

@@ -78,11 +78,10 @@ func (r *runner) runAndWaitForever() error {
 }
 
 func (r *runner) runWithTimeout(ctx context.Context) error {
-	ctxWithTimeout := ctx
 	// context timeout also can be set on caller side
 	if _, ok := ctx.Deadline(); !ok {
 		var cancel context.CancelFunc
-		ctxWithTimeout, cancel = context.WithTimeout(ctx, r.timeoutDuration)
+		ctx, cancel = context.WithTimeout(ctx, r.timeoutDuration)
 		defer cancel()
 	}
 	klog.V(3).Infof("Running task with timeout: %v", r.timeoutDuration)
@@ -92,7 +91,7 @@ func (r *runner) runWithTimeout(ctx context.Context) error {
 		klog.V(3).Infof("Stopping runner on task completion with error: %v", err)
 		return err
 	// Signaled when we run out of time.
-	case <-ctxWithTimeout.Done():
+	case <-ctx.Done():
 		klog.V(3).Info("Stopping runner on timeout")
 		return ErrTimeout
 	}