diff options
Diffstat (limited to 'plugin/metrics/setup.go')
| -rw-r--r-- | plugin/metrics/setup.go | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/plugin/metrics/setup.go b/plugin/metrics/setup.go index ffc0466f3..b50960211 100644 --- a/plugin/metrics/setup.go +++ b/plugin/metrics/setup.go @@ -7,6 +7,7 @@ import ( "github.com/coredns/coredns/core/dnsserver" "github.com/coredns/coredns/coremain" "github.com/coredns/coredns/plugin" + "github.com/coredns/coredns/plugin/metrics/vars" clog "github.com/coredns/coredns/plugin/pkg/log" "github.com/coredns/coredns/plugin/pkg/uniq" @@ -50,6 +51,23 @@ func setup(c *caddy.Controller) error { return nil }) + c.OnRestart(func() error { + vars.PluginEnabled.Reset() + return nil + }) + + c.OnStartup(func() error { + conf := dnsserver.GetConfig(c) + plugins := conf.Handlers() + for _, h := range conf.ListenHosts { + addrstr := conf.Transport + "://" + net.JoinHostPort(h, conf.Port) + for _, p := range plugins { + vars.PluginEnabled.WithLabelValues(addrstr, conf.Zone, p.Name()).Set(1) + } + } + return nil + + }) c.OnRestart(m.OnRestart) c.OnFinalShutdown(m.OnFinalShutdown) |