Redis 日常运维脚本

需求背景

生产 Redis 日常运维需要。

前提

添加 Redis 可执行命令到系统环境变量中。

$ cat redis-profile.sh
export PATH=/usr/local/redis-5.0.7/src/:$PATH

# 添加到系统环境变量
$ cp redis-profile.sh /etc/profile.d/

# 实时生效
$ source /etc/profile.d/redis-profile.sh

# 验证
$ redis-cli -v

简介

Filename         redis-tools.sh
Revision         0.0.1
Date             2020/04/26
Author           jiangliheng
Email            jiang_liheng@163.com
Website          https://jiangliheng.github.io/
Description      Redis 日常运维脚本
Copyright        Copyright (c) jiangliheng
License          GNU General Public License

概要

redis-tools.sh [option] <value> ...

选项

-h<value>, --host=<value>            Redis IP，可设置默认值参数：HOST
-p<value>, --port=<value>            Redis 端口，可设置默认值参数：PORT
-a<value>, --password=<value>        Redis 密码，可设置默认值参数：PASSWORD
-c<value>, --cluster=<value>         集群相关命令，如：nodes, info
-g<value>, --get=<value>             获取指定 key 的值
-d<value>, --del=<value>             删除指定 key
-k<value>, --keys=<value>            查询 key
-f, --flushall                       删除所有 key
--help                               帮助信息
-v, --version                        版本信息

示例

1. 查询集群信息，使用默认参数
sh redis-tools.sh -c info

2. 查询集群节点
sh redis-tools.sh -h 127.0.0.1 -p 8001 -a password -c nodes

3. 获取指定 key 值
sh redis-tools.sh -g key

4. 删除指定 key
sh redis-tools.sh -d key

5. 查询 key
sh redis-tools.sh -k key

6. 删除所有 key
sh redis-tools.sh -f

redis-tools.sh

$ cat redis-tools.sh
#!/bin/bash
#================================================================
# HEADER
#================================================================
#    Filename         redis-tools.sh
#    Revision         0.0.1
#    Date             2020/04/26
#    Author           jiangliheng
#    Email            jiang_liheng@163.com
#    Website          https://jiangliheng.github.io/
#    Description      Redis 日常运维脚本
#    Copyright        Copyright (c) jiangliheng
#    License          GNU General Public License
#
#================================================================
#  Version 0.0.1
#     Redis 集群情况查询、key 查询、key 删除等
#
#================================================================
#%名称(NAME)
#%       ${SCRIPT_NAME} - Redis 日常运维脚本
#%
#%概要(SYNOPSIS)
#%       sh ${SCRIPT_NAME} [option] <value> ...
#%
#%描述(DESCRIPTION)
#%       Redis 日常运维脚本
#%
#%选项(OPTIONS)
#%       -h<value>, --host=<value>            Redis IP，可设置默认值参数：HOST
#%       -p<value>, --port=<value>            Redis 端口，可设置默认值参数：PORT
#%       -a<value>, --password=<value>        Redis 密码，可设置默认值参数：PASSWORD
#%       -c<value>, --cluster=<value>         集群相关命令，如：nodes, info
#%       -g<value>, --get=<value>             获取指定 key 的值
#%       -d<value>, --del=<value>             删除指定 key
#%       -k<value>, --keys=<value>            查询 key
#%       -f, --flushall                       删除所有 key
#%       --help                               帮助信息
#%       -v, --version                        版本信息
#%
#%示例(EXAMPLES)
#%       1. 查询集群信息，使用默认参数
#%       sh ${SCRIPT_NAME} -c info
#%
#%       2. 查询集群节点
#%       sh ${SCRIPT_NAME} -h 127.0.0.1 -p 8001 -a password -c nodes
#%
#%       3. 获取指定 key 值
#%       sh ${SCRIPT_NAME} -g key
#%
#%       4. 删除指定 key
#%       sh ${SCRIPT_NAME} -d key
#%
#%       5. 查询 key
#%       sh ${SCRIPT_NAME} -k key
#%
#%       6. 删除所有 key
#%       sh ${SCRIPT_NAME} -f
#%
#================================================================
# END_OF_HEADER
#================================================================

# header 总行数
SCRIPT_HEADSIZE=$(head -200 "${0}" |grep -n "^# END_OF_HEADER" | cut -f1 -d:)
# 脚本名称
SCRIPT_NAME="$(basename "${0}")"
# 版本
VERSION="0.0.1"

