master/root/bin/monitor_cmd

   1 #!/bin/bash
   2
   3 # default values
   4 PROGNAME=$(basename $0)
   5 FULLPROG=$0
   6
   7 # usage
   8 function usage () {
   9     echo "usage: $PROGNAME [-h] [-r] cmd"
  10     echo " -h: help message"
  11     echo " -s: set stats"
  12     echo " -r: reset stats"
  13     echo "cmd: shell command"
  14 }
  15
  16 # reset
  17 function reset () {
  18     echo "reseting stats"
  19     [ -f $resource ] && rm $resource
  20 }
  21
  22 # variables
  23 set=
  24 cmd=
  25 resource=$HOME/.$PROGNAME
  26
  27 # parameter processing
  28 while [ $# -gt 0 ]; do
  29     if [ "$cmd" ]; then
  30         cmd="$cmd $1"
  31     else
  32         case "$1" in
  33         -h) usage; exit 0;;
  34         -s) set="yes";;
  35         -r) reset; exit 0;;
  36         *) cmd="$1";;
  37         esac
  38     fi
  39     shift
  40 done
  41
  42 # check arguments
  43 [ "$cmd" ] || { echo "no command specified"; exit 1; }
  44 [ "$(which ${cmd/ *})" ] || { echo "command not found '$cmd'"; exit 1; }
  45
  46 # read old values
  47 [ -f $resource ] && source $resource
  48
  49 # main
  50 var=$(echo _$cmd | sed 's/[^A-Za-z0-9]/_/g;s/__*/_/g')
  51 newvar=($(eval $cmd | awk '{ for (i = 1; i <= NF; i++) if ($(i) ~ /^[-+0-9][0-9]*$/) printf " %s", $(i) }'))
  52 eval oldvar=(\${$var[@]})
  53
  54 # compute diff
  55 for ((i=0; i<${#newvar[@]}; i++)); do
  56     [ "${oldvar[$i]}" ] && v[$i]=$((newvar[i] - oldvar[i])) || v[$i]=${newvar[$i]}
  57 done
  58
  59 # ouput
  60 if [ "$set" = "yes" ]; then
  61     echo "$var=(${newvar[@]})" >>$resource
  62     awk -F= '{ a[$1]=$2 } END { for (i in a) print i "=" a[i] }' $resource > $resource.n
  63     mv -f $resource.n $resource
  64 else
  65     str=""
  66     for ((i=0; i<${#v[@]}; i++)); do
  67         str="$str; v[$i]=${v[$i]}"
  68     done
  69     eval $cmd | gawk 'BEGIN { k=0 '"$str"' } { n = 1; for (i=1; i<=NF; i++) if ($(i) ~ /^[-+0-9][0-9]*$/) { if ((i == 1) && ! ($0 ~ /^ /)) $(i) = v[k]; else $0 = gensub(/ [-+0-9][0-9]*/, " " v[k], n); k++; n++ } print }'
  70 fi