From 4b29ed98469ff3bd38116559f4090eff63693321 Mon Sep 17 00:00:00 2001 From: Mahan Zendedel DH Date: Thu, 11 Jul 2024 15:47:05 +0400 Subject: [PATCH] feat: add duration configurations --- pkg/scheduler/service.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pkg/scheduler/service.go b/pkg/scheduler/service.go index 292e4cd..cfa1074 100644 --- 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