123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102 |
- #!/bin/sh -e
- #
- # disk_io: calculate the disk io rate
- #
- # Copyright (C) 2010 Canonical Ltd.
- # Copyright (C) 2011-2014 Dustin Kirkland
- #
- # Authors: Dustin Kirkland <kirkland@byobu.org>
- #
- # This program is free software: you can redistribute it and/or modify
- # it under the terms of the GNU General Public License as published by
- # the Free Software Foundation, version 3 of the License.
- #
- # This program is distributed in the hope that it will be useful,
- # but WITHOUT ANY WARRANTY; without even the implied warranty of
- # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- # GNU General Public License for more details.
- #
- # You should have received a copy of the GNU General Public License
- # along with this program. If not, see <http://www.gnu.org/licenses/>.
- __disk_io_detail() {
- if eval $BYOBU_TEST iostat >/dev/null; then
- iostat -d -m -h
- else
- printf "%s\n" "Please install iostat if you want detailed information on your disk throughput"
- fi
- }
- getdisk() {
- local t=""
- if [ -L "${1}" ]; then
- t=$($BYOBU_READLINK -f "$1")
- else
- t="$1"
- fi
- t="${t##*/}";
- [ -h "/sys/block/$t" ] && _RET="$t" || rtrim "$t" "0-9"
- }
- __disk_io() {
- local part= i=
- # Default to disk providing /, but let users override with MONITORED_DISK
- [ -z "$MONITORED_DISK" ] && mount_point="/" || mount_point="$MONITORED_DISK"
- # By default, we won't bug the user with the display of network traffic
- # below DISK_IO_THRESHOLD in kB/s; override in $BYOBU_CONFIG_DIR/status
- [ -n "$DISK_IO_THRESHOLD" ] || DISK_IO_THRESHOLD=50
- case "$mount_point" in
- /dev/*) part="${mount_point}";;
- *) part=$(awk '$2 == mp { print $1 ; exit(0); }' "mp=$mount_point" /etc/mtab);;
- esac
- [ -e "$part" ] || return
- getdisk "$part"
- local disk=${_RET}
- local t2=$(date +%s) t1=
- for i in "read" "write"; do
- local cache="$BYOBU_RUN_DIR/cache.$BYOBU_BACKEND/disk.$i"
- t1=$(stat -c %Y "$cache") 2>/dev/null || t1=0
- local unit="kB/s"
- local rate=0 x1=0 x2=0 symbol= unit=
- if [ $t2 -le $t1 ]; then
- rate=0
- else
- [ -r "$cache" ] && read x1 < "$cache" || x1=0
- local a1= a2= a3= a4= a5= a6= a7= a8= a9= a10=
- read a1 a2 a3 a4 a5 a6 a7 a8 a9 a10 other < "/sys/block/$disk/stat"
- if [ "$i" = "read" ]; then
- symbol="$ICON_RD"
- [ -n "$a3" ] && x2="$a3" || x2=0
- else
- symbol="$ICON_WR"
- [ -n "$a7" ] && x2="$a7" || x2=0
- fi
- printf "%s" "$x2" > "$cache"
- rate=$((($x2 - $x1) / ($t2 - $t1) * 512 / 1024))
- if [ $rate -lt $DISK_IO_THRESHOLD ]; then
- # Below threshold, don't print
- continue
- elif [ "$rate" -lt 0 ]; then
- rate=0
- elif [ "$rate" -gt 1048576 ]; then
- unit="GB/s"
- fpdiv "$rate" 1048576 0
- rate=${_RET}
- elif [ "$rate" -gt 1024 ]; then
- unit="MB/s"
- fpdiv "$rate" 1024 0
- rate=${_RET}
- else
- unit="kB/s"
- fi
- fi
- if [ -z "$rate" ] || [ "$rate" = "0" ]; then
- rm -f "$BYOBU_RUN_DIR/status.$BYOBU_BACKEND/disk_io"*
- else
- color b M W; printf "%s%s" "$symbol" "$rate"; color -; color M W; printf "%s" "$unit"; color --
- fi
- done
- }
- # vi: syntax=sh ts=4 noexpandtab
|