-
Notifications
You must be signed in to change notification settings - Fork 32
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Using psql command instead of postgres in initialScript (#66)
* Use psql instead of postgres, as a result, use `pg_ctl` to temporarily start the postgres server in `${name}-init` process.
* Avoid creating a custom `postgresPkg` with extensions by using `apply` attribute provided by `mkOption`
* Refactor setup script, use `writeShellApplication` and resolve shell-check errors
---------
Co-authored-by: shivaraj-bh <[email protected]>- Loading branch information
1 parent
8cd80d7
commit 22e121b
Showing
5 changed files
with
150 additions
and
119 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
File renamed without changes.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,124 @@ | ||
| { config, pkgs, lib }: | ||
| let | ||
| setupInitialDatabases = | ||
| if config.initialDatabases != [ ] then | ||
| (lib.concatMapStrings | ||
| (database: '' | ||
| echo "Checking presence of database: ${database.name}" | ||
| # Create initial databases | ||
| dbAlreadyExists=$( | ||
| echo "SELECT 1 as exists FROM pg_database WHERE datname = '${database.name}';" | \ | ||
| psql -d postgres | \ | ||
| grep -c 'exists = "1"' || true | ||
| ) | ||
| echo "$dbAlreadyExists" | ||
| if [ 1 -ne "$dbAlreadyExists" ]; then | ||
| echo "Creating database: ${database.name}" | ||
| echo 'create database "${database.name}";' | psql -d postgres | ||
| ${lib.optionalString (database.schema != null) '' | ||
| echo "Applying database schema on ${database.name}" | ||
| if [ -f "${database.schema}" ] | ||
| then | ||
| echo "Running file ${database.schema}" | ||
| awk 'NF' "${database.schema}" | psql -d ${database.name} | ||
| elif [ -d "${database.schema}" ] | ||
| then | ||
| # Read sql files in version order. Apply one file | ||
| # at a time to handle files where the last statement | ||
| # doesn't end in a ;. | ||
| find "${database.schema}"/*.sql | while read -r f ; do | ||
| echo "Applying sql file: $f" | ||
| awk 'NF' "$f" | psql -d ${database.name} | ||
| done | ||
| else | ||
| echo "ERROR: Could not determine how to apply schema with ${database.schema}" | ||
| exit 1 | ||
| fi | ||
| ''} | ||
| fi | ||
| '') | ||
| config.initialDatabases) | ||
| else | ||
| lib.optionalString config.createDatabase '' | ||
| echo "CREATE DATABASE ''${USER:-$(id -nu)};" | psql -d postgres ''; | ||
| runInitialScript = | ||
| let | ||
| scriptCmd = sqlScript: '' | ||
| echo "${sqlScript}" | psql -d postgres | ||
| ''; | ||
| in | ||
| { | ||
| before = with config.initialScript; | ||
| lib.optionalString (before != null) (scriptCmd before); | ||
| after = with config.initialScript; | ||
| lib.optionalString (after != null) (scriptCmd after); | ||
| }; | ||
| toStr = value: | ||
| if true == value then | ||
| "yes" | ||
| else if false == value then | ||
| "no" | ||
| else if lib.isString value then | ||
| "'${lib.replaceStrings [ "'" ] [ "''" ] value}'" | ||
| else | ||
| toString value; | ||
| configFile = pkgs.writeText "postgresql.conf" (lib.concatStringsSep "\n" | ||
| (lib.mapAttrsToList (n: v: "${n} = ${toStr v}") (config.defaultSettings // config.settings))); | ||
|
|
||
| initdbArgs = | ||
| config.initdbArgs | ||
| ++ (lib.optionals (config.superuser != null) [ "-U" config.superuser ]) | ||
| ++ [ "-D" config.dataDir ]; | ||
| in | ||
| (pkgs.writeShellApplication { | ||
| name = "setup-postgres"; | ||
| runtimeInputs = with pkgs; [ config.package coreutils gnugrep gawk ]; | ||
| text = '' | ||
| set -euo pipefail | ||
| # Setup postgres ENVs | ||
| export PGDATA="${config.dataDir}" | ||
| export PGPORT="${toString config.port}" | ||
| POSTGRES_RUN_INITIAL_SCRIPT="false" | ||
| if [[ ! -d "$PGDATA" ]]; then | ||
| initdb ${lib.concatStringsSep " " initdbArgs} | ||
| POSTGRES_RUN_INITIAL_SCRIPT="true" | ||
| echo | ||
| echo "PostgreSQL initdb process complete." | ||
| echo | ||
| fi | ||
| # Setup config | ||
| cp ${configFile} "$PGDATA/postgresql.conf" | ||
| if [[ "$POSTGRES_RUN_INITIAL_SCRIPT" = "true" ]]; then | ||
| echo | ||
| echo "PostgreSQL is setting up the initial database." | ||
| echo | ||
| PGHOST=$(mktemp -d "$(readlink -f ${config.dataDir})/pg-init-XXXXXX") | ||
| export PGHOST | ||
| function remove_tmp_pg_init_sock_dir() { | ||
| if [[ -d "$1" ]]; then | ||
| rm -rf "$1" | ||
| fi | ||
| } | ||
| trap 'remove_tmp_pg_init_sock_dir "$PGHOST"' EXIT | ||
| pg_ctl -D "$PGDATA" -w start -o "-c unix_socket_directories=$PGHOST -c listen_addresses= -p ${toString config.port}" | ||
| ${runInitialScript.before} | ||
| ${setupInitialDatabases} | ||
| ${runInitialScript.after} | ||
| pg_ctl -D "$PGDATA" -m fast -w stop | ||
| remove_tmp_pg_init_sock_dir "$PGHOST" | ||
| else | ||
| echo | ||
| echo "PostgreSQL database directory appears to contain a database; Skipping initialization" | ||
| echo | ||
| fi | ||
| unset POSTGRES_RUN_INITIAL_SCRIPT | ||
| ''; | ||
| }) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters