Merge pull request #31 from nextmv-io/feature/eng-5070-parallel-solve…

…r-cycle-invocation-do-not-comply-with-max

Feature/eng 5070 parallel solver cycle invocation do not comply with max

solve_solver.go

@@ -10,7 +10,6 @@ import (
 	"time"
 
 	"github.com/nextmv-io/nextroute/common"
-	"github.com/nextmv-io/sdk/run"
 )
 
 // IntParameterOptions are the options for an integer parameter.
@@ -247,7 +246,12 @@ func (s *solveImpl) Solve(
 	s.workSolution = newWorkSolution
 	s.random = rand.New(rand.NewSource(newWorkSolution.Random().Int63()))
 
-	start := ctx.Value(run.Start).(time.Time)
+	start := time.Now()
+
+	ctx, cancel := context.WithDeadline(
+		ctx,
+		start.Add(solveOptions.Duration),
+	)
 
 	solveInformation := &solveInformationImpl{
 		iteration:      0,
@@ -268,7 +272,10 @@ func (s *solveImpl) Solve(
 		Error:    nil,
 	}
 	go func() {
-		defer close(solutions)
+		defer func() {
+			close(solutions)
+			cancel()
+		}()
 
 	Loop:
 		for iteration := 0; iteration < solveOptions.Iterations; iteration++ {

solve_solver_parallel.go

@@ -278,7 +278,11 @@ func (s *parallelSolverImpl) Solve(
 		}
 	}
 
-	start := ctx.Value(run.Start).(time.Time)
+	start := time.Now()
+
+	if ctx.Value(run.Start) != nil {
+		start = ctx.Value(run.Start).(time.Time)
+	}
 
 	ctx, cancel := context.WithDeadline(
 		ctx,

tests/custom_operators/main.go

@@ -6,6 +6,7 @@ package main
 import (
 	"context"
 	"log"
+	"time"
 
 	"github.com/nextmv-io/nextroute"
 	"github.com/nextmv-io/nextroute/check"
@@ -76,6 +77,7 @@ func solver(
 		) (nextroute.SolveOptions, error) {
 			return nextroute.SolveOptions{
 				Iterations: 1000,
+				Duration:   10 * time.Minute,
 			}, nil
 		},
 	)