diff options
Diffstat (limited to 'utils/collectd/files/collectd.init')
| -rw-r--r-- | utils/collectd/files/collectd.init | 202 |
1 files changed, 200 insertions, 2 deletions
diff --git a/utils/collectd/files/collectd.init b/utils/collectd/files/collectd.init index 668858b90..a0029267a 100644 --- a/utils/collectd/files/collectd.init +++ b/utils/collectd/files/collectd.init @@ -1,17 +1,215 @@ #!/bin/sh /etc/rc.common # Copyright (C) 2006-2016 OpenWrt.org +. "/usr/share/libubox/jshn.sh" + START=80 STOP=10 USE_PROCD=1 +COLLECTD_CONF="/tmp/collectd.conf" +LOG="logger -t collectd[$$] -p" NICEPRIO=5 +CONFIG_STRING="" + +process_exec() { + printf "<Plugin exec>\n" >> "$COLLECTD_CONF" + config_foreach process_exec_sections exec_input "Exec" + config_foreach process_exec_sections exec_notify "NotificationExec" + printf "</Plugin>\n\n" >> "$COLLECTD_CONF" +} + +process_exec_sections() { + local cfg="$1" + local section="$2" + + local cmdline cmduser cmdgroup + + config_get cmdline "$cfg" cmdline + [ -z "$cmdline" ] && { + $LOG notice "No cmdline option in config $cfg defined" + return 0 + } + + config_get cmduser "$cfg" cmduser + [ -z "$cmduser" ] && { + $LOG notice "No cmduser option in config $cfg defined" + return 0 + } + + config_get cmdgroup "$cfg" cmdgroup + if [ -z "$cmdgroup" ]; then + printf "\\t%s \"%s\" \"%s\"\n" "${section}" "${cmduser}" "${cmdline}" >> "$COLLECTD_CONF" + else + printf "\\t%s \"%s:%s\" \"%s\"\n" "${section}" "${cmduser}" "${cmdgroup}" "${cmdline}" >> "$COLLECTD_CONF" + fi +} + +CONFIG_LIST="" +add_list_option() { + local value="$1" + local option="$2" + local indent="$3" + + CONFIG_LIST="${CONFIG_LIST}${indent}${option} \"$value\"\n" +} + +process_generic() { + local cfg="$1" + local indent="$2" + local json="$3" + + local config="" + + json_init + json_load_file "$json" + + json_select string 1>/dev/null 2>&1 + if [ $? -eq 0 ]; then + json_get_keys keys + for key in ${keys}; do + json_get_var option "$key" + config_get value "$cfg" "$option" "" + [ -z "$value" ] || { + config="${config}${indent}${option} \"${value}\"\n" + } + done + json_select .. + fi + + json_select bool 1>/dev/null 2>&1 + if [ $? -eq 0 ]; then + json_get_keys keys + for key in ${keys}; do + json_get_var option "$key" + config_get_bool value "$cfg" "$option" + if [ "$value" = "0" ]; then + config="${config}${indent}${option} false\n" + fi + + if [ "$value" = "1" ]; then + config="${config}${indent}${option} true\n" + fi + done + json_select .. + fi + + json_select list 1>/dev/null 2>&1 + if [ $? -eq 0 ]; then + json_get_keys keys + for key in ${keys}; do + json_get_var option "$key" + CONFIG_LIST="" + config_list_foreach "$cfg" "$option" add_list_option "$option" "$indent" + config="${config}${CONFIG_LIST}" + done + json_select .. + fi + + [ -z "$config" ] || { + printf "%s<Plugin %s>\n" "${CONFIG_STRING}" "$cfg" >> "$COLLECTD_CONF" + echo -e "${config}" >> "$COLLECTD_CONF" + printf "%s</Plugin>\n" "${CONFIG_STRING}" >> "$COLLECTD_CONF" + } + + printf "\n" >> "$COLLECTD_CONF" +} + +process_plugins() { + local cfg="$1" + + local enable keys key option value + + config_get enable "$cfg" enable 0 + [ "$enable" = "1" ] || return 0 + + [ -f "/usr/lib/collectd/$cfg.so" ] || { + $LOG notice "Plugin collectd-mod-$cfg not installed" + return 0 + } + + [ -f "/usr/share/collectd/plugin/$cfg.json" ] || { + $LOG notice "Configuration definition file for $cfg not found" + return 0 + } + + printf "LoadPlugin %s\n" "$cfg" >> "$COLLECTD_CONF" + case "$cfg" in + exec) + CONFIG_STRING="" + process_exec + ;; + *) + CONFIG_STRING="" + process_generic "$cfg" "\\t" "/usr/share/collectd/plugin/$cfg.json" + ;; + esac +} + +process_config() { + local alt_config_file BaseDir Include PIDFile PluginDir TypesDB + local Interval ReadThreads Hostname + + rm -f "$COLLECTD_CONF" + + [ -f /etc/config/collectd ] || { + $LOG notice "UCI config not found" + return 0 + } + config_load collectd + config_get alt_config_file globals alt_config_file + + # If "alt_config_file" specified, use that instead + [ -n "$alt_config_file" ] && [ -f "$alt_config_file" ] && { + rm -f "$COLLECTD_CONF" + ln -s "$alt_config_file" "$COLLECTD_CONF" + return 0 + } + + # GOBAL CONFIG + config_get BaseDir globals BaseDir "/var/run/collectd" + printf "BaseDir \"%s\"\n" "$BaseDir" >> "$COLLECTD_CONF" + + config_get Include globals Include "/tmp/collectd.d" + printf "Include \"%s\"\n" "$Include" >> "$COLLECTD_CONF" + mkdir -p "$Include" + + config_get PIDFile globals PIDFile "/var/run/collectd.pid" + printf "PIDFile \"%s\"\n" "$PIDFile" >> "$COLLECTD_CONF" + + config_get PluginDir globals PluginDir "/usr/lib/collectd" + printf "PluginDir \"%s\"\n" "$PluginDir" >> "$COLLECTD_CONF" + + config_get TypesDB globals TypesDB "/usr/share/collectd/types.db" + printf "TypesDB \"%s\"\n" "$TypesDB" >> "$COLLECTD_CONF" + + config_get Interval globals Interval 30 + printf "Interval \"%s\"\n" "$Interval" >> "$COLLECTD_CONF" + + config_get ReadThreads globals ReadThreads 2 + printf "ReadThreads \"%s\"\n" "$ReadThreads" >> "$COLLECTD_CONF" + + config_get Hostname globals Hostname "$(hostname)" + printf "Hostname \"%s\"\n" "$Hostname" >> "$COLLECTD_CONF" + + printf "\n" >> "$COLLECTD_CONF" + + # PLUGIN CONFIG + config_foreach process_plugins plugin +} + start_service() { - mkdir -m 0755 -p /var/lib/collectd procd_open_instance - procd_set_param command /usr/sbin/collectd -f + procd_set_param command /usr/sbin/collectd + procd_append_param command -C "$COLLECTD_CONF" + procd_append_param command -f # don't daemonize, procd will handle that for us procd_set_param nice "$NICEPRIO" + + process_config + + # set auto respawn behavior + procd_set_param respawn procd_close_instance } |