mirror of
https://github.com/datahub-project/datahub.git
synced 2025-11-02 03:39:03 +00:00
40 lines
1.8 KiB
Bash
Executable File
40 lines
1.8 KiB
Bash
Executable File
#!/bin/sh
|
|
export PGPASSWORD=$POSTGRES_PASSWORD
|
|
|
|
POSTGRES_CREATE_DB=${POSTGRES_CREATE_DB:-true}
|
|
POSTGRES_CREATE_DB_CONNECTION_DB=${POSTGRES_CREATE_DB_CONNECTION_DB:-postgres}
|
|
CDC_MCL_PROCESSING_ENABLED=${CDC_MCL_PROCESSING_ENABLED:-false}
|
|
CDC_USER=${CDC_USER:-datahub_cdc}
|
|
CDC_PASSWORD=${CDC_PASSWORD:-datahub_cdc}
|
|
|
|
# workaround create database if not exists, check https://stackoverflow.com/a/36591842
|
|
if [ "$POSTGRES_CREATE_DB" = true ]; then
|
|
psql -d "$POSTGRES_CREATE_DB_CONNECTION_DB" -U "$POSTGRES_USERNAME" -h "$POSTGRES_HOST" -p "$POSTGRES_PORT" -tc "SELECT 1 FROM pg_database WHERE datname = '${DATAHUB_DB_NAME}'" | grep -q 1 || psql -d "$POSTGRES_CREATE_DB_CONNECTION_DB" -U "$POSTGRES_USERNAME" -h "$POSTGRES_HOST" -p "$POSTGRES_PORT" -c "CREATE DATABASE ${DATAHUB_DB_NAME}"
|
|
fi
|
|
|
|
# Process main init script
|
|
sed -e "s/DATAHUB_DB_NAME/${DATAHUB_DB_NAME}/g" /init.sql | tee -a /tmp/init-final.sql
|
|
|
|
# Run main init script with regular user
|
|
psql -d "$DATAHUB_DB_NAME" -U "$POSTGRES_USERNAME" -h "$POSTGRES_HOST" -p "$POSTGRES_PORT" < /tmp/init-final.sql
|
|
|
|
# Conditionally create CDC user if enabled
|
|
if [ "${CDC_MCL_PROCESSING_ENABLED}" = "true" ]; then
|
|
echo "CDC MCL processing is enabled. Creating CDC user..."
|
|
|
|
if [ -z "${POSTGRES_PASSWORD}" ]; then
|
|
echo "ERROR: POSTGRES_PASSWORD must be set when CDC_MCL_PROCESSING_ENABLED=true"
|
|
exit 1
|
|
fi
|
|
|
|
# Create CDC user script
|
|
sed -e "s/DATAHUB_DB_NAME/${DATAHUB_DB_NAME}/g" \
|
|
-e "s/CDC_USER/${CDC_USER}/g" \
|
|
-e "s/CDC_PASSWORD/${CDC_PASSWORD}/g" /init-cdc.sql > /tmp/init-cdc-final.sql
|
|
|
|
# Execute CDC user creation with regular user (who has sufficient privileges)
|
|
psql -d "$POSTGRES_CREATE_DB_CONNECTION_DB" -U "$POSTGRES_USERNAME" -h "$POSTGRES_HOST" -p "$POSTGRES_PORT" < /tmp/init-cdc-final.sql
|
|
|
|
echo "CDC user '$CDC_USER' created successfully."
|
|
fi
|