diff --git a/ingestion/src/metadata/ingestion/source/database/db2/service_spec.py b/ingestion/src/metadata/ingestion/source/database/db2/service_spec.py index e5ec7fdd3ed..616556249b0 100644 --- a/ingestion/src/metadata/ingestion/source/database/db2/service_spec.py +++ b/ingestion/src/metadata/ingestion/source/database/db2/service_spec.py @@ -1,3 +1,4 @@ +from metadata.ingestion.source.database.db2.lineage import Db2LineageSource from metadata.ingestion.source.database.db2.metadata import Db2Source from metadata.profiler.interface.sqlalchemy.db2.profiler_interface import ( DB2ProfilerInterface, @@ -5,5 +6,7 @@ from metadata.profiler.interface.sqlalchemy.db2.profiler_interface import ( from metadata.utils.service_spec.default import DefaultDatabaseSpec ServiceSpec = DefaultDatabaseSpec( - metadata_source_class=Db2Source, profiler_class=DB2ProfilerInterface + metadata_source_class=Db2Source, + profiler_class=DB2ProfilerInterface, + lineage_source_class=Db2LineageSource, ) diff --git a/ingestion/src/metadata/ingestion/source/database/druid/service_spec.py b/ingestion/src/metadata/ingestion/source/database/druid/service_spec.py index e83fe9a41a0..066919a1a39 100644 --- a/ingestion/src/metadata/ingestion/source/database/druid/service_spec.py +++ b/ingestion/src/metadata/ingestion/source/database/druid/service_spec.py @@ -1,4 +1,8 @@ +from metadata.ingestion.source.database.druid.lineage import DruidLineageSource from metadata.ingestion.source.database.druid.metadata import DruidSource from metadata.utils.service_spec.default import DefaultDatabaseSpec -ServiceSpec = DefaultDatabaseSpec(metadata_source_class=DruidSource) +ServiceSpec = DefaultDatabaseSpec( + metadata_source_class=DruidSource, + lineage_source_class=DruidLineageSource, +) diff --git a/ingestion/src/metadata/ingestion/source/database/greenplum/service_spec.py b/ingestion/src/metadata/ingestion/source/database/greenplum/service_spec.py index 9fe3ac14cfa..14b430d2554 100644 --- a/ingestion/src/metadata/ingestion/source/database/greenplum/service_spec.py +++ b/ingestion/src/metadata/ingestion/source/database/greenplum/service_spec.py @@ -1,4 +1,8 @@ +from metadata.ingestion.source.database.greenplum.lineage import GreenplumLineageSource from metadata.ingestion.source.database.greenplum.metadata import GreenplumSource from metadata.utils.service_spec.default import DefaultDatabaseSpec -ServiceSpec = DefaultDatabaseSpec(metadata_source_class=GreenplumSource) +ServiceSpec = DefaultDatabaseSpec( + metadata_source_class=GreenplumSource, + lineage_source_class=GreenplumLineageSource, +) diff --git a/ingestion/src/metadata/ingestion/source/database/hive/service_spec.py b/ingestion/src/metadata/ingestion/source/database/hive/service_spec.py index e303ccfab77..31adc04c667 100644 --- a/ingestion/src/metadata/ingestion/source/database/hive/service_spec.py +++ b/ingestion/src/metadata/ingestion/source/database/hive/service_spec.py @@ -1,4 +1,8 @@ +from metadata.ingestion.source.database.hive.lineage import HiveLineageSource from metadata.ingestion.source.database.hive.metadata import HiveSource from metadata.utils.service_spec.default import DefaultDatabaseSpec -ServiceSpec = DefaultDatabaseSpec(metadata_source_class=HiveSource) +ServiceSpec = DefaultDatabaseSpec( + metadata_source_class=HiveSource, + lineage_source_class=HiveLineageSource, +) diff --git a/ingestion/src/metadata/ingestion/source/database/impala/service_spec.py b/ingestion/src/metadata/ingestion/source/database/impala/service_spec.py index 51e302a88e4..31c04e516f0 100644 --- a/ingestion/src/metadata/ingestion/source/database/impala/service_spec.py +++ b/ingestion/src/metadata/ingestion/source/database/impala/service_spec.py @@ -1,4 +1,8 @@ +from metadata.ingestion.source.database.impala.lineage import ImpalaLineageSource from metadata.ingestion.source.database.impala.metadata import ImpalaSource from metadata.utils.service_spec.default import DefaultDatabaseSpec -ServiceSpec = DefaultDatabaseSpec(metadata_source_class=ImpalaSource) +ServiceSpec = DefaultDatabaseSpec( + metadata_source_class=ImpalaSource, + lineage_source_class=ImpalaLineageSource, +) diff --git a/ingestion/src/metadata/ingestion/source/database/mariadb/service_spec.py b/ingestion/src/metadata/ingestion/source/database/mariadb/service_spec.py index 40f55327c90..39775592628 100644 --- a/ingestion/src/metadata/ingestion/source/database/mariadb/service_spec.py +++ b/ingestion/src/metadata/ingestion/source/database/mariadb/service_spec.py @@ -1,4 +1,12 @@ +from metadata.ingestion.source.database.mariadb.lineage import MariadbLineageSource from metadata.ingestion.source.database.mariadb.metadata import MariadbSource +from metadata.profiler.interface.sqlalchemy.mariadb.profiler_interface import ( + MariaDBProfilerInterface, +) from metadata.utils.service_spec.default import DefaultDatabaseSpec -ServiceSpec = DefaultDatabaseSpec(metadata_source_class=MariadbSource) +ServiceSpec = DefaultDatabaseSpec( + metadata_source_class=MariadbSource, + lineage_source_class=MariadbLineageSource, + profiler_class=MariaDBProfilerInterface, +) diff --git a/ingestion/src/metadata/ingestion/source/database/pinotdb/service_spec.py b/ingestion/src/metadata/ingestion/source/database/pinotdb/service_spec.py index ffc4b1885a1..4178ae9bb5d 100644 --- a/ingestion/src/metadata/ingestion/source/database/pinotdb/service_spec.py +++ b/ingestion/src/metadata/ingestion/source/database/pinotdb/service_spec.py @@ -1,4 +1,8 @@ +from metadata.ingestion.source.database.pinotdb.lineage import PinotdbLineageSource from metadata.ingestion.source.database.pinotdb.metadata import PinotdbSource from metadata.utils.service_spec.default import DefaultDatabaseSpec -ServiceSpec = DefaultDatabaseSpec(metadata_source_class=PinotdbSource) +ServiceSpec = DefaultDatabaseSpec( + metadata_source_class=PinotdbSource, + lineage_source_class=PinotdbLineageSource, +) diff --git a/ingestion/src/metadata/ingestion/source/database/singlestore/service_spec.py b/ingestion/src/metadata/ingestion/source/database/singlestore/service_spec.py index 3175e998f12..c742e5de87d 100644 --- a/ingestion/src/metadata/ingestion/source/database/singlestore/service_spec.py +++ b/ingestion/src/metadata/ingestion/source/database/singlestore/service_spec.py @@ -1,3 +1,6 @@ +from metadata.ingestion.source.database.singlestore.lineage import ( + SinglestoreLineageSource, +) from metadata.ingestion.source.database.singlestore.metadata import SinglestoreSource from metadata.profiler.interface.sqlalchemy.single_store.profiler_interface import ( SingleStoreProfilerInterface, @@ -5,5 +8,7 @@ from metadata.profiler.interface.sqlalchemy.single_store.profiler_interface impo from metadata.utils.service_spec.default import DefaultDatabaseSpec ServiceSpec = DefaultDatabaseSpec( - metadata_source_class=SinglestoreSource, profiler_class=SingleStoreProfilerInterface + metadata_source_class=SinglestoreSource, + profiler_class=SingleStoreProfilerInterface, + lineage_source_class=SinglestoreLineageSource, ) diff --git a/ingestion/src/metadata/ingestion/source/database/sqlite/service_spec.py b/ingestion/src/metadata/ingestion/source/database/sqlite/service_spec.py index 6ffbe21a16c..ea7af10ac9c 100644 --- a/ingestion/src/metadata/ingestion/source/database/sqlite/service_spec.py +++ b/ingestion/src/metadata/ingestion/source/database/sqlite/service_spec.py @@ -1,4 +1,8 @@ +from metadata.ingestion.source.database.sqlite.lineage import SqliteLineageSource from metadata.ingestion.source.database.sqlite.metadata import SqliteSource from metadata.utils.service_spec.default import DefaultDatabaseSpec -ServiceSpec = DefaultDatabaseSpec(metadata_source_class=SqliteSource) +ServiceSpec = DefaultDatabaseSpec( + metadata_source_class=SqliteSource, + lineage_source_class=SqliteLineageSource, +) diff --git a/ingestion/src/metadata/ingestion/source/database/teradata/service_spec.py b/ingestion/src/metadata/ingestion/source/database/teradata/service_spec.py index 4fe31877cdf..83dab69c374 100644 --- a/ingestion/src/metadata/ingestion/source/database/teradata/service_spec.py +++ b/ingestion/src/metadata/ingestion/source/database/teradata/service_spec.py @@ -1,4 +1,8 @@ +from metadata.ingestion.source.database.teradata.lineage import TeradataLineageSource from metadata.ingestion.source.database.teradata.metadata import TeradataSource from metadata.utils.service_spec.default import DefaultDatabaseSpec -ServiceSpec = DefaultDatabaseSpec(metadata_source_class=TeradataSource) +ServiceSpec = DefaultDatabaseSpec( + metadata_source_class=TeradataSource, + lineage_source_class=TeradataLineageSource, +)