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            [email protected]
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            [email protected]
#    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 "\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
}

# 操作確認
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\n" "${SCRIPT_NAME}" "${VERSION}"
      exit 1
      ;;

    --help)
      usage
      exit 1
      ;;

    --)
      shift
      break
      ;;

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

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