support rsa key

snowflakedb · Dec 13, 2023 · 920e6de · 920e6de
1 parent 063f8e2
commit 920e6de
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/fips-pom.xml b/fips-pom.xml
@@ -588,6 +588,7 @@
                                 --add-opens=java.base/java.util=ALL-UNNAMED
                                 --add-exports=java.base/sun.nio.ch=ALL-UNNAMED
                                 --add-exports=jdk.unsupported/sun.misc=ALL-UNNAMED
+                                --add-opens=java.base/sun.security.util=ALL-UNNAMED
                             </argLine>
                         </configuration>
                     </plugin>

diff --git a/pom.xml b/pom.xml
@@ -620,6 +620,7 @@
                                 --add-opens=java.base/java.util=ALL-UNNAMED
                                 --add-exports=java.base/sun.nio.ch=ALL-UNNAMED
                                 --add-exports=jdk.unsupported/sun.misc=ALL-UNNAMED
+                                --add-opens=java.base/sun.security.util=ALL-UNNAMED
                             </argLine>
                         </configuration>
                     </plugin>

diff --git a/src/main/scala/com/snowflake/snowpark/internal/UDXRegistrationHandler.scala b/src/main/scala/com/snowflake/snowpark/internal/UDXRegistrationHandler.scala
@@ -983,7 +983,7 @@ class UDXRegistrationHandler(session: Session) extends Logging {
     } else ""
     val createUdfQuery = s"CREATE $tempType " +
       s"FUNCTION $udfName($sqlFunctionArgs) RETURNS " +
-      s"$returnSqlType LANGUAGE JAVA $getRuntimeVersion" +
+      s"$returnSqlType LANGUAGE JAVA $getRuntimeVersion " +
       s"$importSql HANDLER='$className.$methodName' " +
       s"target_path='$targetJarStageLocation' " + packageSql +
       "AS $$ \n" + code + "\n$$"