#!/bin/sh
#
# $FreeBSD$
#
# Specify gsched_enable="YES" in /etc/rc.conf(.local) to activate scheduling on
# storage devices.  
# 
# gsched_devs is a space separated list of accepted devices.  `ALL' may be used 
# to indicate all storage devices.  
#
# gsched_flags_$dev specifies storage device specific flags (i.e. ``-a rr'' to
# use rr scheduling, see gsched(8) insert command).  

# TODO:
# - add gsched profiles, such as `desktop' for kern.geom.sched.rr tunables
# - add rejected list

# PROVIDE: gsched
# KEYWORD: nojail

. /etc/rc.subr

gsched_enable=${gsched_enable:-NO}

name="gsched"
rcvar=`set_rcvar`
command="/sbin/${name}"
start_cmd="gsched_start"
stop_cmd="gsched_stop"

gsched_filter() {
	local _gsched_devs _devs _devs_recon _kern_disks

	_kern_disks=`sysctl -n kern.disks`

	_devs=$*
	if [ -z "${_devs}" ]; then
		_devs=${_kern_disks}
	fi

	_gsched_devs=${gsched_devs}
	case ${_gsched_devs} in
		[aA][lL][lL])
			_gsched_devs=${_kern_disks}
			;;
	esac

	for _g in ${_devs}; do
		# Filter all devs that are part of gsched_devs
		case " ${_gsched_devs} " in
		    *\ ${_g}\ *)
			# Filter all devs that exist
			case " ${_kern_disks} " in
			    *\ ${_g}\ *)
				# ${_g} is part of ${gsched_dev} and it exists
				_devs_recon="${_devs_recon} ${_g}"
				;;
			esac
			;;
		esac
	done

	echo ${_devs_recon}
}

gsched_start()
{
	local _devs _g _gsched_flags

	# Make sure only accepted (and existing) devices are used
	#
	_devs=`gsched_filter $*`

	echo -n "Starting gsched devices:"

	for _g in ${_devs}; do
		echo -n " $_g"
		eval _gsched_flags=\$gsched_flags_${_g}
		${command} insert ${_gsched_flags} ${_g}
	done

	echo "."
}

gsched_stop() {
	local _devs _g

	# Make sure only accepted (and existing) devices are used
	#
	_devs=`gsched_filter $*`

	echo -n "Stopping gsched devices:"

	for _g in ${_devs}; do
		echo -n " $_g"
		${command} destroy ${_g}.sched.
	done

	echo "."
}

load_rc_config $name
run_rc_command $*