[Bug] upload jar file type check bug fixed (#2321)

apache · Feb 12, 2023 · 0c87c6d · 0c87c6d
1 parent 2a4d219
commit 0c87c6d
Showing 1 changed file with 3 additions and 9 deletions.
diff --git a/streampark-common/src/main/scala/org/apache/streampark/common/util/FileUtils.scala b/streampark-common/src/main/scala/org/apache/streampark/common/util/FileUtils.scala
@@ -42,17 +42,11 @@ object FileUtils extends org.apache.commons.io.FileUtils {
     if (input == null) {
       throw new RuntimeException("The inputStream can not be null")
     }
-    val headerHex = Utils.tryWithResource(input) { in =>
-      val b = new Array[Byte](8)
+    Utils.tryWithResource(input) { in =>
+      val b = new Array[Byte](4)
       in.read(b, 0, b.length)
       bytesToHexString(b)
-    }
-    val jarFileTypes: mutable.Seq[String] = mutable.Seq(
-      "504B03040A000000",
-      "504B03040A000008",
-      "504B030414000808"
-    )
-    jarFileTypes.contains(headerHex)
+    } == "504B0304"
   }
 
   def isJarFileType(file: File): Boolean = {