mirror of https://github.com/zulip/zulip.git
31 lines
752 B
Bash
Executable File
31 lines
752 B
Bash
Executable File
#!/usr/bin/env bash
|
|
set -e
|
|
|
|
DEFAULT_USER="postgres"
|
|
if [ "$(uname)" = "OpenBSD" ]; then
|
|
DEFAULT_USER="_postgresql"
|
|
fi
|
|
|
|
cd /
|
|
if [ "$EUID" -eq 0 ]; then
|
|
version=$(sudo -u "$DEFAULT_USER" sh -c 'psql -A -t -d postgres -c "show server_version"')
|
|
else
|
|
version=$(psql -A -t -d postgres -c "show server_version")
|
|
fi
|
|
|
|
major=$(echo "$version" | cut -d. -f1,2)
|
|
username=$1
|
|
|
|
shift
|
|
tables=$(echo "'$*'" | sed "s/ /','/g")
|
|
|
|
if [ "$EUID" -eq 0 ]; then
|
|
sudo -u "$DEFAULT_USER" sh -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
|