From a8f5b2f8ab59f5bbb93145933a81ea2be4cdb478 Mon Sep 17 00:00:00 2001 From: Thomas Kluyver Date: Mon, 1 Aug 2022 11:51:34 +0100 Subject: [PATCH] Allow custom resolver in WebSocketClientConnection --- tornado/websocket.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tornado/websocket.py b/tornado/websocket.py index 82c29d844..b480cdcee 100644 --- a/tornado/websocket.py +++ b/tornado/websocket.py @@ -34,6 +34,7 @@ from tornado import gen, httpclient, httputil from tornado.ioloop import IOLoop, PeriodicCallback from tornado.iostream import StreamClosedError, IOStream from tornado.log import gen_log, app_log +from tornado.netutil import Resolver from tornado import simple_httpclient from tornado.queues import Queue from tornado.tcpclient import TCPClient @@ -1362,6 +1363,7 @@ class WebSocketClientConnection(simple_httpclient._HTTPConnection): ping_timeout: Optional[float] = None, max_message_size: int = _default_max_message_size, subprotocols: Optional[List[str]] = [], + resolver: Optional[Resolver] = None, ) -> None: self.connect_future = Future() # type: Future[WebSocketClientConnection] self.read_queue = Queue(1) # type: Queue[Union[None, str, bytes]] @@ -1402,7 +1404,7 @@ class WebSocketClientConnection(simple_httpclient._HTTPConnection): # Websocket connection is currently unable to follow redirects request.follow_redirects = False - self.tcp_client = TCPClient() + self.tcp_client = TCPClient(resolver=resolver) super().__init__( None, request, -- 2.47.2