diff --git a/setup.go b/setup.go index 4006c51..6772ef3 100644 --- a/setup.go +++ b/setup.go @@ -32,15 +32,7 @@ func setup(c *caddy.Controller) error { // prometheus plugin has been used - if so we will export metrics. We can only register // this metric once, hence the "once.Do". c.OnStartup(func() error { - once.Do(func() { - m := dnsserver.GetConfig(c).Handler("prometheus") - if m == nil { - return - } - if x, ok := m.(*metrics.Metrics); ok { - x.MustRegister(requestCount) - } - }) + once.Do(func() { metrics.MustRegister(c, requestCount) }) return nil })