Add support for Presto CPP

prestodb · Nov 15, 2024 · 8168904 · 8168904
1 parent f91b719
commit 8168904
Show file tree

Hide file tree

Showing 6 changed files with 31 additions and 6 deletions.
diff --git a/charts/presto/Chart.yaml b/charts/presto/Chart.yaml
@@ -3,7 +3,7 @@ name: presto
 description: The official Helm chart for Presto
 type: application
 version: 0.3.0
-appVersion: "0.288"
+appVersion: "0.289"
 home: https://prestodb.io
 icon: https://prestodb.io/docs/current/_static/logo.png
 keywords:

diff --git a/charts/presto/templates/configmap-coordinator.yaml b/charts/presto/templates/configmap-coordinator.yaml
@@ -29,7 +29,15 @@ data:
     discovery-server.enabled={{ or (eq .Values.mode "single") (eq .Values.mode "cluster") }}
     node-scheduler.include-coordinator={{ eq .Values.mode "single" }}
     resource-manager-enabled={{ eq .Values.mode "ha-cluster" }}
-    {{- .Values.config | nindent 4 }}
+    presto.version={{ .Values.image.tag | default .Chart.AppVersion }}{{- if .Values.prestoCpp.enabled }}wCPP{{ .Values.prestoCpp.image.tag }}{{- end }}
+    {{- if .Values.prestoCpp.enabled }}
+    native-execution-enabled=true
+    optimizer.optimize-hash-generation=false
+    regex-library=RE2J
+    use-alternative-function-signatures=true
+    experimental.table-writer-merge-operator-enabled=false
+    {{- end }}
+    {{- .Values.coordinator.config | default .Values.config | nindent 4 }}
   log.properties: |-
     {{- .Values.log | nindent 4 }}
   {{- if .Values.resourceGroups.manager }}

diff --git a/charts/presto/templates/configmap-resource-manager.yaml b/charts/presto/templates/configmap-resource-manager.yaml
@@ -33,7 +33,8 @@ data:
     resource-manager=true
     thrift.server.port=8081
     thrift.server.ssl.enabled=false
-    {{- .Values.config | nindent 4 }}
+    presto.version={{ .Values.image.tag | default .Chart.AppVersion }}{{- if .Values.prestoCpp.enabled }}wCPP{{ .Values.prestoCpp.image.tag }}{{- end }}
+    {{- .Values.resourceManager.config | default .Values.config | nindent 4 }}
   log.properties: |-
     {{- .Values.log | nindent 4 }}
   {{- if .Values.metrics.enabled }}

diff --git a/charts/presto/templates/configmap-worker.yaml b/charts/presto/templates/configmap-worker.yaml
@@ -28,7 +28,8 @@ data:
     http-server.http.port={{ .Values.service.port }}
     discovery.uri=http://{{ .Release.Name }}-discovery:{{ .Values.service.port }}
     resource-manager-enabled={{ eq .Values.mode "ha-cluster" }}
-    {{- .Values.config | nindent 4 }}
+    presto.version={{ .Values.image.tag | default .Chart.AppVersion }}{{- if .Values.prestoCpp.enabled }}wCPP{{ .Values.prestoCpp.image.tag }}{{- end }}
+    {{- .Values.worker.config | default .Values.config | nindent 4 }}
   log.properties: |-
     {{- .Values.log | nindent 4 }}
   {{- if .Values.metrics.enabled }}

diff --git a/charts/presto/templates/deployment-worker.yaml b/charts/presto/templates/deployment-worker.yaml
@@ -50,8 +50,8 @@ spec:
       {{- end }}
       containers:
         - name: worker
-          image: "{{ .Values.image.repository }}:{{ .Values.image.tag | default .Chart.AppVersion }}"
-          imagePullPolicy: {{ .Values.image.pullPolicy }}
+          image: "{{ if .Values.prestoCpp.enabled }}{{ .Values.prestoCpp.image.repository }}:{{ .Values.prestoCpp.image.tag | default .Chart.AppVersion }}{{ else }}{{ .Values.image.repository }}:{{ .Values.image.tag | default .Chart.AppVersion }}{{ end }}"
+          imagePullPolicy: {{ if .Values.prestoCpp.enabled }}{{ .Values.prestoCpp.image.pullPolicy }}{{ else }}{{ .Values.image.pullPolicy }}{{ end }}
           {{- with .Values.worker.command }}
           command: {{- tpl (toYaml .) $ | nindent 12 }}
           {{- end }}

diff --git a/charts/presto/values.yaml b/charts/presto/values.yaml
@@ -90,6 +90,8 @@ log: |-
 resourceManager:
   # Resource manager JVM options (overwrites common JVM options)
   jvm: ""
+  # Resource manager configuration properties (overwrites common config options)
+  config: ~
   # Command to launch resource manager (templated)
   command: ~
   # Arguments to launch resource manager (templated)
@@ -142,6 +144,8 @@ coordinator:
   replicas: 1
   # Coordinator JVM options (overwrites common JVM options)
   jvm: ""
+  # Coordinator configuration properties (overwrites common config options)
+  config: ~
   # Command to launch coordinator (templated)
   command: ~
   # Arguments to launch coordinator (templated)
@@ -194,6 +198,8 @@ worker:
   replicas: 2
   # Worker JVM options (overwrites common JVM options)
   jvm: ""
+  # Worker configuration properties (overwrites common config options)
+  config: ~
   # Command to launch worker (templated)
   command: ~
   # Arguments to launch worker (templated)
@@ -239,6 +245,15 @@ worker:
   tolerations: []
   # Worker security context (overwrites default security context)
   securityContext: {}
+
+# Enables Presto to use C++ based workers  
+prestoCpp: 
+  enabled: false
+  # Image details for Presto C++ workers (overrides common image details)
+  image:
+    repository: prestodb/presto-native
+    pullPolicy: IfNotPresent
+    tag: ~
 
 # Catalogs
 catalog: {}