From: Thomas Kluyver <thomas@kluyver.me.uk>
Date: Mon, 1 Aug 2022 10:51:34 +0000 (+0100)
Subject: Allow custom resolver in WebSocketClientConnection
X-Git-Tag: v6.3.0b1~29^2
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=refs%2Fpull%2F3172%2Fhead;p=thirdparty%2Ftornado.git

Allow custom resolver in WebSocketClientConnection
---

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,