diff options
Diffstat (limited to 'plugin/metrics/setup.go')
| -rw-r--r-- | plugin/metrics/setup.go | 12 |
1 files changed, 7 insertions, 5 deletions
diff --git a/plugin/metrics/setup.go b/plugin/metrics/setup.go index 52d5775c1..c00f44a83 100644 --- a/plugin/metrics/setup.go +++ b/plugin/metrics/setup.go @@ -8,19 +8,21 @@ import ( "github.com/coredns/coredns/coremain" "github.com/coredns/coredns/plugin" clog "github.com/coredns/coredns/plugin/pkg/log" + "github.com/coredns/coredns/plugin/pkg/uniq" "github.com/mholt/caddy" ) -var log = clog.NewWithPlugin("prometheus") +var ( + log = clog.NewWithPlugin("prometheus") + uniqAddr = uniq.New() +) func init() { caddy.RegisterPlugin("prometheus", caddy.Plugin{ ServerType: "dns", Action: setup, }) - - uniqAddr = newAddress() } func setup(c *caddy.Controller) error { @@ -36,7 +38,7 @@ func setup(c *caddy.Controller) error { c.OncePerServerBlock(func() error { c.OnStartup(func() error { - return uniqAddr.forEachTodo() + return uniqAddr.ForEach() }) return nil }) @@ -54,7 +56,7 @@ func prometheusParse(c *caddy.Controller) (*Metrics, error) { var met = New(defaultAddr) defer func() { - uniqAddr.setAddress(met.Addr, met.OnStartup) + uniqAddr.Set(met.Addr, met.OnStartup) }() i := 0 |