diff options
Diffstat (limited to 'plugin/metrics')
| -rw-r--r-- | plugin/metrics/context.go | 24 | ||||
| -rw-r--r-- | plugin/metrics/metrics.go | 1 |
2 files changed, 25 insertions, 0 deletions
diff --git a/plugin/metrics/context.go b/plugin/metrics/context.go new file mode 100644 index 000000000..4cce871f5 --- /dev/null +++ b/plugin/metrics/context.go @@ -0,0 +1,24 @@ +package metrics + +import ( + "github.com/coredns/coredns/plugin" + + "golang.org/x/net/context" +) + +// WithServer returns the current server handling the request. It returns the +// server listening address: <scheme>://[<bind>]:<port> Normally this is +// something like "dns://:53", but if the bind plugin is used, i.e. "bind +// 127.0.0.53", it will be "dns://127.0.0.53:53", etc. If not address is found +// the empty string is returned. +// +// Basic usage with a metric: +// +// <metric>.WithLabelValues(metrics.WithServer(ctx), labels..).Add(1) +func WithServer(ctx context.Context) string { + srv := ctx.Value(plugin.ServerCtx{}) + if srv == nil { + return "" + } + return srv.(string) +} diff --git a/plugin/metrics/metrics.go b/plugin/metrics/metrics.go index a49984b3d..e16749fc0 100644 --- a/plugin/metrics/metrics.go +++ b/plugin/metrics/metrics.go @@ -12,6 +12,7 @@ import ( "github.com/coredns/coredns/coremain" "github.com/coredns/coredns/plugin" "github.com/coredns/coredns/plugin/metrics/vars" + "github.com/prometheus/client_golang/prometheus" "github.com/prometheus/client_golang/prometheus/promhttp" ) |