Skip to content

Commit 9e4ef2e

Browse files
pass ssl_options into CloudFetchQueue
Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>
1 parent c07f709 commit 9e4ef2e

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

src/databricks/sql/backend/sea/queue.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,8 @@
33
from abc import ABC
44
from typing import List, Optional, Tuple, Union
55

6+
from databricks.sql.cloudfetch.download_manager import ResultFileDownloadManager
7+
68
try:
79
import pyarrow
810
except ImportError:
@@ -33,6 +35,7 @@ def build_queue(
3335
sea_result_data: ResultData,
3436
manifest: ResultManifest,
3537
statement_id: str,
38+
ssl_options: Optional[SSLOptions] = None,
3639
description: List[Tuple] = [],
3740
max_download_threads: Optional[int] = None,
3841
sea_client: Optional[SeaDatabricksClient] = None,

src/databricks/sql/backend/sea/result_set.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,7 @@ def __init__(
6060
result_data,
6161
self.manifest,
6262
statement_id,
63+
ssl_options=connection.session.ssl_options,
6364
description=execute_response.description,
6465
max_download_threads=sea_client.max_download_threads,
6566
sea_client=sea_client,

0 commit comments

Comments
 (0)