|
@@ -14,8 +14,6 @@ PGHOST=${PGHOST:-127.0.0.1}
|
|
PGPORT=${PGPORT:-5432}
|
|
PGPORT=${PGPORT:-5432}
|
|
PGPASSWORD=${PGPASSWORD:-postgres}
|
|
PGPASSWORD=${PGPASSWORD:-postgres}
|
|
PGUSER=${PGUSER:-postgres}
|
|
PGUSER=${PGUSER:-postgres}
|
|
-PGDUMP=${PGDUMP:-pg_dumpall}
|
|
|
|
-PGRESTORE=${PGRESTORE:-psql}
|
|
|
|
export PGHOST
|
|
export PGHOST
|
|
export PGPORT
|
|
export PGPORT
|
|
export PGPASSWORD
|
|
export PGPASSWORD
|
|
@@ -25,8 +23,8 @@ about() {
|
|
die "usage: $script_name [ config_yaml | setup | dump <backup_file> | restore <backup_file> ]"
|
|
die "usage: $script_name [ config_yaml | setup | dump <backup_file> | restore <backup_file> ]"
|
|
}
|
|
}
|
|
|
|
|
|
-check_postgres_client() {
|
|
|
|
- if ! command -v psql --version >/dev/null; then
|
|
|
|
|
|
+check_requirements() {
|
|
|
|
+ if ! command -v psql >/dev/null; then
|
|
die "missing required program 'psql' as a postgres client (package postgres-client-13 on debian like system)"
|
|
die "missing required program 'psql' as a postgres client (package postgres-client-13 on debian like system)"
|
|
fi
|
|
fi
|
|
}
|
|
}
|
|
@@ -36,10 +34,6 @@ exec_sql() {
|
|
psql <<< "$cmd"
|
|
psql <<< "$cmd"
|
|
}
|
|
}
|
|
|
|
|
|
-requirements() {
|
|
|
|
- check_mysql_client
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
setup() {
|
|
setup() {
|
|
exec_sql "DROP DATABASE IF EXISTS crowdsec_test;"
|
|
exec_sql "DROP DATABASE IF EXISTS crowdsec_test;"
|
|
exec_sql "CREATE DATABASE crowdsec_test;"
|
|
exec_sql "CREATE DATABASE crowdsec_test;"
|
|
@@ -50,13 +44,13 @@ setup() {
|
|
|
|
|
|
dump() {
|
|
dump() {
|
|
backup_file="${1?Missing file to backup database to}"
|
|
backup_file="${1?Missing file to backup database to}"
|
|
- "${PGDUMP}" > "$backup_file" 2>/dev/null
|
|
|
|
|
|
+ pg_dump -Ft --dbname crowdsec_test --file "$backup_file"
|
|
}
|
|
}
|
|
|
|
|
|
restore() {
|
|
restore() {
|
|
backup_file="${1?missing file to restore database from}"
|
|
backup_file="${1?missing file to restore database from}"
|
|
[ -f "$backup_file" ] || die "Backup file $backup_file doesn't exist"
|
|
[ -f "$backup_file" ] || die "Backup file $backup_file doesn't exist"
|
|
- "${PGRESTORE}" < "$backup_file" #seems that in some system it needs for two run
|
|
|
|
|
|
+ pg_restore --dbname crowdsec_test --clean "$backup_file"
|
|
}
|
|
}
|
|
|
|
|
|
config_yaml() {
|
|
config_yaml() {
|
|
@@ -74,6 +68,8 @@ config_yaml() {
|
|
|
|
|
|
[ $# -lt 1 ] && about
|
|
[ $# -lt 1 ] && about
|
|
|
|
|
|
|
|
+check_requirements
|
|
|
|
+
|
|
case "$1" in
|
|
case "$1" in
|
|
setup)
|
|
setup)
|
|
setup
|
|
setup
|