美文网首页
Redis 日常运维脚本

Redis 日常运维脚本

作者: 叨叨软件测试 | 来源:发表于2020-04-28 10:16 被阅读0次

    需求背景

    生产 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 "\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
    

    微信公众号:daodaotest

    相关文章

      网友评论

          本文标题:Redis 日常运维脚本

          本文链接:https://www.haomeiwen.com/subject/iwomwhtx.html