zabbix监控redis服务

#!/bin/bash
ARGS=("$@")
DISCOVERY_TYPE=$1
REDIS_CLI_DEFAULT_PATH="usr/bin/redis-cli"
STBDBUF_DEFAULT_PATH="/usr/bin/stdbuf"
# USE FIRST ARGUMENT TO UNDERSTAND WHICH DISCOVERY TO PERFORM
shift
IFS=$‘\n‘
PASSWORDS=( "$@" )
LIST=$(ps -eo user,args | grep -v grep | grep redis-server | tr -s [:blank:] ":")

if [[ " ${ARGS[@]} " =~ " debug " ]]; then
    set -x
else
    set -e  # RUDIMENTARY ERROR MECHANISM
fi

# REQUIRED UTILS TO BE ABLE TO RUN
if [ -e /tmp/redis-cli ]; then
    REDIS_CLI=$(cat /tmp/redis-cli)
else
    REDIS_CLI=$(locate redis-cli | head -n 1)
    if [ "$REDIS_CLI" = "" ]; then
        if [ -e $REDIS_CLI_DEFAULT_PATH ]; then
            REDIS_CLI_FILE=$(echo $REDIS_CLI_DEFAULT_PATH > /tmp/redis-cli)
        else
            echo "REDIS-CLI not found ...."
            exit 1
        fi
    else
        REDIS_CLI_FILE=$(echo $REDIS_CLI > /tmp/redis-cli)
    fi
fi

if [ -a /tmp/stdbuf ]; then
    STDBUF=$(cat /tmp/stdbuf)
else
    STDBUF=$(locate stdbuf | head -n 1)
    if [ "$STDBUF" = "" ]; then
        if [ -e $STBDBUF_DEFAULT_PATH ]; then
            STDBUF_FILE=$(echo $STBDBUF_DEFAULT_PATH > /tmp/stdbuf)
        else
            echo "STDBUF-CLI not found..."
            exit 1
        fi
    else
        STDBUF_FILE=$(echo $STDBUF > /tmp/stdbuf)
    fi
fi

if [ "$DISCOVERY_TYPE" != "general" ] && [ "$DISCOVERY_TYPE" != "stats" ] && [ "$DISCOVERY_TYPE" != "replication" ]; then
    echo "USAGE: ./zbx_redis_discovery.sh where"
    echo "general - argument generate report with discovered instances"
    echo "stats - generates report for avalable commands"
    echo "replication - generates report for avalable slaves"
    exit 1
fi

