chore(database): add database extension

Author: duanhongyi

Dockerfile

@@ -3,15 +3,15 @@ FROM registry.drycc.cc/drycc/base:${CODENAME}
 
 COPY rootfs/usr /usr/
 COPY rootfs/entrypoint.sh /entrypoint.sh
-ENV PYTHON_VERSION="3.11" \
-  PG_MAJOR=14 \
-  POSTGRESQL_VERSION="15.3"
+ENV PG_MAJOR=15 \
+  PG_MINOR=3 \
+  PYTHON_VERSION="3.11"
 
 ENV PGDATA /data/${PG_MAJOR}
 
 RUN install-packages gcc \
   && install-stack python $PYTHON_VERSION \
-  && install-stack postgresql $POSTGRESQL_VERSION \
+  && install-stack postgresql $PG_MAJOR.$PG_MINOR \
   && . init-stack \
   && set -eux; pip3 install --disable-pip-version-check --no-cache-dir psycopg[binary] patroni[kubernetes] 2>/dev/null; set +eux \
   && apt-get purge -y --auto-remove gcc \

charts/database/templates/_helper.tpl

@@ -38,6 +38,8 @@ env:
   value: '0.0.0.0:8008'
 - name: "DRYCC_DATABASE_INIT_NAMES"
   value: "{{.Values.initDatabases}}"
+- name: "DRYCC_DATABASE_EXTENSIONS"
+  value: "{{.Values.databaseExtensions}}"
 - name: DRYCC_DATABASE_SUPERUSER
   valueFrom:
     secretKeyRef:

charts/database/values.yaml

@@ -56,7 +56,8 @@ replicator: "standby"
 replicatorPassword: ""
 user: ""
 password: ""
-initDatabases: "controller,passport"
+initDatabases: "controller,passport,monitor"
+databaseExtensions: "postgis,timescaledb"
 
 global:
   # Set the location of Workflow's Object Storage

rootfs/entrypoint.sh

@@ -26,6 +26,9 @@ restapi:
   connect_address: '${PATRONI_KUBERNETES_POD_IP}:8008'
 postgresql:
   data_dir: '${PGDATA}'
+  parameters:
+    timescaledb.license: 'timescale'
+    shared_preload_libraries: 'auto_explain,timescaledb,pg_stat_statements'
   connect_address: '${PATRONI_KUBERNETES_POD_IP}:5432'
   authentication:
     superuser:

rootfs/usr/share/scripts/patroni/post_init.sh

@@ -4,11 +4,18 @@ set -Eeu
 if [[ ( -n "$DRYCC_DATABASE_USER") &&  ( -n "$DRYCC_DATABASE_PASSWORD")]]; then
   echo "Creating user ${DRYCC_DATABASE_USER}"
   psql "$1" -w -c "create user ${DRYCC_DATABASE_USER} WITH LOGIN ENCRYPTED PASSWORD '${DRYCC_DATABASE_PASSWORD}'"
-
   for dbname in ${DRYCC_DATABASE_INIT_NAMES//,/ }
   do
     echo "Creating database ${dbname}"
     psql "$1" -w -c "CREATE DATABASE ${dbname} OWNER ${DRYCC_DATABASE_USER}"
+    for extension in ${DRYCC_DATABASE_EXTENSIONS//,/ }
+    do
+      echo "Creating extension ${extension}"
+      psql "$1" -w << EOF
+\c ${dbname};
+create extension ${extension};
+EOF
+    done
   done
 else
   echo "Skipping user creation"