Fixes MSSQL default query issue (#4614)

This commit is contained in:
Ayush Shah 2022-05-02 16:22:34 +05:30 committed by GitHub
parent c674408dbb
commit dba6bf2adf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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)