SNOW-612517 reimplement how gzip compression is applied (snowflakedb#1200)

sfc-gh-mkeller · web-flow · commit 939f70f6d041 · 2022-07-22T13:51:18.000-07:00
diff --git a/DESCRIPTION.md b/DESCRIPTION.md
@@ -14,6 +14,8 @@ Source code is also available at: https://github.com/snowflakedb/snowflake-conne
    - Release wheels are now built on manylinux2014
    - Bumped supported pyarrow version to >=8.0.0,<8.1.0
    - Added overwrite option to write_pandas
+   - Fixed a bug where gzip compressed http requests might be garbled by an unflushed buffer
+
 
 
 - v2.7.9(June 26,2022)
diff --git a/src/snowflake/connector/network.py b/src/snowflake/connector/network.py
@@ -14,7 +14,6 @@
 import time
 import traceback
 import uuid
-from io import BytesIO
 from threading import Lock
 from typing import TYPE_CHECKING
 
@@ -1007,11 +1006,8 @@ def _request_exec(
         logger.debug("socket timeout: %s", socket_timeout)
         try:
             if not catch_okta_unauthorized_error and data and len(data) > 0:
-                gzdata = BytesIO()
-                gzip.GzipFile(fileobj=gzdata, mode="wb").write(data.encode("utf-8"))
-                gzdata.seek(0, 0)
                 headers["Content-Encoding"] = "gzip"
-                input_data = gzdata
+                input_data = gzip.compress(data.encode("utf-8"))
             else:
                 input_data = data
 
