support java 17 in UDF

snowflakedb · Dec 12, 2023 · b0f3274 · b0f3274
1 parent 5456174
commit b0f3274
Showing 1 changed file with 13 additions and 1 deletion.
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,8 @@ class UDXRegistrationHandler(session: Session) extends Logging {
     } else ""
     val createUdfQuery = s"CREATE $tempType " +
       s"FUNCTION $udfName($sqlFunctionArgs) RETURNS " +
-      s"$returnSqlType LANGUAGE JAVA $importSql HANDLER='$className.$methodName' " +
+      s"$returnSqlType LANGUAGE JAVA $getRuntimeVersion" +
+      s"$importSql HANDLER='$className.$methodName' " +
       s"target_path='$targetJarStageLocation' " + packageSql +
       "AS $$ \n" + code + "\n$$"
     logDebug(s"""
@@ -1156,4 +1157,15 @@ class UDXRegistrationHandler(session: Session) extends Logging {
       }
     }
   }
+
+  private def getRuntimeVersion: String = {
+    val version = Utils.JavaVersion
+    if (version.startsWith("17")) {
+      "runtime_version = '17'"
+    } else {
+      // for any other version of JVM, let's use the default jvm, which is java 11.
+      // it is current behavior.
+      ""
+    }
+  }
 }