remove scheduler stop build and move container instance set back

pkg/scheduler/scheduler.go

@@ -164,21 +164,6 @@ func (s *Scheduler) Stop(stopArgs *types.StopContainerArgs) error {
 	return nil
 }
 
-func (s *Scheduler) StopBuild(containerId string) error {
-	_, err := s.eventBus.Send(&common.Event{
-		Type:          common.EventType("stop-build" + "-" + containerId),
-		Args:          map[string]any{"container_id": containerId},
-		LockAndDelete: false,
-	})
-	if err != nil {
-		log.Error().Err(err).Msg("failed to send stop build event")
-		return err
-	}
-
-	log.Info().Str("container_id", containerId).Msg("sent stop build event")
-	return nil
-}
-
 func (s *Scheduler) getControllers(request *types.ContainerRequest) ([]WorkerPoolController, error) {
 	controllers := []WorkerPoolController{}
 

pkg/worker/lifecycle.go

@@ -156,6 +156,13 @@ func (s *Worker) deleteContainer(containerId string, err error) {
 func (s *Worker) RunContainer(ctx context.Context, request *types.ContainerRequest) error {
 	containerId := request.ContainerId
 
+	s.containerInstances.Set(containerId, &ContainerInstance{
+		Id:        containerId,
+		StubId:    request.StubId,
+		LogBuffer: common.NewLogBuffer(),
+		Request:   request,
+	})
+
 	bundlePath := filepath.Join(s.imageMountPath, request.ImageId)
 
 	// Set worker hostname

pkg/worker/worker.go

@@ -280,12 +280,6 @@ func (s *Worker) Run() error {
 				}})
 				go eventbus.ReceiveEvents(ctx)
 
-				s.containerInstances.Set(containerId, &ContainerInstance{
-					Id:        containerId,
-					StubId:    request.StubId,
-					LogBuffer: common.NewLogBuffer(),
-					Request:   request,
-				})
 				err := s.RunContainer(ctx, request)
 				if err != nil {
 					log.Error().Str("container_id", containerId).Err(err).Msg("unable to run container")