Add runit service dir flag

This commit is contained in:
Johannes 'fish' Ziemke 2016-12-19 13:10:38 +01:00
parent 677ed28575
commit 9039a425d0
1 changed files with 8 additions and 1 deletions

View File

@ -16,11 +16,18 @@
package collector package collector
import ( import (
"flag"
"github.com/prometheus/client_golang/prometheus" "github.com/prometheus/client_golang/prometheus"
"github.com/prometheus/common/log" "github.com/prometheus/common/log"
"github.com/soundcloud/go-runit/runit" "github.com/soundcloud/go-runit/runit"
) )
var runitServiceDir = flag.String(
"collector.runit.servicecdir",
"/etc/service",
"Path to runit service directory.")
type runitCollector struct { type runitCollector struct {
state, stateDesired, stateNormal, stateTimestamp *prometheus.GaugeVec state, stateDesired, stateNormal, stateTimestamp *prometheus.GaugeVec
} }
@ -81,7 +88,7 @@ func NewRunitCollector() (Collector, error) {
} }
func (c *runitCollector) Update(ch chan<- prometheus.Metric) error { func (c *runitCollector) Update(ch chan<- prometheus.Metric) error {
services, err := runit.GetServices("/etc/service") services, err := runit.GetServices(*runitServiceDir)
if err != nil { if err != nil {
return err return err
} }