diff options
| -rwxr-xr-x | rust/tests/pg_tmp.sh | 131 |
1 files changed, 131 insertions, 0 deletions
diff --git a/rust/tests/pg_tmp.sh b/rust/tests/pg_tmp.sh new file mode 100755 index 00000000..cad4fc78 --- /dev/null +++ b/rust/tests/pg_tmp.sh @@ -0,0 +1,131 @@ +#!/bin/sh +# +# Copyright (c) 2014 Eric Radman <ericshane@eradman.com> +# +# Permission to use, copy, modify, and distribute this software for any +# purpose with or without fee is hereby granted, provided that the above +# copyright notice and this permission notice appear in all copies. +# +# THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES +# WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF +# MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR +# ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES +# WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN +# ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF +# OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. + +usage() { + >&2 echo "release: ${release}" + >&2 echo "usage: pg_tmp [-t [-p port]] [-w timeout] [-o extra-options] [-d datadir]" + exit 1 +} + +trap 'printf "$0: exit code $? on line $LINENO\n" >&2; exit 1' ERR \ + 2> /dev/null || exec bash $0 "$@" +trap '' HUP +set +o posix + +USER_OPTS="" +>/dev/null getopt tp:w:o:d: "$@" || usage +while [ $# -gt 0 ]; do + case "$1" in + -t) LISTENTO="127.0.0.1" ;; + -p) PGPORT="$2"; shift ;; + -w) TIMEOUT="$2"; shift ;; + -o) USER_OPTS="$2"; shift ;; + -d) TD="$2"; shift ;; + *) CMD=$1 ;; + esac + shift +done + +initdb -V > /dev/null || exit 1 +PGVER=$(psql -V | awk '{print $NF}') + +[ -n "$LISTENTO" ] && [ -z "$PGPORT" ] && { + PGPORT="$(getsocket)" +} + +case ${CMD:-start} in +initdb) + [ -z $TD ] || mkdir -p $TD + [ -z $TD ] && TD="$(mktemp -d ${SYSTMP:-/tmp}/ephemeralpg.XXXXXX)" + initdb --nosync -D $TD/$PGVER -E UNICODE -A trust > $TD/initdb.out + cat <<-EOF >> $TD/$PGVER/postgresql.conf + unix_socket_directories = '$TD' + listen_addresses = '' + shared_buffers = 12MB + fsync = off + synchronous_commit = off + full_page_writes = off + log_min_duration_statement = 0 + log_connections = on + log_disconnections = on + EOF + touch $TD/NEW + echo $TD + ;; +start) + # 1. Find a temporary database directory owned by the current user + # 2. Create a new datadir if nothing was found + # 3. Launch a background task to create a datadir for future invocations + if [ -z $TD ]; then + for d in $(ls -d ${SYSTMP:-/tmp}/ephemeralpg.*/$PGVER 2> /dev/null); do + td=$(dirname "$d") + test -O $td/NEW && rm $td/NEW 2> /dev/null && { TD=$td; break; } + done + [ -z $TD ] && { TD=$($0 initdb); rm $TD/NEW; } + nice -n 19 $0 initdb > /dev/null & + else + [ -O $TD/$PGVER ] || TD=$($0 initdb -d $TD) + fi + nice -n 19 $0 -w ${TIMEOUT:-60} -d $TD -p ${PGPORT:-5432} stop > $TD/stop.log 2>&1 & + [ -n "$PGPORT" ] && OPTS="-c listen_addresses='$LISTENTO' -c port=$PGPORT" + LOGFILE="$TD/$PGVER/postgres.log" + pg_ctl -W -o "$OPTS $USER_OPTS" -s -D $TD/$PGVER -l $LOGFILE start + PGHOST=$TD + export PGPORT PGHOST + if [ -n "$PGPORT" ]; then + url="postgresql://$(whoami)@$LISTENTO:$PGPORT/test" + else + url="postgresql:///test?host=$(echo $PGHOST | sed 's:/:%2F:g')" + fi + for n in 1 2 3 4 5; do + sleep 0.1 + createdb -E UNICODE test > /dev/null 2>&1 && break + done + [ $? != 0 ] && { >&2 cat $LOGFILE; exit 1; } + [ -t 1 ] && echo "$url" || echo -n "$url" + ;; +stop) + [ -O $TD/$PGVER/postgresql.conf ] || { + >&2 echo "Please specify a PostgreSQL data directory using -d" + exit 1 + } + trap "rm -r $TD" EXIT + PGHOST=$TD + export PGHOST PGPORT + q="SELECT count(*) FROM pg_stat_activity WHERE datname='test';" + until [ "${count:-2}" -lt "2" ]; do + sleep ${TIMEOUT:-0} + count=$(psql test --no-psqlrc -At -c "$q" || echo 0) + done + pg_ctl -W -D $TD/$PGVER stop + sleep 1 + ;; +selftest) + export SYSTMP=$(mktemp -d /tmp/ephemeralpg-selftest.XXXXXX) + trap "rm -r $SYSTMP" EXIT + printf "Running: " + printf "initdb "; dir=$($0 initdb) + printf "start " ; url=$($0 -w 3 -o '-c log_temp_files=100' start) + printf "psql " ; [ "$(psql --no-psqlrc -At -c 'select 5' $url)" == "5" ] + printf "stop " ; sleep 10 + printf "verify "; ! [ -d dir ] + echo; echo "OK" + ;; +*) + usage + ;; +esac + |