feat: add building affinity (#1824)

* feat: add building affinity
goodrain · Jan 8, 2024 · 3464882 · 3464882
1 parent dd6ba6f
commit 3464882
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 0 deletions.
diff --git a/builder/build/code_build.go b/builder/build/code_build.go
@@ -383,6 +383,11 @@ func (s *slugBuild) runBuildJob(re *Request) error {
 								Operator: corev1.NodeSelectorOpIn,
 								Values:   []string{re.Arch},
 							},
+							{
+								Key:      "kubernetes.io/hostname",
+								Operator: corev1.NodeSelectorOpIn,
+								Values:   []string{os.Getenv("HOST_IP")},
+							},
 						},
 					},
 					},

diff --git a/builder/build/dockerfile_build.go b/builder/build/dockerfile_build.go
@@ -111,6 +111,11 @@ func (d *dockerfileBuild) runBuildJob(re *Request, buildImageName string) error
 								Operator: corev1.NodeSelectorOpIn,
 								Values:   []string{re.Arch},
 							},
+							{
+								Key:      "kubernetes.io/hostname",
+								Operator: corev1.NodeSelectorOpIn,
+								Values:   []string{os.Getenv("HOST_IP")},
+							},
 						},
 					},
 					},

diff --git a/builder/sources/image.go b/builder/sources/image.go
@@ -434,6 +434,11 @@ func ImageBuild(arch, contextDir, cachePVCName, cacheMode, RbdNamespace, Service
 								Operator: corev1.NodeSelectorOpIn,
 								Values:   []string{arch},
 							},
+							{
+								Key:      "kubernetes.io/hostname",
+								Operator: corev1.NodeSelectorOpIn,
+								Values:   []string{os.Getenv("HOST_IP")},
+							},
 						},
 					},
 					},