refactor(strm-2708): minor stuff

getstrm · Oct 24, 2023 · 123b0bc · 123b0bc
1 parent 422df9f
commit 123b0bc
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 13 deletions.
diff --git a/app/src/main/kotlin/com/getstrm/pace/common/AbstractDynamicViewGenerator.kt b/app/src/main/kotlin/com/getstrm/pace/common/AbstractDynamicViewGenerator.kt
@@ -1,6 +1,7 @@
 package com.getstrm.pace.common
 
 import build.buf.gen.getstrm.api.data_policies.v1alpha.DataPolicy
+import com.getstrm.pace.domain.SqlParseException
 import headTailFold
 import org.jooq.* // ktlint-disable no-wildcard-imports
 import org.jooq.conf.ParseNameCase

diff --git a/app/src/main/kotlin/com/getstrm/pace/common/Exceptions.kt b/app/src/main/kotlin/com/getstrm/pace/common/Exceptions.kt
diff --git a/app/src/main/kotlin/com/getstrm/pace/snowflake/SnowflakeClient.kt b/app/src/main/kotlin/com/getstrm/pace/snowflake/SnowflakeClient.kt
@@ -49,7 +49,9 @@ class SnowflakeClient(
     }
 
     override suspend fun applyPolicy(dataPolicy: DataPolicy) {
-        val statement = SnowflakeDynamicViewGenerator(dataPolicy).toDynamicViewSQL().replace("\\n".toRegex(), "\\\\n")
+        val statement = SnowflakeDynamicViewGenerator(dataPolicy).toDynamicViewSQL()
+            .replace("\\n".toRegex(), "\\\\n")
+            .replace("""(\\\d)""".toRegex(), """\\\\""" + "\$1")
         val statementCount = statement.mapNotNull { element -> element.takeIf { it == ';' } }.size.toString()
         val request = SnowflakeRequest(
             statement = statement,

diff --git a/app/src/test/kotlin/com/getstrm/pace/common/AbstractDynamicViewGeneratorTest.kt b/app/src/test/kotlin/com/getstrm/pace/common/AbstractDynamicViewGeneratorTest.kt
@@ -2,6 +2,7 @@ package com.getstrm.pace.common
 
 import com.getstrm.pace.util.TestDynamicViewGenerator
 import build.buf.gen.getstrm.api.data_policies.v1alpha.DataPolicy
+import com.getstrm.pace.domain.SqlParseException
 import io.kotest.matchers.shouldBe
 import org.junit.jupiter.api.Test
 import org.junit.jupiter.api.assertThrows

diff --git a/scripts/pp/upsert-data-policy.sh b/scripts/pp/upsert-data-policy.sh
@@ -5,7 +5,5 @@ data_policy=$(json-case $f)
 request=$(echo '{}' | jq -r --argjson p "$data_policy" '{data_policy: $p}')
 
 echo $request | evans -r cli \
-    --header "authorization=Bearer $(strm auth print-access-token)" \
-    --host api.dev.getstrm.io --tls --port 443 \
     --package getstrm.api.data_policies.v1alpha --service DataPolicyService \
     call UpsertDataPolicy