diff --git a/Corefile b/Corefile new file mode 100644 index 0000000..4467cca --- /dev/null +++ b/Corefile @@ -0,0 +1,10 @@ +.:9999 { + reload 2s + log + errors + dnsstat + + prometheus 0.0.0.0:51273 + + forward . 1.1.1.1 +} \ No newline at end of file diff --git a/dnsstat.go b/dnsstat.go index 5feef30..004a0b6 100644 --- a/dnsstat.go +++ b/dnsstat.go @@ -8,6 +8,8 @@ import ( "github.com/coredns/coredns/plugin" "github.com/coredns/coredns/plugin/metrics" "github.com/coredns/coredns/request" + "github.com/prometheus/client_golang/prometheus" + "github.com/prometheus/client_golang/prometheus/promauto" "github.com/miekg/dns" ) @@ -19,6 +21,7 @@ type DNSStat struct { func init() { plugin.Register("dnsstat", setup) } +// Register func setup(c *caddy.Controller) error { for c.Next() { if c.NextArg() { @@ -33,6 +36,21 @@ func setup(c *caddy.Controller) error { return nil } +// Metrics counter +var requestCount = promauto.NewCounterVec(prometheus.CounterOpts{ + Namespace: plugin.Namespace, + Subsystem: "dnsstat", + Name: "dnsstat_request_count_total", + Help: "Counter of requests made.", +}, []string{ + "server", + "zone", + "class", + "type", + "name", + "client_ip", +}) + // ServeDNS implements the plugin.Handler interface. func (d DNSStat) ServeDNS(ctx context.Context, w dns.ResponseWriter, r *dns.Msg) (int, error) { state := request.Request{W: w, Req: r} diff --git a/metrics.go b/metrics.go deleted file mode 100644 index 4b3deb5..0000000 --- a/metrics.go +++ /dev/null @@ -1,27 +0,0 @@ -package dnsstat - -import ( - "sync" - - "github.com/coredns/coredns/plugin" - - "github.com/prometheus/client_golang/prometheus" - "github.com/prometheus/client_golang/prometheus/promauto" -) - -// Track total requests made to given domain name + misc -var requestCount = promauto.NewCounterVec(prometheus.CounterOpts{ - Namespace: plugin.Namespace, - Subsystem: "dnsstat", - Name: "dnsstat_request_count_total", - Help: "Counter of requests made.", -}, []string{ - "server", - "zone", - "class", - "type", - "name", - "client_ip", -}) - -var once sync.Once