From 3464882cf544dbbd3480aff049dc564a13dacb81 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=80=86=E6=B5=81=E8=80=8C=E4=B8=8A?=
 <62740231+DokiDoki1103@users.noreply.github.com>
Date: Mon, 8 Jan 2024 09:58:48 +0800
Subject: [PATCH] feat: add building affinity (#1824)

* feat: add building affinity
---
 builder/build/code_build.go       | 5 +++++
 builder/build/dockerfile_build.go | 5 +++++
 builder/sources/image.go          | 5 +++++
 3 files changed, 15 insertions(+)

diff --git a/builder/build/code_build.go b/builder/build/code_build.go
index e421a7b0c7..cfbfadb62a 100644
--- 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
index d0b9579058..db5f42c462 100644
--- 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
index 1fa2cecb77..ca4c47c088 100644
--- 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")},
+							},
 						},
 					},
 					},