From 4327bf017f0075b55d47665f6efcf4444269e430 Mon Sep 17 00:00:00 2001
From: Zebing Lin <linzebing1995@gmail.com>
Date: Tue, 1 Mar 2022 14:10:18 -0800
Subject: [PATCH] Add exchange manager properties to charts

---
 charts/trino/README.md                            | 3 +++
 charts/trino/templates/configmap-coordinator.yaml | 7 +++++++
 charts/trino/templates/configmap-worker.yaml      | 7 +++++++
 charts/trino/values.yaml                          | 5 +++++
 4 files changed, 22 insertions(+)

diff --git a/charts/trino/README.md b/charts/trino/README.md
index 79665042..4a1a1b6e 100644
--- a/charts/trino/README.md
+++ b/charts/trino/README.md
@@ -24,6 +24,8 @@ The following table lists the configurable parameters of the Trino chart and the
 | `server.config.query.maxMemory` |  | `"4GB"` |
 | `server.config.query.maxMemoryPerNode` |  | `"1GB"` |
 | `server.config.memory.heapHeadroomPerNode` |  | `"1GB"` |
+| `server.exchangeManager.name` |  | `"filesystem"` |
+| `server.exchangeManager.baseDir` |  | `"/tmp/trino-local-file-system-exchange-manager"` |
 | `server.jvm.maxHeapSize` |  | `"8G"` |
 | `server.jvm.gcMethod.type` |  | `"UseG1GC"` |
 | `server.jvm.gcMethod.g1.heapRegionSize` |  | `"32M"` |
@@ -34,6 +36,7 @@ The following table lists the configurable parameters of the Trino chart and the
 | `additionalJVMConfig` |  | `{}` |
 | `additionalConfigProperties` |  | `{}` |
 | `additionalLogProperties` |  | `{}` |
+| `additionalExchangeManagerProperties` |  | `{}` |
 | `additionalCatalogs` |  | `{}` |
 | `env` |  | `[]` |
 | `securityContext.runAsUser` |  | `1000` |
diff --git a/charts/trino/templates/configmap-coordinator.yaml b/charts/trino/templates/configmap-coordinator.yaml
index 57a15cde..5564381a 100644
--- a/charts/trino/templates/configmap-coordinator.yaml
+++ b/charts/trino/templates/configmap-coordinator.yaml
@@ -53,6 +53,13 @@ data:
     {{ $configValue }}
   {{- end }}
 
+  exchange-manager.properties: |
+    exchange-manager.name={{ .Values.server.exchangeManager.name }}
+    exchange.base-directory={{ .Values.server.exchangeManager.baseDir }}
+  {{- range $configValue := .Values.additionalExchangeManagerProperties }}
+    {{ $configValue }}
+  {{- end }}
+
   log.properties: |
     io.trino={{ .Values.server.log.trino.level }}
   {{- range $configValue := .Values.additionalLogProperties }}
diff --git a/charts/trino/templates/configmap-worker.yaml b/charts/trino/templates/configmap-worker.yaml
index 4df74f6a..a72f336a 100644
--- a/charts/trino/templates/configmap-worker.yaml
+++ b/charts/trino/templates/configmap-worker.yaml
@@ -48,6 +48,13 @@ data:
     {{ $configValue }}
   {{- end }}
 
+  exchange-manager.properties: |
+    exchange-manager.name={{ .Values.server.exchangeManager.name }}
+    exchange.encryption-enabled={{ .Values.server.exchangeManager.encryptionEnabled }}
+  {{- range $configValue := .Values.additionalExchangeManagerProperties }}
+    {{ $configValue }}
+  {{- end }}
+
   log.properties: |
     io.trino={{ .Values.server.log.trino.level }}
   {{- range $configValue := .Values.additionalLogProperties }}
diff --git a/charts/trino/values.yaml b/charts/trino/values.yaml
index 190c08cf..481efa47 100644
--- a/charts/trino/values.yaml
+++ b/charts/trino/values.yaml
@@ -26,6 +26,9 @@ server:
       maxMemoryPerNode: "1GB"
     memory:
       heapHeadroomPerNode: "1GB"
+  exchangeManager:
+    name: "filesystem"
+    baseDir: "/tmp/trino-local-file-system-exchange-manager"
   jvm:
     maxHeapSize: "8G"
     gcMethod:
@@ -45,6 +48,8 @@ additionalConfigProperties: {}
 
 additionalLogProperties: {}
 
+additionalExchangeManagerProperties: {}
+
 additionalCatalogs: {}
 
 # Array of EnvVar (https://v1-18.docs.kubernetes.io/docs/reference/generated/kubernetes-api/v1.18/#envvar-v1-core)