reproduce retry attempt

src/snowflake/connector/network.py

@@ -381,6 +381,7 @@ def __init__(
         self._sessions_map: dict[str | None, SessionPool] = collections.defaultdict(
             lambda: SessionPool(self)
         )
+        self._counter = 0
 
         # OCSP mode (OCSPMode.FAIL_OPEN by default)
         ssl_wrap_socket.FEATURE_OCSP_MODE = (
@@ -890,6 +891,11 @@ def _request_exec_wrapper(
                 token=token,
                 **kwargs,
             )
+            if method.upper() == 'GET' and "/queries/" in full_url:
+                if self._counter == 0:
+                    pass
+                    #return_object = None
+                self._counter += 1
             if return_object is not None:
                 return return_object
             self._handle_unknown_error(method, full_url, headers, data, conn)

src/snowflake/connector/vendored/urllib3/connection.py

@@ -126,7 +126,7 @@ def __init__(self, *args, **kw):
         # Proxy options provided by the user.
         self.proxy = kw.pop("proxy", None)
         self.proxy_config = kw.pop("proxy_config", None)
-
+        self._counter = 0
         _HTTPConnection.__init__(self, *args, **kw)
 
     @property
@@ -241,7 +241,15 @@ def request(self, method, url, body=None, headers=None):
             headers = headers.copy()
         if "user-agent" not in (six.ensure_str(k.lower()) for k in headers):
             headers["User-Agent"] = _get_default_user_agent()
+        # this goes to python built in client
         super(HTTPConnection, self).request(method, url, body=body, headers=headers)
+        if method.upper() == 'GET' and "/queries/" in url:
+            # try immediately close socket here when performing get query request, not waiting for result
+            # TODO: not sure if this is fast enough to trigger 499 at the backend, check snowhouse traffic if this can trigger 499 error
+            if self._counter == 1:
+                self.sock.close()
+                self._counter = 0
+            self._counter += 1
 
     def request_chunked(self, method, url, body=None, headers=None):
         """

src/snowflake/connector/vendored/urllib3/connectionpool.py

@@ -237,6 +237,7 @@ def __init__(
         # Close all the HTTPConnections in the pool before the
         # HTTPConnectionPool object is garbage collected.
         weakref_finalize(self, _close_pool_connections, pool)
+        self._get_counter = 0
 
     def _new_conn(self):
         """
@@ -639,7 +640,13 @@ def urlopen(
             Additional parameters are passed to
             :meth:`urllib3.response.HTTPResponse.from_httplib`
         """
-
+        """
+        post queries are like the following:
+        '/queries/v1/query-request?requestId=xxx&request_guid=xxx'
+        """
+        if method == "GET" and url.startswith("/queries/"):
+            print('urlopen counter', self._get_counter)
+            self._get_counter += 1
         parsed_url = parse_url(url)
         destination_scheme = parsed_url.scheme
 
@@ -721,6 +728,9 @@ def urlopen(
                 headers=headers,
                 chunked=chunked,
             )
+            # attempt 1: mock SocketError, after calling GET query twice, raise the error
+            if self._get_counter == 2:
+                raise SocketError("Remote Disconnected")
 
             # If we're going to release the connection in ``finally:``, then
             # the response doesn't need to know about the connection. Otherwise
@@ -740,6 +750,10 @@ def urlopen(
                 **response_kw
             )
 
+            # attempt 2: mock raise 499 error
+            # if self._get_counter == 2:
+            #     response = HTTPResponse(status=499)
+
             # Everything went great!
             clean_exit = True