vesta/bin/v_upd_sys_rrd_net

#!/bin/bash
# info: updating network rrd

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

# Argument defenition
rrd_start=${1--1d}
rrd_end=${2-now}
rrd_grid=${3-MINUTE:30:HOUR:1:HOUR:4:0:%H:%M}

# Importing variables
source $VESTA/conf/vars.conf
source $V_CONF/vesta.conf


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

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

# Parsing network interfaces
ifaces=$(cat /proc/net/dev |grep : |cut -f 1 -d : |sed -e "s/ //g")

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

for iface in $ifaces; do
    # Checking database
    if [ ! -e "$V_RRD/net/$iface.rrd" ]; then
        # Adding database
        rrdtool create $V_RRD/net/$iface.rrd --step $V_RRD_STEP \
            DS:RX:COUNTER:600:U:U \
            DS:TX: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

    # Parsing device stats
    if [ -z "$1" ]; then
        raw_iface=$(grep "$iface:" /proc/net/dev |sed -e "s/:/ /")
        rx=$(echo "$raw_iface" |awk '{print $2}')
        tx=$(echo "$raw_iface" |awk '{print $10}')

        # Updating rrd database
        rrdtool update $V_RRD/net/$iface.rrd N:$rx:$tx
    fi

    # Updating rrd graph
    rrdtool graph $V_RRD/net/$iface.png \
        --imgformat PNG \
        --height="120" \
        --width="440" \
        --start "$rrd_start" \
        --end "$rrd_end" \
        --title "Bandwidth Usage $iface" \
        --vertical-label "KBytes" \
        --x-grid "$rrd_grid" \
        -c "BACK#484439" \
        -c "SHADEA#484439" \
        -c "SHADEB#484439" \
        -c "FONT#DDDDDD" \
        -c "CANVAS#202020" \
        -c "GRID#666666" \
        -c "MGRID#AAAAAA" \
        -c "FRAME#202020" \
        -c "ARROW#FFFFFF" \
        DEF:inoctets=$V_RRD/net/$iface.rrd:RX:AVERAGE \
        DEF:outoctets=$V_RRD/net/$iface.rrd:TX:AVERAGE \
        "CDEF:in=inoctets,8,*" \
        "CDEF:out=outoctets,8,*" \
        COMMENT:'\r' \
        AREA:in#C8EA2E:"Input (rx) "\
        GPRINT:in:'LAST: Current\:''%8.0lf'  \
        GPRINT:in:'MIN: Min\:''%8.0lf'  \
        GPRINT:in:'MAX: Max\:''%8.0lf\j'  \
        LINE1:out#1c74cd:"Output (tx)" \
        GPRINT:out:'LAST:Current\:''%8.0lf' \
        GPRINT:out:'MIN:Min\:''%8.0lf' \
        GPRINT:out:'MAX:Max\:''%8.0lf\j'  >/dev/null 2>/dev/null; result=$?

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

done


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

exit