Add heap_size setting env for engine image (#582)

* add heap_size setting env for engine * rename to min_heap_size
Canner · May 29, 2024 · 33b3b98 · 33b3b98
1 parent dfd6d7a
commit 33b3b98
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/docker/Dockerfile b/docker/Dockerfile
@@ -24,4 +24,4 @@ RUN pip3 install -r ./wren-sqlglot-server/requirements.txt
 COPY entrypoint.sh ./
 RUN chmod +x ./entrypoint.sh
 
-CMD ./entrypoint.sh ${WREN_JAR}
+CMD ./entrypoint.sh ${WREN_JAR} ${MAX_HEAP_SIZE} ${MIN_HEAP_SIZE}
diff --git a/docker/entrypoint.sh b/docker/entrypoint.sh
@@ -1,6 +1,9 @@
+#!/bin/bash
+export ENV_MAX_HEAP_SIZE=$2
+export ENV_MIN_HEAP_SIZE=$3
 python3 wren-sqlglot-server/main.py &
 
 # Required add-opens=java.nio=ALL-UNNAMED for Apache arrow in the Snowflake
-java -Dconfig=etc/config.properties \
+java -Xmx${ENV_MAX_HEAP_SIZE:-"512m"} -Xms${ENV_MIN_HEAP_SIZE:-"64m"}  -Dconfig=etc/config.properties \
      --add-opens=java.base/java.nio=ALL-UNNAMED \
      -jar $1