move call and clean up

beam-cloud · Mar 6, 2025 · dfddb44 · dfddb44
1 parent 000a522
commit dfddb44
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/pkg/scheduler/scheduler.go b/pkg/scheduler/scheduler.go
@@ -259,6 +259,9 @@ func (s *Scheduler) StartProcessingRequests() {
 			continue
 		}
 
+		// Add the container cost per ms to the request if a cost endpoint is provided
+		s.addContainerCostPerMs(request)
+
 		// Find a worker to schedule ContainerRequests on
 		worker, err := s.selectWorker(request)
 		if err != nil || worker == nil {
@@ -294,7 +297,6 @@ func (s *Scheduler) StartProcessingRequests() {
 					}
 				}
 
-				s.addContainerCostPerMs(request)
 				log.Error().Str("container_id", request.ContainerId).Err(err).Msg("unable to add worker")
 				s.addRequestToBacklog(request)
 			}()

diff --git a/pkg/worker/lifecycle.go b/pkg/worker/lifecycle.go
@@ -649,6 +649,8 @@ func (s *Worker) spawn(request *types.ContainerRequest, spec *specs.Spec, output
 	}
 
 	// Log metrics
+
+	log.Info().Str("container_id", request.ContainerId).Str("cost_per_ms", fmt.Sprintf("%f", request.CostPerMs)).Msg("container cost per ms")
 	go s.workerMetrics.EmitContainerUsage(ctx, request)
 	go s.eventRepo.PushContainerStartedEvent(containerId, s.workerId, request)
 	defer func() { go s.eventRepo.PushContainerStoppedEvent(containerId, s.workerId, request) }()