Add dnsConfig to head, worker and additional workers (#2377)

ray-project · Oct 10, 2024 · 9a0b9d0 · 9a0b9d0
1 parent bf21d2d
commit 9a0b9d0
Show file tree

Hide file tree

Showing 2 changed files with 45 additions and 0 deletions.
diff --git a/helm-chart/ray-cluster/templates/raycluster-cluster.yaml b/helm-chart/ray-cluster/templates/raycluster-cluster.yaml
@@ -38,6 +38,9 @@ spec:
     {{- end }}
     template:
       spec:
+        {{- if .Values.head.dnsConfig }}
+        dnsConfig: {{- toYaml .Values.head.dnsConfig | nindent 10 }}
+        {{- end }}
         imagePullSecrets: {{- toYaml .Values.imagePullSecrets | nindent 10 }}
         {{- if .Values.head.serviceAccountName }}
         serviceAccountName: {{ .Values.head.serviceAccountName }}
@@ -135,6 +138,9 @@ spec:
     groupName: {{ $groupName }}
     template:
       spec:
+        {{- if $values.dnsConfig }}
+        dnsConfig: {{- toYaml $values.dnsConfig | nindent 10 }}
+        {{- end }}
         imagePullSecrets: {{- toYaml $.Values.imagePullSecrets | nindent 10 }}
         {{- if $values.serviceAccountName }}
         serviceAccountName: {{ $values.serviceAccountName }}
@@ -232,6 +238,9 @@ spec:
     groupName: {{ .Values.worker.groupName }}
     template:
       spec:
+        {{- if .Values.worker.dnsConfig }}
+        dnsConfig: {{- toYaml .Values.worker.dnsConfig | nindent 10 }}
+        {{- end }}
         imagePullSecrets: {{- toYaml .Values.imagePullSecrets | nindent 10 }}
         {{- if .Values.worker.serviceAccountName }}
         serviceAccountName: {{ .Values.worker.serviceAccountName }}

diff --git a/helm-chart/ray-cluster/values.yaml b/helm-chart/ray-cluster/values.yaml
@@ -116,6 +116,18 @@ head:
     #   annotations:
     #     prometheus.io/scrape: "true"
 
+  # Custom pod DNS configuration
+  # See https://kubernetes.io/docs/concepts/services-networking/dns-pod-service/#pod-dns-config
+  # dnsConfig:
+  #   nameservers:
+  #     - 8.8.8.8
+  #   searches:
+  #     - example.local
+  #   options:
+  #     - name: ndots
+  #       value: "2"
+  #     - name: edns0
+
 
 worker:
   # If you want to disable the default workergroup
@@ -179,6 +191,18 @@ worker:
   command: []
   args: []
 
+  # Custom pod DNS configuration
+  # See https://kubernetes.io/docs/concepts/services-networking/dns-pod-service/#pod-dns-config
+  # dnsConfig:
+  #   nameservers:
+  #     - 8.8.8.8
+  #   searches:
+  #     - example.local
+  #   options:
+  #     - name: ndots
+  #       value: "2"
+  #     - name: edns0
+
 # The map's key is used as the groupName.
 # For example, key:small-group in the map below
 # will be used as the groupName
@@ -241,6 +265,18 @@ additionalWorkerGroups:
     command: []
     args: []
 
+    # Custom pod DNS configuration
+    # See https://kubernetes.io/docs/concepts/services-networking/dns-pod-service/#pod-dns-config
+    # dnsConfig:
+    #   nameservers:
+    #     - 8.8.8.8
+    #   searches:
+    #     - example.local
+    #   options:
+    #     - name: ndots
+    #       value: "2"
+    #     - name: edns0
+
 # Configuration for Head's Kubernetes Service
 service:
   # This is optional, and the default is ClusterIP.