Merge pull request #4 from dominodatalab/ddl-giuliocapolino.DOM-33794…

….avoid-timeout-if-pod-dies

datasourceSDK: avoid timeouts when ds proxy dies

.gitignore

@@ -52,6 +52,7 @@ Temporary Items
 .idea/**/dynamic.xml
 .idea/**/uiDesigner.xml
 .idea/**/dbnavigator.xml
+.idea/
 
 # Gradle
 .idea/**/gradle.xml

domino_data/data_sources.py

@@ -490,15 +490,7 @@ def __attrs_post_init__(self):
             proxy_host=proxy_host,
         )
 
-        self.proxy = flight.FlightClient(
-            flight_host,
-            middleware=[
-                AuthMiddlewareFactory(
-                    self.api_key,
-                    self.token_file,
-                )
-            ],
-        )
+        self._set_proxy()
         self.proxy_http = AuthenticatedClient(
             base_url=proxy_host,
             api_key=self.api_key,
@@ -511,6 +503,18 @@ def __attrs_post_init__(self):
             headers=ACCEPT_HEADERS,
         )
 
+    def _set_proxy(self):
+        flight_host = os.getenv("DOMINO_DATASOURCE_PROXY_FLIGHT_HOST")
+        self.proxy = flight.FlightClient(
+            flight_host,
+            middleware=[
+                AuthMiddlewareFactory(
+                    self.api_key,
+                    self.token_file,
+                )
+            ],
+        )
+
     def get_datasource(self, name: str) -> Datasource:
         """Fetch a datasource by name.
 
@@ -647,6 +651,7 @@ def execute(
         logger.info("execute", datasource_id=datasource_id, query=query)
 
         try:
+            self._set_proxy()
             reader = self._do_get(
                 BoardingPass(
                     datasource_id=datasource_id,

pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "dominodatalab-data"
-version = "0.1.2"
+version = "0.1.3"
 description = "Domino Data SDK for interacting with Access Data features"
 readme = "README.md"
 authors = [