fix(ingest/snowflake): fix error on stored procs in non-SQL languages (#13127)

This commit is contained in:
Harshal Sheth 2025-04-08 17:00:56 -07:00 committed by GitHub
parent 742d060722
commit 1fca9855ee
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 17 additions and 1 deletions

View File

@ -26,6 +26,7 @@ from datahub.metadata.schema_classes import (
DataPlatformInstanceClass, DataPlatformInstanceClass,
DataTransformClass, DataTransformClass,
DataTransformLogicClass, DataTransformLogicClass,
QueryLanguageClass,
QueryStatementClass, QueryStatementClass,
SubTypesClass, SubTypesClass,
) )
@ -176,7 +177,17 @@ def _generate_job_workunits(
DataTransformClass( DataTransformClass(
queryStatement=QueryStatementClass( queryStatement=QueryStatementClass(
value=procedure.procedure_definition, value=procedure.procedure_definition,
language=procedure.language, language=(
QueryLanguageClass.SQL
if procedure.language == "SQL"
# The language field uses a pretty limited enum.
# The "UNKNOWN" enum value is pretty new, so we don't want to
# emit it until it has broader server-side support. As a
# short-term solution, we map all languages to "SQL".
# TODO: Once we've released server 1.1.0, we should change
# this to be "UNKNOWN" for all languages except "SQL".
else QueryLanguageClass.SQL
),
), ),
) )
] ]

View File

@ -17,5 +17,10 @@ record QueryStatement {
* A SQL Query * A SQL Query
*/ */
SQL SQL
/**
* Unknown query language
*/
UNKNOWN
} = "SQL" } = "SQL"
} }