feat: Allow to pass secrets and environment variables to transformation service (#2632)

hermesdt · Alvaro Duran Tovar · web-flow · commit ffa33ad9860e · 2022-05-10T14:03:29.000-07:00
Signed-off-by: Alvaro Duran Tovar &lt;hermesdt@gmail.com&gt;

Co-authored-by: Alvaro Duran Tovar &lt;hermest@gmail.com&gt;
diff --git a/infra/charts/feast/Chart.yaml b/infra/charts/feast/Chart.yaml
@@ -1,7 +1,7 @@
 apiVersion: v1
 description: Feature store for machine learning
 name: feast
-version: 0.20.0
+version: 0.20.1
 keywords:
   - machine learning
   - big data
diff --git a/infra/charts/feast/charts/transformation-service/templates/deployment.yaml b/infra/charts/feast/charts/transformation-service/templates/deployment.yaml
@@ -30,19 +30,44 @@ spec:
       nodeSelector:
         {{- toYaml . | nindent 8 }}
       {{- end }}
+
+      volumes:
+      {{- range $secret := .Values.secrets }}
+      - name: {{ $secret }}
+        secret:
+          secretName: {{ $secret }}
+      {{- end }}
+
       containers:
       - name: {{ .Chart.Name }}
         image: {{ .Values.image.repository }}:{{ .Values.image.tag }}
         imagePullPolicy: {{ .Values.image.pullPolicy }}
         ports:
         - name: grpc
           containerPort: {{ .Values.service.grpc.targetPort }}
-        
+
+        volumeMounts:
+        {{- range $secret := .Values.secrets }}
+        - name: {{ $secret }}
+          mountPath: "/etc/secrets/{{ $secret }}"
+          readOnly: true
+        {{- end }}
+
         env:
         - name: FEATURE_TRANSFORMATION_SERVER_PORT
           value: {{ .Values.service.grpc.targetPort | quote }}
         - name: FEATURE_STORE_YAML_BASE64
           value: {{ tpl (.Files.Get "config/feature_store.yaml") . | b64enc | quote }}
-        
+
+        {{- range $key, $value := .Values.envOverrides }}
+        - name: {{ printf "%s" $key | replace "." "_" | upper | quote }}
+          {{- if eq (kindOf $value) "map" }}
+          valueFrom:
+            {{- toYaml $value | nindent 12 }}
+          {{- else }}
+          value: {{ $value | quote }}
+          {{- end }}
+        {{- end }}
+
         resources:
           {{- toYaml .Values.resources | nindent 10 }}
diff --git a/infra/charts/feast/charts/transformation-service/values.yaml b/infra/charts/feast/charts/transformation-service/values.yaml
@@ -30,5 +30,8 @@ nodeSelector: {}
 # envOverrides -- Extra environment variables to set
 envOverrides: {}
 
+# secrets -- List of Kubernetes secrets to be mounted. These secrets will be mounted on /etc/secrets/<secret name>.
+secrets: []
+
 # podLabels -- Labels to be added to Feast Serving pods
 podLabels: {}
