Skip to content

Commit 0e5f106

Browse files
authored
SNOW-266042 Add logging for request id, guid and move logging for query id before ping pong (#614)
1 parent 7d92c1e commit 0e5f106

File tree

2 files changed

+8
-1
lines changed

2 files changed

+8
-1
lines changed

src/snowflake/connector/cursor.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -328,6 +328,8 @@ def _execute_helper(self,
328328
self._sequence_counter = self._connection._next_sequence_counter()
329329
self._request_id = uuid.uuid4()
330330

331+
logger.debug(f'Request id: {self._request_id}')
332+
331333
if logger.getEffectiveLevel() <= logging.DEBUG:
332334
logger.debug(
333335
'running query [%s]', self._format_query_for_log(query))

src/snowflake/connector/network.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -561,6 +561,9 @@ def _post_request(self, url, headers, body, token=None,
561561
return self._post_request(
562562
url, headers, body, token=self.token, timeout=timeout)
563563

564+
if ret.get('data') and ret['data'].get('queryId'):
565+
logger.debug('Query id: {}'.format(ret['data']['queryId']))
566+
564567
if ret.get('code') == QUERY_IN_PROGRESS_ASYNC_CODE and _no_results:
565568
return ret
566569

@@ -628,9 +631,11 @@ def add_request_guid(full_url):
628631
parsed_url = urlparse(full_url)
629632
if not parsed_url.hostname.endswith(SNOWFLAKE_HOST_SUFFIX):
630633
return full_url
634+
request_guid = str(uuid.uuid4())
631635
suffix = urlencode({
632-
REQUEST_GUID: str(uuid.uuid4())
636+
REQUEST_GUID: request_guid
633637
})
638+
logger.debug(f'Request guid: {request_guid}')
634639
sep = '&' if parsed_url.query else '?'
635640
# url has query string already, just add fields
636641
return full_url + sep + suffix

0 commit comments

Comments
 (0)