diff options
Diffstat (limited to 'plugin/dnstap/handler.go')
| -rw-r--r-- | plugin/dnstap/handler.go | 9 |
1 files changed, 2 insertions, 7 deletions
diff --git a/plugin/dnstap/handler.go b/plugin/dnstap/handler.go index b09c70406..1178dad79 100644 --- a/plugin/dnstap/handler.go +++ b/plugin/dnstap/handler.go @@ -44,12 +44,6 @@ const ( DnstapSendOption ContextKey = "dnstap-send-option" ) -// TapperFromContext will return a Tapper if the dnstap plugin is enabled. -func TapperFromContext(ctx context.Context) (t Tapper) { - t, _ = ctx.(Tapper) - return -} - // TapMessage implements Tapper. func (h Dnstap) TapMessage(m *tap.Message) { t := tap.Dnstap_MESSAGE @@ -71,6 +65,7 @@ func (h Dnstap) ServeDNS(ctx context.Context, w dns.ResponseWriter, r *dns.Msg) // message to be sent out sendOption := taprw.SendOption{Cq: true, Cr: true} newCtx := context.WithValue(ctx, DnstapSendOption, &sendOption) + newCtx = ContextWithTapper(newCtx, h) rw := &taprw.ResponseWriter{ ResponseWriter: w, @@ -80,7 +75,7 @@ func (h Dnstap) ServeDNS(ctx context.Context, w dns.ResponseWriter, r *dns.Msg) QueryEpoch: time.Now(), } - code, err := plugin.NextOrFailure(h.Name(), h.Next, tapContext{newCtx, h}, rw, r) + code, err := plugin.NextOrFailure(h.Name(), h.Next, newCtx, rw, r) if err != nil { // ignore dnstap errors return code, err |