# 默认 host
HOST=127.0.0.1
# 默认 port
PORT=8001
# 默认 password
PASSWORD=password

# usage
usage() {
  head -"${SCRIPT_HEADSIZE:-99}" "${0}" 
  | grep -e "^#%" 
  | sed -e "s/^#%//g" -e "s/${SCRIPT_NAME}/${SCRIPT_NAME}/g" -e "s/${VERSION}/${VERSION}/g"
}

# redis-cli 方法
cli() {
  printf "33[36mredis-cli -c -h %s -p %s -a %s %s "%s" 33[0m

" "${HOST}" "${PORT}" "${PASSWORD}" "$1" "$2"
  eval redis-cli -c -h "${HOST}" -p "${PORT}" -a "${PASSWORD}" "$1" ""$2""
}

# clusterCli 方法
clusterCli() {
  printf "33[36mredis-cli -c -h %s -p %s -a %s cluster %s 33[0m

" "${HOST}" "${PORT}" "${PASSWORD}" "$1"
  eval redis-cli -c -h "${HOST}" -p "${PORT}" -a "${PASSWORD}" cluster "$1"
}

# 查询 master 节点
masterNodes() {
  masterNodes=$(clusterCli nodes | awk '{if($3=="master" || $3=="myself,master") print $2}' | awk -F@ '{print $1}')
  printf "33[36mRedis master nodes: 
%s
33[0m" "${masterNodes}"
}

# 清理所有 key
flushallCli() {
  masterNodes

  # 循环清理
  for master in ${masterNodes}
  do
    thost=${master%:*}
    tport=${master#*:}
    printf "33[36m
redis-cli -c -h %s -p %s -a %s flushall 33[0m
" "${thost}" "${tport}" "${PASSWORD}"
    eval redis-cli -c -h "${thost}" -p "${tport}" -a "${PASSWORD}" flushall
  done
}

# 查询 key
keysCli() {
  masterNodes

  # 循环查询 key
  for master in ${masterNodes}
  do
    thost=${master%:*}
    tport=${master#*:}
    printf "33[36m
redis-cli -c -h %s -p %s -a %s keys "%s" 33[0m
" "${thost}" "${tport}" "${PASSWORD}" "$1"
    eval redis-cli -c -h "${thost}" -p "${tport}" -a "${PASSWORD}" keys ""$1""
  done
}

# 操作确认
inputYN() {
  read -r -p "是否继续 "$1" 操作(y/n)：" choose
  case "${choose}" in
    [yY])
      return 0
      ;;

    [nN])
      exit 1
      ;;

    *)
      inputYN
      ;;
  esac
}

# 判断参数个数
if [ $# -eq 0 ];
then
  usage
  exit 1
fi

# getopt 命令行参数
if ! ARGS=$(getopt -o vfh:p:a:g:d:c:k: --long flushall,help,version,host:,port:,password:,get:,del:,password:,cluster:,keys: -n "${SCRIPT_NAME}" -- "$@")
then
  # 无效选项，则退出
  exit 1
fi

# 命令行参数格式化
eval set -- "${ARGS}"

while [ -n "$1" ]
do
  case "$1" in
    -a|--password)
      PASSWORD=$2
      shift 2
      ;;

    -h|--host)
      HOST=$2
      shift 2
      ;;

    -p|--port)
      PORT="$2"
      shift 2
      ;;

    -c|--cluster)
      clusterCli "$2"
      exit 1
      ;;

    -g|--get)
      cli get "$2"
      exit 1
      ;;

    -d|--del)
      # 先显示内容
      cli get "$2"
      # 确认是否删除
      inputYN "del"
      # 删除
      cli del "$2"
      exit 1
      ;;

    -k|--keys)
      keysCli "$2"
      exit 1
      ;;

    -f|--flushall)
      # 确认是否删除
      inputYN "flushall"
      # 删除所有 key
      flushallCli
      exit 1
      ;;

    -v|--version)
      printf "%s version %s
" "${SCRIPT_NAME}" "${VERSION}"
      exit 1
      ;;

    --help)
      usage
      exit 1
      ;;

    --)
      shift
      break
      ;;

    *)
      printf "%s is not an option!" "$1"
      exit 1
      ;;

  esac
done

微信公众号：daodaotest