From 49307745f906195b72b945e21489a690e08647d0 Mon Sep 17 00:00:00 2001
From: yetone <yetoneful@gmail.com>
Date: Mon, 13 Mar 2023 04:59:16 +0000
Subject: [PATCH] feat: installation is supported by specifying the proxy image

---
 controllers/bentodeployment_controller.go       | 7 ++++++-
 helm/yatai-deployment/templates/secret-env.yaml | 1 +
 helm/yatai-deployment/values.yaml               | 1 +
 3 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/controllers/bentodeployment_controller.go b/controllers/bentodeployment_controller.go
index fb2fb5a..a467bde 100644
--- a/controllers/bentodeployment_controller.go
+++ b/controllers/bentodeployment_controller.go
@@ -2188,9 +2188,14 @@ monitoring.options.insecure=true`
 				},
 			},
 		})
+		proxyImage := "quay.io/bentoml/bentoml-proxy:0.0.1"
+		proxyImage_ := os.Getenv("INTERNAL_IMAGES_PROXY")
+		if proxyImage_ != "" {
+			proxyImage = proxyImage_
+		}
 		containers = append(containers, corev1.Container{
 			Name:  "proxy",
-			Image: "quay.io/bentoml/bentoml-proxy:0.0.1",
+			Image: proxyImage,
 			Command: []string{
 				"envoy",
 				"--config-path",
diff --git a/helm/yatai-deployment/templates/secret-env.yaml b/helm/yatai-deployment/templates/secret-env.yaml
index 9774760..3659eaf 100644
--- a/helm/yatai-deployment/templates/secret-env.yaml
+++ b/helm/yatai-deployment/templates/secret-env.yaml
@@ -13,6 +13,7 @@ stringData:
   INTERNAL_IMAGES_METRICS_TRANSFORMER: {{ .Values.internalImages.metricsTransformer | quote }}
   INTERNAL_IMAGES_DEBUGGER: {{ .Values.internalImages.debugger | quote }}
   INTERNAL_IMAGES_MONITOR_EXPORTER: {{ .Values.internalImages.monitorExporter | quote }}
+  INTERNAL_IMAGES_PROXY: {{ .Values.internalImages.proxy | quote }}
 
   {{- if .Values.disableAutomateBentoImageBuilder }}
   DISABLE_AUTOMATE_BENTO_IMAGE_BUILDER: "true"
diff --git a/helm/yatai-deployment/values.yaml b/helm/yatai-deployment/values.yaml
index f82d3cd..df3ed45 100644
--- a/helm/yatai-deployment/values.yaml
+++ b/helm/yatai-deployment/values.yaml
@@ -111,6 +111,7 @@ internalImages:
   metricsTransformer: quay.io/bentoml/yatai-bento-metrics-transformer:0.0.3
   debugger: quay.io/bentoml/bento-debugger:0.0.5
   monitorExporter: quay.io/bentoml/bentoml-monitor-exporter:0.0.3
+  proxy: quay.io/bentoml/bentoml-proxy:0.0.1
 
 disableAutomateBentoImageBuilder: false
 enableRestrictedSecurityContext: false