Files
zulip/scripts/setup/terminate-psql-sessions
2019-04-17 12:15:37 -07:00

26 lines
535 B
Bash
Executable File

#!/usr/bin/env bash
set -e
DEFAULT_USER="postgres"
if [ "$(uname)" = "OpenBSD" ]; then
DEFAULT_USER="_postgresql"
fi
cd /
username=$1
shift
tables="$(printf "'%s'," "${@//\'/\'\'}")"
tables="${tables%,}"
if [ "$EUID" -eq 0 ]; then
su - "$DEFAULT_USER" -c "psql postgres '$DEFAULT_USER'" <<EOF
SELECT pg_terminate_backend(pid) FROM pg_stat_activity WHERE datname IN ($tables);
EOF
else
psql -h localhost postgres "$username" <<EOF
SELECT pg_terminate_backend(pid) FROM pg_stat_activity WHERE datname IN ($tables);
EOF
fi