yujunjun/datahub
2
0
Fork 0
mirror of https://github.com/datahub-project/datahub.git synced 2025-07-25 02:19:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(ingest/snowflake): set platform instance for foreign keys (#9577)

Browse Source
This commit is contained in:
Harshal Sheth 2024-01-09 16:07:49 -05:00 committed by GitHub
parent aa28837eed
commit e3f2c52d7e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
metadata-ingestion/src/datahub/ingestion/source/snowflake/snowflake_v2.py
View File

@ -13,7 +13,6 @@ from snowflake.connector import SnowflakeConnection
from datahub.configuration.pattern_utils import is_schema_allowed from datahub.configuration.pattern_utils import is_schema_allowed
from datahub.emitter.mce_builder import ( from datahub.emitter.mce_builder import (
make_data_platform_urn, make_data_platform_urn,
make_dataset_urn,
make_dataset_urn_with_platform_instance, make_dataset_urn_with_platform_instance,
make_schema_field_urn, make_schema_field_urn,
make_tag_urn, make_tag_urn,
@ -1237,12 +1236,13 @@ class SnowflakeV2Source(
) -> List[ForeignKeyConstraint]: ) -> List[ForeignKeyConstraint]:
foreign_keys = [] foreign_keys = []
for fk in table.foreign_keys: for fk in table.foreign_keys:
foreign_dataset = make_dataset_urn( foreign_dataset = make_dataset_urn_with_platform_instance(
self.platform, platform=self.platform,
self.get_dataset_identifier( name=self.get_dataset_identifier(
fk.referred_table, fk.referred_schema, fk.referred_database fk.referred_table, fk.referred_schema, fk.referred_database
), ),
self.config.env, env=self.config.env,
platform_instance=self.config.platform_instance,
) )
foreign_keys.append( foreign_keys.append(
ForeignKeyConstraint( ForeignKeyConstraint(