清空Redis集群所有节点的数据工具

#!/bin/bash
# Batch to clear all nodes using FLUSHALL command
# 用来清空一个redis集群中的所有数据，要求 FLUSHALL 命令可用，
# 如果在 redis.conf 中使用 rename 改名了 FLUSHALL，则不能执行本脚本。
# 可带两个参数：
# 1）参数1 集群中的任一可用节点（必须）
# 2）连接redis的密码（设置了密码才需要）
REDIS_CLI=${REDIS_CLI:-redis-cli}
REDIS_IP=${REDIS_IP:-127.0.0.1}
REDIS_PORT=${REDIS_PORT:-6379}
# 显示用法函数
function usage()
{
echo "Usage: clear_redis_cluster.sh a_redis_node_of_cluster redis_password"
echo "Example1: clear_redis_cluster.sh '127.0.0.1:6379'"
echo "Example2: clear_redis_cluster.sh '127.0.0.1:6379' '123456'"
}
# 检查参数个数
if test $# -lt 1 -o $# -gt 2; then
usage
exit 1
fi
# 第一个参数为集群中的节点，
REDIS_NODE="$1"
# 第二个参数为密码
REDIS_PASSWORD=""
if test $# -ge 2; then
REDIS_PASSWORD="$2"
fi
# 取得IP和端口
eval $(echo "$REDIS_NODE" | awk -F[\:] '{ printf("REDIS_IP=%s\nREDIS_PORT=%s\n",$1,$2) }')
if test -z "$REDIS_IP" -o -z "$REDIS_PORT"; then
echo "Parameter error: \`$REDIS_NODE\`."
usage
exit 1
fi
# 确保redis-cli可用
echo "Checking \`redis-cli\` ..."
which "$REDIS_CLI" > /dev/null 2>&1
if test $? -ne 0; then
echo "Command \`redis-cli\` is not exists or not executable."
echo "You can set environment variable \`REDIS_CLI\` to point to the redis-cli."
echo "Example: export REDIS_CLI=/usr/local/bin/redis-cli"
exit 1
fi
if test -z "$REDIS_PASSWORD"; then
redis_nodes=`redis-cli -h $REDIS_IP -p $REDIS_PORT cluster nodes | awk -F[\ \:\@] '!/ERR/{ printf("%s:%s\n",$2,$3); }'`
else
redis_nodes=`redis-cli --no-auth-warning -a "$REDIS_PASSWORD" -h $REDIS_IP -p $REDIS_PORT cluster nodes | awk -F[\ \:\@] '!/ERR/{ printf("%s:%s\n",$2,$3); }'`
fi
if test -z "$redis_nodes"; then
# Standlone（非集群）
if test -z "$REDIS_PASSWORD"; then
$REDIS_CLI -h $REDIS_IP -p $REDIS_PORT FLUSHALL ASYNC
$REDIS_CLI -h $REDIS_IP -p $REDIS_PORT BGREWRITEAOF
else
$REDIS_CLI --no-auth-warning -a "$REDIS_PASSWORD" -h $REDIS_IP -p $REDIS_PORT FLUSHALL ASYNC
$REDIS_CLI --no-auth-warning -a "$REDIS_PASSWORD" -h $REDIS_IP -p $REDIS_PORT BGREWRITEAOF
fi
else
# Cluster（集群）
for redis_node in $redis_nodes;
do
if test ! -z "$redis_node"; then
eval $(echo "$redis_node" | awk -F[\:] '{ printf("redis_node_ip=%s\nredis_node_port=%s\n",$1,$2) }')
if test ! -z "$redis_node_ip" -a ! -z "$redis_node_port"; then
# clear
echo -e "Clearing \033[1;33m${redis_node_ip}:${redis_node_port}\033[m ..."
if test -z "$REDIS_PASSWORD"; then
result=`$REDIS_CLI -h $redis_node_ip -p $redis_node_port FLUSHALL ASYNC`
$REDIS_CLI -h $redis_node_ip -p $redis_node_port BGREWRITEAOF
else
result=`$REDIS_CLI --no-auth-warning -a "$REDIS_PASSWORD" -h $redis_node_ip -p $redis_node_port FLUSHALL ASYNC`
$REDIS_CLI --no-auth-warning -a "$REDIS_PASSWORD" -h $redis_node_ip -p $redis_node_port BGREWRITEAOF
fi
if test ! -z "$result"; then
# SUCCESS
if test "$result" = "OK"; then
echo -e "\033[0;32;32m$result\033[m"
else
echo -e "\033[0;32;31m$result\033[m"
fi
fi
fi
fi
done
fi