feat: add duration configurations

kaytu-io · Jul 11, 2024 · 4b29ed9 · 4b29ed9
1 parent 38d6ce1
commit 4b29ed9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pkg/scheduler/service.go b/pkg/scheduler/service.go
@@ -163,9 +163,9 @@ func (s *Service) runOptimizationJob(ctx context.Context, job *database.Optimiza
 		return
 	}
 
-	ctx, cancel := context.WithTimeout(ctx, s.cfg.GetOptimizationJobRunTimeout())
+	jobCtx, cancel := context.WithTimeout(ctx, s.cfg.GetOptimizationJobRunTimeout())
 	defer cancel()
-	err = s.kaytuCmd.Optimize(ctx, job.Command)
+	err = s.kaytuCmd.Optimize(jobCtx, job.Command)
 	if err != nil {
 		s.logger.Error("failed to run kaytu optimization", zap.String("command", job.Command), zap.Error(err))
 		jobStatus = database.OptimizationJobStatusFailed