Redis 日常運維腳本(增加集羣批量刪除 key 功能)

需求背景

由於應用 bug 造成某些特定前綴的緩存出錯,需要批量清理。

Redis 日常運維腳本 中的 redis-tools.sh僅支持刪除單個 key 和 全部清理,並未實現匹配特定前綴批量刪除的功能。

批量刪除 key

$ redis-cli --scan --pattern "party*" | xargs -L 1 redis-cli del

變更記錄

  • Version 0.0.2 2020/05/16

    • 修正 inputYN 多次回車,導致參數丟失問題
    • 修正 部分描述信息,調整格式等
    • 增加 “批量刪除 key,支持正則表達式” 方法
  • Version 0.0.1 2020/04/26

    • 創建 Redis 集羣情況查詢、key 查詢、key 刪除等功能腳本

選項

sh ${SCRIPT_NAME} [options] <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
    -k<pattern>, --keys=<pattern>        查詢 key,支持正則表達式
    -g<value>, --get=<value>             獲取指定 key 的值
    -d<value>, --del=<value>             刪除指定 key,不支持正則表達式,原因:redis 的 del 命令不支持正則表達式
    -b<pattern>, --bdel=<pattern>        批量刪除 key,支持正則表達式
    -f, --flushall                       刪除所有 key
    --help                               幫助信息
    -v, --version                        版本信息

示例

redis key示例數據格式:
       "party::123"
       "party::456"

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 -k "party::123"
sh redis-tools.sh -k "party*"

4. 獲取指定 key 值
sh redis-tools.sh -g "party::123"

5. 刪除指定 key,不支持正則表達式,原因:redis 的 del 命令不支持正則表達式
sh redis-tools.sh -d "party::123"

6. 批量刪除 key,支持正則表達式
sh redis-tools.sh -b "party::123"
sh redis-tools.sh -b "party*"

7. 刪除所有 key
sh redis-tools.sh -f

腳本

$ cat redis-tools.sh
#!/bin/bash
#================================================================
# HEADER
#================================================================
#    Filename         redis-tools.sh
#    Revision         0.0.2
#    Date             2020/04/26
#    Author           jiangliheng
#    Email            [email protected]
#    Website          https://jiangliheng.github.io/
#    Description      Redis 日常運維腳本
#    Copyright        Copyright (c) jiangliheng
#    License          GNU General Public License
#
#================================================================
#  Version 0.0.2 2020/05/16
#     修正 inputYN 多次回車,導致參數丟失問題
#     修正 部分描述信息,調整格式等
#     增加 “批量刪除 key,支持正則表達式” 方法
#
#  Version 0.0.1 2020/04/26
#     創建 Redis 集羣情況查詢、key 查詢、key 刪除等功能腳本
#
#================================================================
#%名稱(NAME)
#%       ${SCRIPT_NAME} - Redis 日常運維腳本
#%
#%概要(SYNOPSIS)
#%       sh ${SCRIPT_NAME} [options] <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
#%       -k<pattern>, --keys=<pattern>        查詢 key,支持正則表達式
#%       -g<value>, --get=<value>             獲取指定 key 的值
#%       -d<value>, --del=<value>             刪除指定 key,不支持正則表達式,原因:redis 的 del 命令不支持正則表達式
#%       -b<pattern>, --bdel=<pattern>        批量刪除 key,支持正則表達式
#%       -f, --flushall                       刪除所有 key
#%       --help                               幫助信息
#%       -v, --version                        版本信息
#%
#%示例(EXAMPLES)
#%       redis key示例數據格式:
#%              "party::123"
#%              "party::456"
#%
#%       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} -k "party::123"
#%       sh ${SCRIPT_NAME} -k "party*"
#%
#%       4. 獲取指定 key 值
#%       sh ${SCRIPT_NAME} -g "party::123"
#%
#%       5. 刪除指定 key,不支持正則表達式,原因:redis 的 del 命令不支持正則表達式
#%       sh ${SCRIPT_NAME} -d "party::123"
#%
#%       6. 批量刪除 key,支持正則表達式
#%       sh ${SCRIPT_NAME} -b "party::123"
#%       sh ${SCRIPT_NAME} -b "party*"
#%
#%       7. 刪除所有 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.2"

# 默認 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 "\033[36mredis-cli -c -h %s -p %s -a %s %s \"%s\" \033[0m\n\n" "${HOST}" "${PORT}" "${PASSWORD}" "$1" "$2"
  eval redis-cli -c -h "${HOST}" -p "${PORT}" -a "${PASSWORD}" "$1" \""$2"\"
}

# clusterCli 方法
clusterCli() {
  printf "\033[36mredis-cli -c -h %s -p %s -a %s cluster %s \033[0m\n\n" "${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 "\033[36mRedis master nodes: \n%s\n\033[0m" "${masterNodes}"
}

# 清理所有 key
flushallCli() {
  masterNodes

  # 循環清理
  for master in ${masterNodes}
  do
    thost=${master%:*}
    tport=${master#*:}
    printf "\033[36m\nredis-cli -c -h %s -p %s -a %s flushall \033[0m\n" "${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 "\033[36m\nredis-cli -c -h %s -p %s -a %s keys \"%s\" \033[0m\n" "${thost}" "${tport}" "${PASSWORD}" "$1"
    eval redis-cli -c -h "${thost}" -p "${tport}" -a "${PASSWORD}" keys \""$1"\"
  done
}

# 批量刪除相同前綴的 key,支持正則表達式
bdelCli() {
  masterNodes

  # 循環查詢 key
  for master in ${masterNodes}
  do
    thost=${master%:*}
    tport=${master#*:}
    printf "\033[36m\nredis-cli -h %s -p %s -a %s --scan --pattern \"%s\" | xargs -L 1 redis-cli -h %s -p %s -a %s del\033[0m\n" "${thost}" "${tport}" "${PASSWORD}" "$1" "${thost}" "${tport}" "${PASSWORD}"
    eval redis-cli -h "${thost}" -p "${tport}" -a "${PASSWORD}" --scan --pattern \""$1"\" | xargs -L 1 redis-cli -h "${thost}" -p "${tport}" -a "${PASSWORD}" del
  done
}

# 操作確認
inputYN() {
  read -r -p "是否繼續 \"$1\" 操作(y/n):" choose
  case "${choose}" in
    [yY])
      return 0
      ;;

    [nN])
      exit 1
      ;;

    *)
      inputYN "$1"
      ;;
  esac
}

# 判斷參數個數
if [ $# -eq 0 ];
then
  usage
  exit 1
fi

# getopt 命令行參數
if ! ARGS=$(getopt -o vfh:p:a:g:d:c:k:b: --long flushall,help,version,host:,port:,password:,get:,del:,bdel:,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
      ;;

    -b|--bdel)
      # 先顯示匹配項
      keysCli "$2"
      # 確認是否刪除
      inputYN "batch del"
      # 刪除
      bdelCli "$2"
      exit 1
      ;;

    -f|--flushall)
      # 確認是否刪除
      inputYN "flushall"
      # 刪除所有 key
      flushallCli
      exit 1
      ;;

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

    --help)
      usage
      exit 1
      ;;

    --)
      shift
      break
      ;;

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

  esac
done

微信公衆號:daodaotest

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章