mirror of
https://github.com/ntop/ntopng.git
synced 2026-04-28 06:59:33 +00:00
87747eae7b
* Implemented the factory reset script * Added options for CH to the factory reset script and fixed Redis reset * Added help option for the factory reset script
104 lines
No EOL
2.4 KiB
Bash
Executable file
104 lines
No EOL
2.4 KiB
Bash
Executable file
#!/bin/bash
|
|
help() {
|
|
echo "Usage: $0 [OPTIONS]"
|
|
echo ""
|
|
echo "This script resets ntopng data by:"
|
|
echo " - Removing /var/lib/ntopng/ contents"
|
|
echo " - Flushing ntopng Redis keys"
|
|
echo " - Dropping ClickHouse ntopng database"
|
|
echo ""
|
|
echo "Options:"
|
|
echo " -c, --clickhouse Enable ClickHouse reset"
|
|
echo " -h, --host <hostname> ClickHouse host (default: localhost)"
|
|
echo " -u, --user <username> ClickHouse username"
|
|
echo " -p, --password <pass> ClickHouse password"
|
|
echo " --help Show this help message and exit"
|
|
echo ""
|
|
echo "Example:"
|
|
echo " sudo $0 --clickhouse --host localhost --user admin --password secret"
|
|
exit 0
|
|
}
|
|
|
|
CLICKHOUSE=0
|
|
HOST=""
|
|
USER=""
|
|
PASS=""
|
|
|
|
while [[ $# -gt 0 ]]; do
|
|
case "$1" in
|
|
--help)
|
|
help
|
|
;;
|
|
-c|--clickhouse)
|
|
CLICKHOUSE=1
|
|
shift
|
|
;;
|
|
-h|--host)
|
|
HOST="$2"
|
|
shift 2
|
|
;;
|
|
-u|--user)
|
|
USER="$2"
|
|
shift 2
|
|
;;
|
|
-p|--password)
|
|
PASS="$2"
|
|
shift 2
|
|
;;
|
|
*)
|
|
echo "Unknown option: $1"
|
|
exit 1
|
|
;;
|
|
esac
|
|
done
|
|
if [ "$EUID" -ne 0 ]; then
|
|
echo "Please run this script as root"
|
|
exit 1
|
|
fi
|
|
|
|
echo "Removing contents of /var/lib/ntopng/..."
|
|
rm -rf /var/lib/ntopng/*
|
|
echo "ntopng data removed."
|
|
|
|
NTOPNG_KEYS=$(redis-cli --raw keys "ntopng.*" | wc -l)
|
|
if [ "$NTOPNG_KEYS" -gt 1 ]; then
|
|
echo "Flushing Redis ($NTOPNG_KEYS keys)..."
|
|
redis-cli --raw keys "ntopng.*" | xargs -n 100 redis-cli del >/dev/nul
|
|
echo "Redis flushed."
|
|
else
|
|
echo "No ntopng-related keys found in Redis. Skipping flush."
|
|
fi
|
|
|
|
check_clickhouse_installed() {
|
|
command -v clickhouse-client >/dev/null 2>&1
|
|
}
|
|
|
|
if [ "$CLICKHOUSE" -eq 1 ]; then
|
|
if check_clickhouse_installed; then
|
|
|
|
echo "Dropping ntopng database from ClickHouse..."
|
|
CMD=("clickhouse-client")
|
|
if [ -n "$HOST" ]; then
|
|
CMD+=(--host="$HOST")
|
|
fi
|
|
if [ -n "$USER" ]; then
|
|
CMD+=(--user="$USER")
|
|
fi
|
|
if [ -n "$PASS" ]; then
|
|
CMD+=(--password="$PASS")
|
|
fi
|
|
CMD+=(--query="DROP DATABASE IF EXISTS ntopng")
|
|
|
|
"${CMD[@]}"
|
|
|
|
if [ $? -eq 0 ]; then
|
|
echo "ClickHouse database ntopng dropped."
|
|
else
|
|
echo "Failed to drop ClickHouse database."
|
|
fi
|
|
else
|
|
echo "ClickHouse is not installed."
|
|
fi
|
|
else
|
|
echo "Skipping ClickHouse operations."
|
|
fi |