From dba6bf2adf47c5cceefb982a46ca18bcef3a0e01 Mon Sep 17 00:00:00 2001 From: Ayush Shah Date: Mon, 2 May 2022 16:22:34 +0530 Subject: [PATCH] Fixes MSSQL default query issue (#4614) --- ingestion/src/metadata/ingestion/source/mssql.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/ingestion/src/metadata/ingestion/source/mssql.py b/ingestion/src/metadata/ingestion/source/mssql.py index f777d62024c..98b086e26ea 100644 --- a/ingestion/src/metadata/ingestion/source/mssql.py +++ b/ingestion/src/metadata/ingestion/source/mssql.py @@ -16,6 +16,9 @@ from metadata.generated.schema.entity.services.connections.database.mssqlConnect from metadata.generated.schema.entity.services.connections.metadata.openMetadataConnection import ( OpenMetadataConnection, ) +from metadata.generated.schema.metadataIngestion.databaseServiceMetadataPipeline import ( + DatabaseServiceMetadataPipeline, +) from metadata.generated.schema.metadataIngestion.workflow import ( Source as WorkflowSource, ) @@ -41,4 +44,8 @@ class MssqlSource(SQLSource): raise InvalidSourceException( f"Expected MssqlConnection, but got {connection}" ) + if config.sourceConfig.config.sampleDataQuery == "select * from {}.{} limit 50": + config.sourceConfig.config.sampleDataQuery = ( + "select top 50 * from [{}].[{}]" + ) return cls(config, metadata_config)