[MINOR] Avoid NPE when native lib is not found (#668)

kwai · Nov 28, 2024 · 0b3fb9f · 0b3fb9f
1 parent f038f82
commit 0b3fb9f
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/spark-extension/src/main/scala/org/apache/spark/sql/blaze/BlazeCallNativeWrapper.scala b/spark-extension/src/main/scala/org/apache/spark/sql/blaze/BlazeCallNativeWrapper.scala
@@ -199,10 +199,11 @@ object BlazeCallNativeWrapper extends Logging {
       val tempFile = File.createTempFile("libblaze-", ".tmp")
       tempFile.deleteOnExit()
 
-      Utils.tryWithResource(classLoader.getResourceAsStream(libName)) { is =>
+      Utils.tryWithResource {
+        val is = classLoader.getResourceAsStream(libName)
         assert(is != null, s"cannot load $libName")
-        Files.copy(is, tempFile.toPath, StandardCopyOption.REPLACE_EXISTING)
-      }
+        is
+      }(Files.copy(_, tempFile.toPath, StandardCopyOption.REPLACE_EXISTING))
       System.load(tempFile.getAbsolutePath)
 
     } catch {