Skip to content

Commit

Permalink
[NOSNOW] Correct server-connection.py in merge-decoder (#2958)
Browse files Browse the repository at this point in the history
  • Loading branch information
sfc-gh-vbudati authored Jan 29, 2025
1 parent 2a23e90 commit 20441c6
Showing 1 changed file with 8 additions and 2 deletions.
10 changes: 8 additions & 2 deletions src/snowflake/snowpark/_internal/server_connection.py
Original file line number Diff line number Diff line change
Expand Up @@ -51,6 +51,8 @@
from snowflake.snowpark._internal.error_message import SnowparkClientExceptionMessages
from snowflake.snowpark._internal.telemetry import TelemetryClient
from snowflake.snowpark._internal.utils import (
create_rlock,
create_thread_local,
escape_quotes,
get_application_name,
get_version,
Expand Down Expand Up @@ -171,8 +173,12 @@ def __init__(
except TypeError:
pass

self._lock = threading.RLock()
self._thread_store = threading.local()
# thread safe param protection
self._thread_safe_session_enabled = self._get_client_side_session_parameter(
"PYTHON_SNOWPARK_ENABLE_THREAD_SAFE_SESSION", False
)
self._lock = create_rlock(self._thread_safe_session_enabled)
self._thread_store = create_thread_local(self._thread_safe_session_enabled)

if "password" in self._lower_case_parameters:
self._lower_case_parameters["password"] = None
Expand Down

0 comments on commit 20441c6

Please sign in to comment.