find jar name

Author: yanxi0227

core/src/main/java/com/dtstack/flink/sql/util/PluginUtil.java

@@ -21,13 +21,15 @@
 package com.dtstack.flink.sql.util;
 
 import com.dtstack.flink.sql.classloader.DtClassLoader;
+import org.apache.commons.lang3.StringUtils;
 import org.apache.flink.shaded.jackson2.com.fasterxml.jackson.core.JsonGenerationException;
 import org.apache.flink.shaded.jackson2.com.fasterxml.jackson.core.JsonParseException;
 import org.apache.flink.shaded.jackson2.com.fasterxml.jackson.databind.JsonMappingException;
 import org.apache.flink.shaded.jackson2.com.fasterxml.jackson.databind.ObjectMapper;
 
 import java.io.ByteArrayInputStream;
 import java.io.File;
+import java.io.FilenameFilter;
 import java.io.IOException;
 import java.net.MalformedURLException;
 import java.net.URL;
@@ -105,15 +107,19 @@ public static Properties stringToProperties(String str) throws IOException{
         return properties;
     }
 
-    public static URL getRemoteJarFilePath(String pluginType, String tableType, String remoteSqlRootDir) throws MalformedURLException {
+    public static URL getRemoteJarFilePath(String pluginType, String tableType, String remoteSqlRootDir) throws Exception {
         String dirName = pluginType + tableType.toLowerCase();
-        String jarName = String.format("%s-%s.jar", pluginType, tableType.toLowerCase());
+        String prefix = String.format("%s-%s", pluginType, tableType.toLowerCase());
+        String jarPath = remoteSqlRootDir + SP + dirName;
+        String jarName = getCoreJarFileName(jarPath, prefix);
         return new URL("file:" + remoteSqlRootDir + SP + dirName + SP + jarName);
     }
 
-    public static URL getRemoteSideJarFilePath(String pluginType, String sideOperator, String tableType, String remoteSqlRootDir) throws MalformedURLException {
+    public static URL getRemoteSideJarFilePath(String pluginType, String sideOperator, String tableType, String remoteSqlRootDir) throws Exception {
         String dirName = pluginType + sideOperator + tableType.toLowerCase();
-        String jarName = String.format("%s-%s-%s.jar", pluginType, sideOperator, tableType.toLowerCase());
+        String prefix = String.format("%s-%s-%s", pluginType, sideOperator, tableType.toLowerCase());
+        String jarPath = remoteSqlRootDir + SP + dirName;
+        String jarName = getCoreJarFileName(jarPath, prefix);
         return new URL("file:" + remoteSqlRootDir + SP + dirName + SP + jarName);
     }
 
@@ -138,4 +144,27 @@ public static void addPluginJar(String pluginDir, DtClassLoader classLoader) thr
         }
     }
 
+    public static String getCoreJarFileName (String path, String prefix) throws Exception {
+        String coreJarFileName = null;
+        File pluginDir = new File(path);
+        if (pluginDir.exists() && pluginDir.isDirectory()){
+            File[] jarFiles = pluginDir.listFiles(new FilenameFilter() {
+                @Override
+                public boolean accept(File dir, String name) {
+                    return name.toLowerCase().startsWith(prefix) && name.toLowerCase().endsWith(".jar");
+                }
+            });
+
+            if (jarFiles != null && jarFiles.length > 0){
+                coreJarFileName = jarFiles[0].getName();
+            }
+        }
+
+        if (StringUtils.isEmpty(coreJarFileName)){
+            throw new Exception("Can not find core jar file in path:" + path);
+        }
+
+        return coreJarFileName;
+    }
+
 }

launcher/src/main/java/com/dtstack/flink/sql/launcher/LauncherMain.java

@@ -26,6 +26,7 @@
 import com.dtstack.flink.sql.ClusterMode;
 import com.dtstack.flink.sql.Main;
 import com.dtstack.flink.sql.launcher.perjob.PerJobSubmitter;
+import com.dtstack.flink.sql.util.PluginUtil;
 import org.apache.flink.client.program.ClusterClient;
 import org.apache.flink.client.program.PackagedProgram;
 import org.apache.flink.client.program.PackagedProgramUtils;
@@ -52,13 +53,14 @@
  */
 
 public class LauncherMain {
-    private static final String CORE_JAR = "core.jar";
+    private static final String CORE_JAR = "core";
 
     private static String SP = File.separator;
 
 
-    private static String getLocalCoreJarPath(String localSqlRootJar){
-        return localSqlRootJar + SP + CORE_JAR;
+    private static String getLocalCoreJarPath(String localSqlRootJar) throws Exception {
+        String jarPath = PluginUtil.getCoreJarFileName(localSqlRootJar, CORE_JAR);
+        return jarPath;
     }
 
     public static void main(String[] args) throws Exception {