From bf8a0b729051723b48c698c75876f05992f65873 Mon Sep 17 00:00:00 2001 From: Shelikhoo Date: Sat, 11 Dec 2021 22:56:26 +0000 Subject: [PATCH] apply dispatcher interface --- app/dns/nameserver_udp.go | 2 +- transport/internet/udp/dispatcher_split.go | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/app/dns/nameserver_udp.go b/app/dns/nameserver_udp.go index 2621571c2..b746db1a4 100644 --- a/app/dns/nameserver_udp.go +++ b/app/dns/nameserver_udp.go @@ -33,7 +33,7 @@ type ClassicNameServer struct { ips map[string]record requests map[uint16]dnsRequest pub *pubsub.Service - udpServer *udp.Dispatcher + udpServer udp.DispatcherI cleanup *task.Periodic reqID uint32 } diff --git a/transport/internet/udp/dispatcher_split.go b/transport/internet/udp/dispatcher_split.go index ae3030cd5..6c89e9d85 100644 --- a/transport/internet/udp/dispatcher_split.go +++ b/transport/internet/udp/dispatcher_split.go @@ -32,7 +32,7 @@ type Dispatcher struct { callback ResponseCallback } -func NewSplitDispatcher(dispatcher routing.Dispatcher, callback ResponseCallback) *Dispatcher { +func NewSplitDispatcher(dispatcher routing.Dispatcher, callback ResponseCallback) DispatcherI { return &Dispatcher{ conns: make(map[net.Destination]*connEntry), dispatcher: dispatcher, @@ -133,7 +133,7 @@ func DialDispatcher(ctx context.Context, dispatcher routing.Dispatcher) (net.Pac } d := NewSplitDispatcher(dispatcher, c.callback) - c.dispatcher = d + c.dispatcher = d.(*Dispatcher) return c, nil }