collect-iostat.atop-sar.sh

#!/bin/sh

# Copyright 2015-2020 Rivoreo

# Permission is hereby granted, free of charge, to any person obtaining
# a copy of this software and associated documentation files (the
# "Software"), to deal in the Software without restriction, including
# without limitation the rights to use, copy, modify, merge, publish,
# distribute, sublicense, and/or sell copies of the Software, and to
# permit persons to whom the Software is furnished to do so, subject to
# the following conditions:

# The above copyright notice and this permission notice shall be
# included in all copies or substantial portions of the Software.

# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
# NONINFRINGEMENT. IN NO EVENT SHALL THE COPYRIGHT HOLDERS BE LIABLE
# FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF
# CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

if [ -z "${PROGRAM_DIR+set}" ]; then
	echo "You should not run this script directly" 1>&2
	exit 255
fi

. "$PROGRAM_DIR/disk-name-functions.sh"

multiply_10() {
	case $1 in
		0|0.0*)
			_n=0
			;;
		0.*)
			_n=${1#0.}
			if [ -n "$_n" ]; then
				_n=${_n%${_n#?}}
			else
				_n=0
			fi
			;;
		*)
			_n=${1%%.*}
			if [ $_n = $1 ]; then
				_n=${_n}0
			else
				_d=${1#*.}
				if [ -n "$_d" ]; then
					_n=${_n}${_d%${_d#?}}
				else
					_n=${_n}0
				fi
				unset _d
			fi
			;;
	esac
	echo $_n
	unset _n
}

kilo_to_512() {
	_n=`multiply_10 $1`
	echo $((_n*100/512))
	unset _n
}

export LC_TIME=C
[ "`id -u`" = 0 ] && nice="nice -n 15" || nice=
$nice atopsar -ad "$INTERVAL" | sed -ur -e 1,5d -e "s/^.{9}//" | while read -r line; do
	set -- $line
	if [ "$1" = ----------------- ]; then
		refresh_device_name_map
		continue
	fi
	if [ $# -lt 8 ]; then
		echo "data error: # $# less than 8" 1>&2
		continue
	fi
	if [ "$1" = disk ]; then
		refresh_device_name_map
		continue
	fi
	device="`get_physical_path \"$1\"`" || continue
	blocks_per_read_transfer=`kilo_to_512 $4`
	blocks_per_write_transfer=`kilo_to_512 $6`
	read_rate=$((`multiply_10 $3`*blocks_per_read_transfer/10))
	write_rate=$((`multiply_10 $5`*blocks_per_write_transfer/10))
	if [ -n "$INFLUXDB_BASE_URL" ]; then
		http_post "${INFLUXDB_BASE_URL}write?db=$INFLUXDB_DB_NAME" \
			"iostat,host=$HOSTNAME,device=$device busy=${2%\%},rtps=$3,blocks_per_read_transfer=$blocks_per_read_transfer,read_rate=$read_rate,wtps=$5,blocks_per_write_transfer=$blocks_per_write_transfer,write_rate=$write_rate" 1>&2
	fi
	if [ -n "$CARBON_SERVER" ]; then
		device="`printf %s \"$device\" | sed -r 's/[.:;]/\\-/g'`"
		printf '%s\n%s\n%s\n%s\n%s\n' \
			"iostat.$HOSTNAME_FOR_CARBON.$device.busy ${2%\%} -1" \
			"iostat.$HOSTNAME_FOR_CARBON.$device.rtps $3 -1" \
			"iostat.$HOSTNAME_FOR_CARBON.$device.blocks_per_read_transfer $blocks_per_read_transfer -1" \
			"iostat.$HOSTNAME_FOR_CARBON.$device.read_rate $read_rate -1" \
			"iostat.$HOSTNAME_FOR_CARBON.$device.wtps $5 -1" \
			"iostat.$HOSTNAME_FOR_CARBON.$device.blocks_per_write_transfer $blocks_per_write_transfer -1" \
			"iostat.$HOSTNAME_FOR_CARBON.$device.write_rate $write_rate -1"
	fi
done | send_carbon_lines