mirror of
https://github.com/datahub-project/datahub.git
synced 2025-11-02 03:39:03 +00:00
fix(ingest/snowflake): fix error on stored procs in non-SQL languages (#13127)
This commit is contained in:
parent
742d060722
commit
1fca9855ee
@ -26,6 +26,7 @@ from datahub.metadata.schema_classes import (
|
||||
DataPlatformInstanceClass,
|
||||
DataTransformClass,
|
||||
DataTransformLogicClass,
|
||||
QueryLanguageClass,
|
||||
QueryStatementClass,
|
||||
SubTypesClass,
|
||||
)
|
||||
@ -176,7 +177,17 @@ def _generate_job_workunits(
|
||||
DataTransformClass(
|
||||
queryStatement=QueryStatementClass(
|
||||
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
|
||||
),
|
||||
),
|
||||
)
|
||||
]
|
||||
|
||||
@ -17,5 +17,10 @@ record QueryStatement {
|
||||
* A SQL Query
|
||||
*/
|
||||
SQL
|
||||
|
||||
/**
|
||||
* Unknown query language
|
||||
*/
|
||||
UNKNOWN
|
||||
} = "SQL"
|
||||
}
|
||||
Loading…
x
Reference in New Issue
Block a user