diff --git a/dnsstat.go b/dnsstat.go index 9487584..02986a3 100644 --- a/dnsstat.go +++ b/dnsstat.go @@ -3,6 +3,8 @@ package dnsstat import ( "context" + "github.com/google/uuid" + "github.com/coredns/caddy" "github.com/coredns/coredns/core/dnsserver" "github.com/coredns/coredns/plugin" @@ -19,7 +21,13 @@ type DNSStat struct { Next plugin.Handler } -func init() { plugin.Register("dnsstat", setup) } +// Session ID for current run for future better log partitioning in prometheus +var sessionID string + +func init() { + sessionID = uuid.New().String() + plugin.Register("dnsstat", setup) +} // Register func setup(c *caddy.Controller) error { @@ -49,6 +57,7 @@ var requestCount = promauto.NewCounterVec(prometheus.CounterOpts{ "type", "name", "client_ip", + "session_id", }) // ServeDNS implements the plugin.Handler interface. @@ -61,6 +70,7 @@ func (d DNSStat) ServeDNS(ctx context.Context, w dns.ResponseWriter, r *dns.Msg) state.Type(), state.Name(), state.IP(), + sessionID, ).Inc() return plugin.NextOrFailure(d.Name(), d.Next, ctx, w, r) }