From 920e6de59ba19524ca67919ffdd73bdcedab5c60 Mon Sep 17 00:00:00 2001
From: Bing Li <bing.li@snowflake.com>
Date: Wed, 13 Dec 2023 11:45:08 -0800
Subject: [PATCH] support rsa key

---
 fips-pom.xml                                                    | 1 +
 pom.xml                                                         | 1 +
 .../snowflake/snowpark/internal/UDXRegistrationHandler.scala    | 2 +-
 3 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/fips-pom.xml b/fips-pom.xml
index b9e539fe..3b2c7fbd 100644
--- 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
index 1de39efe..c12b8455 100644
--- 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
index 77140cd9..92ddf0ec 100644
--- 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$$"