ports/sysutils/cbsd/files/cbsd-statsd-bhyve.in

131 lines
2.8 KiB
Bash

#!/bin/sh
#
# PROVIDE: cbsd_statsd_bhyve
# REQUIRE: LOGIN FILESYSTEMS cbsdd
# KEYWORD: shutdown
#
# cbsd_statsd_bhyve_enable="YES"
#
. /etc/rc.subr
name=cbsd_statsd_bhyve
rcvar=cbsd_statsd_bhyve_enable
load_rc_config $name
# Set defaults
: ${cbsd_statsd_bhyve_enable:="NO"}
export workdir="${cbsd_workdir}"
globalconf=${cbsd_globalconf:-"%%PREFIX%%/cbsd/cbsd.conf"}
start_cmd=${name}_start
stop_cmd=${name}_stop
restart_cmd=${name}_restart
extra_commands="restart"
command="%%PREFIX%%/cbsd/tools/racct-bhyve-statsd"
pidfile="${cbsd_workdir}/var/run/${name}.pid"
daemon_pidfile="${cbsd_workdir}/var/run/${name}-daemon.pid"
daemon_logdir="/var/log/${name}"
workdir_normalize=$( echo ${cbsd_workdir} | /usr/bin/tr "/" "_" )
daemon_logfile="${daemon_logdir}/cbsd-statsd-bhyve${workdir_normalize}.log"
command_args="&"
if [ ! -f ${globalconf} ]; then
echo "cbsd: no such ${globalconf}";
exit 1
fi
if [ ! -f ${subr} ]; then
echo "cbsd: no such ${subr}";
exit 1
fi
. ${globalconf}
. ${subrdir}/nc.subr
cbsd_statsd_bhyve_stop()
{
local _pids
if [ -f "${daemon_pidfile}" ]; then
_pids=$( pgrep -F ${daemon_pidfile} 2>&1 )
_err=$?
/bin/rm -f ${daemon_pidfile}
if [ ${_err} -eq 0 ]; then
kill -9 ${_pids}
else
echo "pgrep: ${_pids}"
fi
fi
if [ -f "${pidfile}" ]; then
_pids=$( pgrep -F ${pidfile} 2>&1 )
_err=$?
/bin/rm -f ${pidfile}
if [ ${_err} -eq 0 ]; then
kill -9 ${_pids}
else
echo "pgrep: ${_pids}"
fi
fi
return 0
}
cbsd_statsd_bhyve_start()
{
local args= racct_enabled=
# RACCT enabled?
racct_enabled=$( sysctl -qn kern.racct.enable )
[ $? -ne 0 ] && err 1 "cbsd-statsd-bhyve: RACCT not enabled"
[ "${racct_enabled}" != "1" ] && err 1 "cbsd-statsd-bhyve: RACCT not enabled"
readconf racct-bhyve-statsd.conf
[ -n "${log_file}" ] && args="${args} --log_file=${log_file}"
[ -n "${log_level}" ] && args="${args} --log_level=${log_level}"
[ -n "${loop_interval}" ] && args="${args} --loop_interval=${loop_interval}"
[ -n "${save_loop_count}" ] && args="${args} --save_loop_count=${save_loop_count}"
case "${prometheus_exporter_enable}" in
[Yy][Ee][Ss])
args="${args} --prometheus_exporter=1"
;;
[No][Oo])
args="${args} --prometheus_exporter=0"
;;
esac
case "${beanstald_enable}" in
[Yy][Ee][Ss])
args="${args} --save_beanstalkd=1"
;;
[No][Oo])
args="${args} --save_beanstalkd=0"
;;
esac
case "${sqlite3_enable}" in
[Yy][Ee][Ss])
args="${args} --save_sqlite=1"
;;
[No][Oo])
args="${args} --save_sqlite=0"
;;
esac
[ ! -d ${daemon_logdir} ] && mkdir -p ${daemon_logdir}
touch ${daemon_logfile}
/usr/sbin/daemon -f -R5 -P ${daemon_pidfile} -o ${daemon_logfile} /usr/bin/env workdir=${cbsd_workdir} ${command} ${args}
}
cbsd_statsd_bhyve_restart()
{
cbsd_statsd_bhyve_stop
cbsd_statsd_bhyve_start
}
run_rc_command "$1"