aboutsummaryrefslogtreecommitdiff
path: root/middleware/trace/trace.go
diff options
context:
space:
mode:
Diffstat (limited to 'middleware/trace/trace.go')
-rw-r--r--middleware/trace/trace.go84
1 files changed, 0 insertions, 84 deletions
diff --git a/middleware/trace/trace.go b/middleware/trace/trace.go
deleted file mode 100644
index 86561871a..000000000
--- a/middleware/trace/trace.go
+++ /dev/null
@@ -1,84 +0,0 @@
-// Package trace implements OpenTracing-based tracing
-package trace
-
-import (
- "fmt"
- "sync"
- "sync/atomic"
-
- "github.com/coredns/coredns/middleware"
- // Plugin the trace package.
- _ "github.com/coredns/coredns/middleware/pkg/trace"
-
- "github.com/miekg/dns"
- ot "github.com/opentracing/opentracing-go"
- zipkin "github.com/openzipkin/zipkin-go-opentracing"
- "golang.org/x/net/context"
-)
-
-type trace struct {
- Next middleware.Handler
- ServiceEndpoint string
- Endpoint string
- EndpointType string
- tracer ot.Tracer
- serviceName string
- clientServer bool
- every uint64
- count uint64
- Once sync.Once
-}
-
-func (t *trace) Tracer() ot.Tracer {
- return t.tracer
-}
-
-// OnStartup sets up the tracer
-func (t *trace) OnStartup() error {
- var err error
- t.Once.Do(func() {
- switch t.EndpointType {
- case "zipkin":
- err = t.setupZipkin()
- default:
- err = fmt.Errorf("unknown endpoint type: %s", t.EndpointType)
- }
- })
- return err
-}
-
-func (t *trace) setupZipkin() error {
-
- collector, err := zipkin.NewHTTPCollector(t.Endpoint)
- if err != nil {
- return err
- }
-
- recorder := zipkin.NewRecorder(collector, false, t.ServiceEndpoint, t.serviceName)
- t.tracer, err = zipkin.NewTracer(recorder, zipkin.ClientServerSameSpan(t.clientServer))
-
- return err
-}
-
-// Name implements the Handler interface.
-func (t *trace) Name() string {
- return "trace"
-}
-
-// ServeDNS implements the middleware.Handle interface.
-func (t *trace) ServeDNS(ctx context.Context, w dns.ResponseWriter, r *dns.Msg) (int, error) {
- trace := false
- if t.every > 0 {
- queryNr := atomic.AddUint64(&t.count, 1)
-
- if queryNr%t.every == 0 {
- trace = true
- }
- }
- if span := ot.SpanFromContext(ctx); span == nil && trace {
- span := t.Tracer().StartSpan("servedns")
- defer span.Finish()
- ctx = ot.ContextWithSpan(ctx, span)
- }
- return middleware.NextOrFailure(t.Name(), t.Next, ctx, w, r)
-}