feat(airflow): respect port parameter if provided (datahub-project#7945)

tejbloke · May 2, 2023 · 6833494 · 6833494
1 parent bf86235
commit 6833494
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/metadata-ingestion/src/datahub_provider/hooks/datahub.py b/metadata-ingestion/src/datahub_provider/hooks/datahub.py
@@ -58,6 +58,12 @@ def _get_config(self) -> Tuple[str, Optional[str], Optional[int]]:
         host = conn.host
         if host is None:
             raise AirflowException("host parameter is required")
+        if conn.port:
+            if ":" in host:
+                raise AirflowException(
+                    "host parameter should not contain a port number if the port is specified separately"
+                )
+            host = f"{host}:{conn.port}"
         password = conn.password
         timeout_sec = conn.extra_dejson.get("timeout_sec")
         return (host, password, timeout_sec)