Browse Source

apply dispatcher interface

Shelikhoo 4 years ago
parent
commit
bf8a0b7290
2 changed files with 3 additions and 3 deletions
  1. 1 1
      app/dns/nameserver_udp.go
  2. 2 2
      transport/internet/udp/dispatcher_split.go

+ 1 - 1
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
 }

+ 2 - 2
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
 }