#!/bin/bash
# info: update PostgreSQL rrd
# options: PERIOD
#
# The function is for updating postgresql rrd database and graphic.


#----------------------------------------------------------#
#                    Variable&Function                     #
#----------------------------------------------------------#

# Argument definition
period=${1-daily}

# Includes
source $VESTA/func/main.sh
source $VESTA/conf/vesta.conf


#----------------------------------------------------------#
#                       Action                             #
#----------------------------------------------------------#

# Switching on time period
case $period in
    daily)   start='-1d'; end='now'; grid='MINUTE:30:HOUR:1:HOUR:4:0:%H:%M';;
    weekly)  start='-7d'; end='now'; grid='HOUR:8:DAY:1:DAY:1:0:%a %d';;
    monthly) start='-1m'; end='now'; grid='WEEK:1:WEEK:1:WEEK:1:0:%b %d';;
    yearly)  start='-1y'; end='now'; grid='MONTH:1:YEAR:1:MONTH:2:2419200:%b';;
    *) exit $E_RRD ;;
esac

# Checking directory
if [ ! -d "$RRD/db" ]; then
    mkdir $RRD/db
fi

# Parsing db hosts
conf="$VESTA/conf/pgsql.conf"
hosts=$(grep HOST $conf |awk '{print $1}' |cut -f 2 -d \')
check_row=$(echo "$hosts" |wc -l)
if [ 0 -eq "$check_row" ]; then
    exit
fi

# Parsing excludes
for exclude in $(echo ${RRD_PGSQL_EXCLUDE//,/ }); do
    hosts=$(echo "$hosts" |grep -vw "$exclude" )
done

for host in $hosts; do
    # Checking database
    if [ ! -e "$RRD/db/pgsql_$host.rrd" ]; then
        # Adding database
        rrdtool create $RRD/db/pgsql_$host.rrd --step $RRD_STEP \
            DS:A:GAUGE:600:U:U \
            DS:T:COUNTER:600:U:U \
            RRA:AVERAGE:0.5:1:600 \
            RRA:AVERAGE:0.5:6:700 \
            RRA:AVERAGE:0.5:24:775 \
            RRA:AVERAGE:0.5:288:797 \
            RRA:MAX:0.5:1:600 \
            RRA:MAX:0.5:6:700 \
            RRA:MAX:0.5:24:775 \
            RRA:MAX:0.5:288:797
    fi

    if [ "$period" = 'daily' ]; then
        # Defining host credentials
        host_str=$(grep "HOST='$host'" $conf)
        for key in $host_str; do
            eval ${key%%=*}=${key#*=}
        done

        export PGPASSWORD="$PASSWORD"
        sql="psql -h $HOST -U $USER -c"

        # Checking empty vars
        if [ -z $HOST ] || [ -z $USER ] || [ -z $PASSWORD ]; then
            echo "Error: config is broken"
            log_event "$E_PARSING" "$ARGUMENTS"
            exit $E_PARSING
        fi

        # Parsing data
        q='SELECT SUM(xact_commit + xact_rollback), SUM(numbackends)
                FROM pg_stat_database;'
        status=$($sql psql -d postgres -c "$q" 2>/dev/null); code="$?"
        if [ '0' -ne "$code" ]; then
            active=0
            slow=0
        else
            active=$(echo "$status"|head -n 3|tail -n 1|awk '{print $3}')
            trans=$(echo "$status"|head -n 3 |tail -n 1|awk '{print $1}')
        fi

        # Updating rrd
        export PGPASSWORD='pgsql'
        rrdtool update $RRD/db/pgsql_$host.rrd N:$active:$trans
    fi

    # Updating rrd graph
    rrdtool graph  $RRD/db/$period-pgsql_$host.png \
        --imgformat PNG \
        --height="150" \
        --width="670" \
        --start "$start" \
        --end "$end" \
        --vertical-label "Queries" \
        --x-grid "$grid" \
        -c "BACK#ffffff" \
        -c "SHADEA#ffffff" \
        -c "SHADEB#ffffff" \
        -c "FONT#555555" \
        -c "CANVAS#302c2d" \
        -c "GRID#666666" \
        -c "MGRID#AAAAAA" \
        -c "FRAME#302c2d" \
        -c "ARROW#FFFFFF" \
        DEF:a=$RRD/db/pgsql_$host.rrd:A:AVERAGE \
        DEF:t=$RRD/db/pgsql_$host.rrd:T:AVERAGE \
        COMMENT:'\r' \
        LINE1:a#fefda0:"Queries     "\
        GPRINT:a:'LAST: Current\:''%8.0lf'  \
        GPRINT:a:'MIN: Min\:''%8.0lf'  \
        GPRINT:a:'MAX: Max\:''%8.0lf\j'  \
        LINE2:t#f57900:"Transactions" \
        GPRINT:t:'LAST:Current\:''%8.0lf' \
        GPRINT:t:'MIN:Min\:''%8.0lf' \
        GPRINT:t:'MAX:Max\:''%8.0lf\j'  &>/dev/null; result=$?

    if [ "$result" -ne 0 ]; then
        exit $E_RRD
    fi

done


#----------------------------------------------------------#
#                       Vesta                              #
#----------------------------------------------------------#

exit