mirror of
https://github.com/pdf/zfs_exporter.git
synced 2025-06-03 03:58:59 +00:00
056b386307
Add github actions CI. BREAKING CHANGE: Go API has changed somewhat, but metrics remain unaffected.
80 lines
1.9 KiB
Go
80 lines
1.9 KiB
Go
package collector
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/mistifyio/go-zfs"
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
"gopkg.in/alecthomas/kingpin.v2"
|
|
)
|
|
|
|
const (
|
|
defaultEnabled = true
|
|
defaultDisabled = false
|
|
namespace = `zfs`
|
|
helpDefaultStateEnabled = `enabled`
|
|
helpDefaultStateDisabled = `disabled`
|
|
)
|
|
|
|
var (
|
|
collectorStates = make(map[string]State)
|
|
scrapeDurationDescName = prometheus.BuildFQName(namespace, `scrape`, `collector_duration_seconds`)
|
|
scrapeDurationDesc = prometheus.NewDesc(
|
|
scrapeDurationDescName,
|
|
`zfs_exporter: Duration of a collector scrape.`,
|
|
[]string{`collector`},
|
|
nil,
|
|
)
|
|
scrapeSuccessDescName = prometheus.BuildFQName(namespace, `scrape`, `collector_success`)
|
|
scrapeSuccessDesc = prometheus.NewDesc(
|
|
scrapeSuccessDescName,
|
|
`zfs_exporter: Whether a collector succeeded.`,
|
|
[]string{`collector`},
|
|
nil,
|
|
)
|
|
)
|
|
|
|
type factoryFunc func() (Collector, error)
|
|
|
|
type State struct {
|
|
Enabled *bool
|
|
factory factoryFunc
|
|
}
|
|
|
|
type Collector interface {
|
|
update(ch chan<- metric, pools []*zfs.Zpool) error
|
|
}
|
|
|
|
type metric struct {
|
|
name string
|
|
prometheus prometheus.Metric
|
|
}
|
|
|
|
type desc struct {
|
|
name string
|
|
prometheus *prometheus.Desc
|
|
}
|
|
|
|
func registerCollector(collector string, isDefaultEnabled bool, factory factoryFunc) {
|
|
helpDefaultState := helpDefaultStateDisabled
|
|
if isDefaultEnabled {
|
|
helpDefaultState = helpDefaultStateEnabled
|
|
}
|
|
|
|
flagName := fmt.Sprintf("collector.%s", collector)
|
|
flagHelp := fmt.Sprintf("Enable the %s collector (default: %s)", collector, helpDefaultState)
|
|
defaultValue := fmt.Sprintf("%t", isDefaultEnabled)
|
|
|
|
flag := kingpin.Flag(flagName, flagHelp).Default(defaultValue).Bool()
|
|
|
|
collectorStates[collector] = State{
|
|
Enabled: flag,
|
|
factory: factory,
|
|
}
|
|
}
|
|
|
|
func expandMetricName(prefix string, context ...string) string {
|
|
return strings.Join(append(context, prefix), `-`)
|
|
}
|