Skip to content

Commit 2f96c4f

Browse files
committed
Pass Endpoint#wrapper to handlers.
1 parent 8875296 commit 2f96c4f

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

lib/async/dns/server.rb

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -104,13 +104,15 @@ def run
104104
Console.info "Starting Async::DNS server (v#{Async::DNS::VERSION})..."
105105

106106
Async do |task|
107+
wrapper = @endpoint.wrapper
108+
107109
@endpoint.bind do |server|
108110
Console.info "<> Listening for connections on #{server.local_address.inspect}"
109111
case server.local_address.socktype
110112
when Socket::SOCK_DGRAM
111-
DatagramHandler.new(self, server).run
113+
DatagramHandler.new(self, server).run(wrapper)
112114
when Socket::SOCK_STREAM
113-
StreamHandler.new(self, server).run
115+
StreamHandler.new(self, server).run(wrapper)
114116
else
115117
raise ArgumentError.new("Don't know how to handle #{server}")
116118
end

0 commit comments

Comments
 (0)