mirror of
https://github.com/datahub-project/datahub.git
synced 2025-11-13 17:59:48 +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,
|
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
|
||||||
|
),
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
]
|
]
|
||||||
|
|||||||
@ -17,5 +17,10 @@ record QueryStatement {
|
|||||||
* A SQL Query
|
* A SQL Query
|
||||||
*/
|
*/
|
||||||
SQL
|
SQL
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Unknown query language
|
||||||
|
*/
|
||||||
|
UNKNOWN
|
||||||
} = "SQL"
|
} = "SQL"
|
||||||
}
|
}
|
||||||
Loading…
x
Reference in New Issue
Block a user