fix cancel build to cancel pending stages

zc2638 · Feb 4, 2024 · 1d5ccfc · 1d5ccfc
1 parent 908bf73
commit 1d5ccfc
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/core/service/build/build.go b/core/service/build/build.go
@@ -272,6 +272,15 @@ func (s *srv) Cancel(ctx context.Context, namespace, name string, number uint64)
 		return errors.New("already done")
 	}
 
+	if err := db.Model(&storageV1.Stage{}).
+		Where(&storageV1.Stage{
+			BoxID:   buildS.BoxID,
+			BuildID: buildS.ID,
+			Phase:   v1.PhasePending.String(),
+		}).Update("phase", v1.PhaseCanceled).Error; err != nil {
+		return err
+	}
+
 	sched := scheduler.FromContext(ctx)
 	return sched.Cancel(ctx, int64(build.ID))
 }