Skip to content

Commit 41e1907

Browse files
committed
Update Snowflake CTE code to use new "mkNativeQueryCTEs" fn
1 parent f4bc041 commit 41e1907

File tree

1 file changed

+4
-8
lines changed

1 file changed

+4
-8
lines changed

ndc-connector-snowflake/src/main/kotlin/io/hasura/snowflake/CTEQueryGenerator.kt

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -45,15 +45,11 @@ object CTEQueryGenerator : BaseQueryGenerator() {
4545
}
4646

4747
private fun buildCTEs(request: QueryRequest, varCTE: List<CommonTableExpression<*>> = emptyList()): WithStep {
48-
val isNativeQuery = ConnectorConfiguration.Loader.config.nativeQueries.containsKey(request.collection)
48+
val withStep = mkNativeQueryCTEs(request)
49+
.with(varCTE)
50+
.with(forEachQueryLevelRecursively(request, CTEQueryGenerator::buildCTE).distinct())
4951

50-
return DSL.with(
51-
buildList {
52-
if (isNativeQuery) add(mkNativeQueryCTESnowflake(request))
53-
addAll(varCTE)
54-
addAll(forEachQueryLevelRecursively(request, CTEQueryGenerator::buildCTE).distinct())
55-
}
56-
)
52+
return withStep
5753
}
5854

5955
private fun getCollectionAsjOOQName(collection: String): Name {

0 commit comments

Comments
 (0)