chore: create appsmith schema for postgres

deploy/docker/fs/opt/appsmith/pg-default-schema.sh

@@ -0,0 +1,57 @@
+#!/bin/bash
+
+run_pg_db() {
+  # Create the appsmith schema
+  echo "Initializing PostgreSQL with schema..."
+
+  # Check if APPSMITH_DB_URL is a PostgreSQL URL
+  if [[ -n "$APPSMITH_DB_URL" && "$APPSMITH_DB_URL" == postgres*://* ]]; then
+    echo "APPSMITH_DB_URL is a valid PostgreSQL URL."
+    # Extract username, password, host, port, and database name from APPSMITH_DB_URL
+    DB_USER=$(echo "$APPSMITH_DB_URL" | sed -n 's#.*://\([^:]*\):.*#\1#p')
+    DB_PASSWORD=$(echo "$APPSMITH_DB_URL" | sed -n 's#.*://[^:]*:\([^@]*\)@.*#\1#p')
+    DB_HOST=$(echo "$APPSMITH_DB_URL" | sed -n 's#.*://[^@]*@\([^:]*\):.*#\1#p')
+    DB_PORT=$(echo "$APPSMITH_DB_URL" | sed -n 's#.*://[^@]*@[^:]*:\([^/]*\)/.*#\1#p')
+    DB_NAME=$(echo "$APPSMITH_DB_URL" | sed -n 's#.*://[^@]*@[^/]*/\([^?]*\).*#\1#p')
+
+    # Set defaults for host, port, and database name if not set
+    DB_HOST=${DB_HOST:-localhost}
+    DB_PORT=${DB_PORT:-5432}
+    DB_NAME=${DB_NAME:-appsmith}
+
+    for i in {1..60}; do
+      PGPASSWORD=$DB_PASSWORD psql -h "$DB_HOST" -p "$DB_PORT" -U "$DB_USER" -d "$DB_NAME" -c "SELECT 1;" &> /dev/null
+      if [ $? -eq 0 ]; then
+        echo "PostgreSQL is available!"
+        break
+      else
+        echo "PostgreSQL is unavailable - waiting..."
+        sleep 10
+      fi
+
+      # If PostgreSQL isn't available after 30 attempts (60 seconds), exit with failure
+      if [ $i -eq 30 ]; then
+        echo "PostgreSQL is still unavailable after multiple attempts. Exiting..."
+        exit 1
+      fi
+    done
+
+    echo "Connecting to PostgreSQL at $DB_HOST:$DB_PORT with user $DB_USER"
+
+    # Execute SQL to create the appsmith schema if it doesn't exist (single line)
+    PGPASSWORD=$DB_PASSWORD psql -h "$DB_HOST" -p "$DB_PORT" -U "$DB_USER" -d "$DB_NAME" -c "CREATE SCHEMA IF NOT EXISTS appsmith;"
+
+    # Check if the schema creation was successful
+    if [ $? -eq 0 ]; then
+      echo "Schema 'appsmith' created or already exists."
+    else
+      echo "Failed to create schema 'appsmith'."
+      exit 1
+    fi
+    echo "PostgreSQL initialization completed."
+  else
+      echo "APPSMITH_DB_URL is either empty or not a PostgreSQL URL. Skipping PostgreSQL initialization."
+    fi
+}
+
+run_pg_db

deploy/docker/fs/opt/appsmith/pg-utils.sh

@@ -0,0 +1,92 @@
+#!/bin/bash
+
+waitForPostgresAvailability() {
+  if [ -z "$PG_DB_HOST" ]; then
+    tlog "PostgreSQL host name is empty. Check env variables. Error. Exiting java setup"
+    exit 2
+  else
+
+    MAX_RETRIES=50
+    RETRYSECONDS=10
+    retry_count=0
+    while true; do
+      su postgres -c "pg_isready -h '${PG_DB_HOST}' -p '${PG_DB_PORT}'"
+      status=$?
+
+      case $status in
+      0)
+        tlog "PostgreSQL host '$PG_DB_HOST' is ready."
+        break
+        ;;
+      1)
+        tlog "PostgreSQL host '$PG_DB_HOST' is rejecting connections e.g. due to being in recovery mode or not accepting connections eg. connections maxed out."
+        ;;
+      2)
+        tlog "PostgreSQL host '$PG_DB_HOST' is not responding or running."
+        ;;
+      3)
+        tlog "The connection check failed e.g. due to network issues or incorrect parameters."
+        ;;
+      *)
+        tlog "pg_isready exited with unexpected status code: $status"
+        break
+        ;;
+      esac
+
+      retry_count=$((retry_count + 1))
+      if [ $retry_count -le $MAX_RETRIES ]; then
+        tlog "PostgreSQL connection failed. Retrying attempt $retry_count/$MAX_RETRIES in $RETRYSECONDS seconds..."
+        sleep $RETRYSECONDS
+      else
+        tlog "Exceeded maximum retry attempts ($MAX_RETRIES). Exiting."
+        # use exit code 2 to indicate that the script failed to connect to postgres and supervisor conf is set not to restart the program for 2.
+        exit 2
+      fi
+
+    done
+  fi
+}
+
+# for PostgreSQL, we use APPSMITH_DB_URL=postgresql://username:password@postgresserver:5432/dbname
+# Args:
+#     conn_string (string): PostgreSQL connection string
+# Returns:
+#     None
+# Example:
+#     postgres syntax
+#       "postgresql://user:password@localhost:5432/appsmith"
+#       "postgresql://user:password@localhost/appsmith"
+#       "postgresql://user@localhost:5432/appsmith"
+#       "postgresql://user@localhost/appsmith"
+extract_postgres_db_params() {
+  local conn_string=$1
+
+  # Use node to parse the URI and extract components
+  IFS=' ' read -r USER PASSWORD HOST PORT DB <<<$(node -e "
+    const connectionString = process.argv[1];
+    const pgUri = connectionString.startsWith(\"postgresql://\")
+      ? connectionString
+      : 'http://' + connectionString; //Prepend a fake scheme for URL parsing
+    const url = require('url');
+    const parsedUrl = new url.URL(pgUri);
+
+    // Extract the pathname and remove the leading '/'
+    const db = parsedUrl.pathname.substring(1);
+
+    // Default the port to 5432 if it's empty
+    const port = parsedUrl.port || '5432';
+
+    console.log(\`\${parsedUrl.username || '-'} \${parsedUrl.password || '-'} \${parsedUrl.hostname} \${port} \${db}\`);
+  " "$conn_string")
+
+  # Now, set the environment variables
+  export PG_DB_USER="$USER"
+  export PG_DB_PASSWORD="$PASSWORD"
+  export PG_DB_HOST="$HOST"
+  export PG_DB_PORT="$PORT"
+  export PG_DB_NAME="$DB"
+}
+
+# Example usage of the functions
+# waitForPostgresAvailability
+# extract_postgres_db_params "postgresql://user:password@localhost:5432/dbname"

deploy/docker/fs/opt/appsmith/run-java.sh

@@ -1,5 +1,8 @@
 #!/bin/bash
 
+# Source the helper script
+source pg-utils.sh
+
 set -o errexit
 set -o pipefail
 set -o nounset
@@ -29,6 +32,12 @@ match-proxy-url() {
   [[ -n $proxy_host ]]
 }
 
+# Extract the database parameters from the APPSMITH_DB_URL and wait for the database to be available
+if [[ "$mode" == "pg" ]]; then
+  extract_postgres_db_params "$APPSMITH_DB_URL"
+  waitForPostgresAvailability
+fi
+
 if match-proxy-url "${HTTP_PROXY-}"; then
   extra_args+=(-Dhttp.proxyHost="$proxy_host" -Dhttp.proxyPort="$proxy_port")
   if [[ -n $proxy_user ]]; then
@@ -72,6 +81,8 @@ while ! curl --fail --silent localhost:"${APPSMITH_RTS_PORT:-8091}"/rts-api/v1/h
 done
 tlog 'RTS started.'
 
+sh /opt/appsmith/pg-default-schema.sh &
+
 sh /opt/appsmith/run-starting-page-init.sh &
 
 # Ref -Dlog4j2.formatMsgNoLookups=true https://spring.io/blog/2021/12/10/log4j2-vulnerability-and-spring-boot