# PROBE DISCOVERED REDIS INSTACES - TO GET INSTANCE NAME#
discover_redis_instance() {
    HOST=$1
    PORT=$2
    PASSWORD=$3

    ALIVE=$($REDIS_CLI -h $HOST -p $PORT -a "$PASSWORD" ping)

    if [[ $ALIVE != "PONG" ]]; then
        return 1
    else
        INSTANCE=$($REDIS_CLI -h $HOST -p $PORT -a "$PASSWORD" info | grep config_file | cut -d ":" -f2 | sed ‘s/.conf//g‘ | rev | cut -d "/" -f1 | rev | tr -d [:space:] | tr [:lower:] [:upper:])
        # WHEN UNABLE TO IDENTIFY INSTANCE NAME BASED ON CONFIG
        if [ "$INSTANCE" = "" ]; then
            INSTANCE=$(echo "$HOST:$PORT")
        fi
        INSTANCE_RDB_PATH=$($REDIS_CLI -h $HOST -p $PORT -a "$PASSWORD" config get "dir" | cut -d " " -f2 | sed -n 2p)
        INSTANCE_RDB_FILE=$($REDIS_CLI -h $HOST -p $PORT -a "$PASSWORD" config get "dbfilename" | cut -d " " -f2 | sed -n 2p)
    fi

    echo $INSTANCE
}

# PROBE DISCOVERED REDIS INSTACES - TO GET RDB DATABASE#
discover_redis_rdb_database() {
    HOST=$1
    PORT=$2
    PASSWORD=$3

    ALIVE=$($REDIS_CLI -h $HOST -p $PORT -a "$PASSWORD" ping)

    if [[ $ALIVE != "PONG" ]]; then
        return 1
    else
        INSTANCE_RDB_PATH=$($REDIS_CLI -h $HOST -p $PORT -a "$PASSWORD" config get "dir" | cut -d " " -f2 | sed -n 2p)
        INSTANCE_RDB_FILE=$($REDIS_CLI -h $HOST -p $PORT -a "$PASSWORD" config get "dbfilename" | cut -d " " -f2 | sed -n 2p)
    fi

    echo $INSTANCE_RDB_PATH/$INSTANCE_RDB_FILE
}

discover_redis_avalable_commands() {
    HOST=$1
    PORT=$2
    PASSWORD=$3

    ALIVE=$($REDIS_CLI -h $HOST -p $PORT -a "$PASSWORD" ping)

    if [[ $ALIVE != "PONG" ]]; then
        return 1
    else
        REDIS_COMMANDS=$($REDIS_CLI -h $HOST -p $PORT -a "$PASSWORD" info all | grep cmdstat | cut -d":" -f1)
    fi

    ( IFS=$‘\n‘; echo "${REDIS_COMMANDS[*]}" )
}

discover_redis_avalable_slaves() {
    HOST=$1
    PORT=$2
    PASSWORD=$3

    ALIVE=$($REDIS_CLI -h $HOST -p $PORT -a "$PASSWORD" ping)

    if [[ $ALIVE != "PONG" ]]; then
        return 1
    else
        REDIS_SLAVES=$($REDIS_CLI -h $HOST -p $PORT -a "$PASSWORD" info all | grep ^slave | cut -d ":" -f1 | grep [0-1024])
    fi

    ( IFS=$‘\n‘; echo "${REDIS_SLAVES[*]}" )
}

# GENERATE ZABBIX DISCOVERY JSON REPONSE #
generate_general_discovery_json() {
    HOST=$1
    PORT=$2
    INSTANCE=$3
    RDB_PATH=$4

    echo -n ‘{‘
    echo -n ‘"{#HOST}":"‘$HOST‘",‘
    echo -n ‘"{#PORT}":"‘$PORT‘",‘
    echo -n ‘"{#INSTANCE}":"‘$INSTANCE‘",‘
    echo -n ‘"{#RDB_PATH}":"‘$RDB_PATH‘"‘
    echo -n ‘},‘
}

# GENERATE ZABBIX DISCOVERY JSON REPONSE #
generate_commands_discovery_json() {
    HOST=$1
    PORT=$2
    COMMAND=$3
    INSTANCE=$4

    echo -n ‘{‘
    echo -n ‘"{#HOST}":"‘$HOST‘",‘
    echo -n ‘"{#PORT}":"‘$PORT‘",‘
    echo -n ‘"{#COMMAND}":"‘$COMMAND‘",‘
    echo -n ‘"{#INSTANCE}":"‘$INSTANCE‘"‘
    echo -n ‘},‘
}

# GENERATE ZABBIX DISCOVERY JSON REPONSE #
generate_replication_discovery_json() {
    HOST=$1
    PORT=$2
    SLAVE=$3
    INSTANCE=$4

    echo -n ‘{‘
    echo -n ‘"{#HOST}":"‘$HOST‘",‘
    echo -n ‘"{#PORT}":"‘$PORT‘",‘
    echo -n ‘"{#SLAVE}":"‘$SLAVE‘",‘
    echo -n ‘"{#INSTANCE}":"‘$INSTANCE‘"‘
    echo -n ‘},‘
}


# GENERATE ALL REPORTS REQUIRED FOR REDIS MONITORING #
generate_redis_stats_report() {
    HOST=$1
    PORT=$2
    PASSWORD=$3

    REDIS_REPORT=$(stdbuf -oL $REDIS_CLI -h $HOST -p $PORT -a "$PASSWORD" info all &> /tmp/redis-$HOST-$PORT)
    REDIS_SLOWLOG_LEN=$(stdbuf -oL $REDIS_CLI -h $HOST -p $PORT -a "$PASSWORD" slowlog len | cut -d " " -f2 &> /tmp/redis-$HOST-$PORT-slowlog-len; $REDIS_CLI -h $HOST -p $PORT -a $PASSWORD slowlog reset > /dev/null  )
    REDIS_SLOWLOG_RAW=$(stdbuf -oL $REDIS_CLI -h $HOST -p $PORT -a "$PASSWORD" slowlog get &> /tmp/redis-$HOST-$PORT-slowlog-raw)
    REDIS_MAX_CLIENTS=$(stdbuf -oL $REDIS_CLI -h $HOST -p $PORT -a "$PASSWORD" config get *"maxclients"* | cut -d " " -f2 | sed -n 2p &> /tmp/redis-$HOST-$PORT-maxclients)
}

# MAIN LOOP #

echo -n ‘{"data":[‘
for s in $LIST; do
    HOST=$(echo $s | sed ‘s/*/127.0.0.1/g‘ | cut -d":" -f3)
    PORT=$(echo $s | sed ‘s/*/127.0.0.1/g‘ | cut -d":" -f4)

    # TRY PASSWORD PER EACH DISCOVERED INSTANCE
    if [[ ${#PASSWORDS[@]} -ne 0 ]]; then
        for (( i=0; i<${#PASSWORDS[@]}; i++ ));
        do
            PASSWORD=${PASSWORDS[$i]}
            INSTANCE=$(discover_redis_instance $HOST $PORT $PASSWORD)
            RDB_PATH=$(discover_redis_rdb_database $HOST $PORT $PASSWORD)
            COMMANDS=$(discover_redis_avalable_commands $HOST $PORT $PASSWORD)
            SLAVES=$(discover_redis_avalable_slaves $HOST $PORT $PASSWORD)

            if [[ -n $INSTANCE ]]; then

                # DECIDE WHICH REPORT TO GENERATE FOR DISCOVERY
                if [[ $DISCOVERY_TYPE == "general" ]]; then
                    generate_redis_stats_report $HOST $PORT $PASSWORD
                    generate_general_discovery_json $HOST $PORT $INSTANCE $RDB_PATH
                elif [[ $DISCOVERY_TYPE == "stats" ]]; then
                    for COMMAND in ${COMMANDS}; do
                        generate_commands_discovery_json $HOST $PORT $COMMAND $INSTANCE
                    done
                elif [[ $DISCOVERY_TYPE == "replication" ]]; then
                    for SLAVE in ${SLAVES}; do
                        generate_replication_discovery_json $HOST $PORT $SLAVE $INSTANCE
                    done
                fi
            fi
        done
    else
        INSTANCE=$(discover_redis_instance $HOST $PORT "")
        RDB_PATH=$(discover_redis_rdb_database $HOST $PORT "")
        COMMANDS=$(discover_redis_avalable_commands $HOST $PORT "")
        SLAVES=$(discover_redis_avalable_slaves $HOST $PORT "")

        if [[ -n $INSTANCE ]]; then

            # DECIDE WHICH REPORT TO GENERATE FOR DISCOVERY
            if [[ $DISCOVERY_TYPE == "general" ]]; then
                generate_redis_stats_report $HOST $PORT ""
                generate_general_discovery_json $HOST $PORT $INSTANCE $RDB_PATH
            elif [[ $DISCOVERY_TYPE == "stats" ]]; then
                for COMMAND in ${COMMANDS}; do
                    generate_commands_discovery_json $HOST $PORT $COMMAND $INSTANCE
                done
            elif [[ $DISCOVERY_TYPE == "replication" ]]; then
                for SLAVE in ${SLAVES}; do
                    generate_replication_discovery_json $HOST $PORT $SLAVE $INSTANCE
                done
            fi
        fi
    fi
    unset
done | sed -e ‘s:\},$:\}:‘
echo -n ‘]}‘
echo ‘‘
unset IFS

# DISCOVERY (@PARAMS = #1 PASSWORDS LIST)
UserParameter=redis.discovery[*], /etc/zabbix/custom/zbx_redis_discovery.sh $1 $2
# METRICS (@PARAMS = #1 - METRIC, #2 - REDIS HOST, #3 - REDIS PORT)
# EXAMPLE: master_last_io_seconds_ago:7
UserParameter=redis.stat.get[*],output=`cat /tmp/redis-$2-$3 | grep "$1:" | cut -d":" -f2`; if [ -z $output ]; then echo 0; else echo $output; fi
# METRICS (@PARAMS = #1 - METRIC, #2 - METRIC SUBKEY, #3 - REDIS HOST, #4 - REDIS PORT)
# EXAMPLE: cmdstat_ping:calls=142434,usec=210078,usec_per_call=1.47
UserParameter=redis.stat.mget[*],output=`cat /tmp/redis-$3-$4 | grep "$1:" | grep -oP ‘(?<=\b$2=)[^,]+‘`; if [ -z $output ]; then echo 0; else echo $output; fi
# METRICS (@PARAMS = #1 - REDIS HOST, #2 - REDIS PORT)
UserParameter=redis.slowlog.count[*],output=`cat /tmp/redis-$1-$2-slowlog-len`; if [ -z $output ]; then echo 0; else echo $output; fi
UserParameter=redis.slowlog.get[*],output=`cat /tmp/redis-$1-$2-slowlog-raw`; if [[ -z $output ]]; then echo 0; else echo $output; fi
UserParameter=redis.maxclients[*],output=`cat /tmp/redis-$1-$2-maxclients`; if [ -z $output ]; then echo 0; else echo $output; fi