kafka-manager 2.0

kafka-manager-dao/src/main/java/com/xiaojukeji/kafka/manager/vfs/SpringBootVFS.java

@@ -0,0 +1,54 @@
+package com.xiaojukeji.kafka.manager.vfs;
+
+import java.io.IOException;
+import java.io.UncheckedIOException;
+import java.net.URL;
+import java.util.List;
+import java.util.stream.Collectors;
+import java.util.stream.Stream;
+
+import org.apache.ibatis.io.VFS;
+
+import org.springframework.core.io.Resource;
+import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
+import org.springframework.core.io.support.ResourcePatternResolver;
+
+/**
+ * SpringBoot执行jar包无法找到mybatis的domain的解决
+ * @author huangyiminghappy@163.com
+ * @date 2019-04-28
+ */
+public class SpringBootVFS extends VFS {
+    private final ResourcePatternResolver resourceResolver;
+
+    public SpringBootVFS() {
+        this.resourceResolver = new PathMatchingResourcePatternResolver(getClass().getClassLoader());
+    }
+
+    @Override
+    public boolean isValid() {
+        return true;
+    }
+
+    @Override
+    protected List<String> list(URL url, String path) throws IOException {
+        String urlString = url.toString();
+        String baseUrlString = urlString.endsWith("/") ? urlString : urlString.concat("/");
+        Resource[] resources = resourceResolver.getResources(baseUrlString + "**/*.class");
+        return Stream.of(resources)
+                .map(resource -> preserveSubpackageName(baseUrlString, resource, path))
+                .collect(Collectors.toList());
+    }
+
+    private static String preserveSubpackageName(final String baseUrlString,
+                                                 final Resource resource,
+                                                 final String rootPath) {
+        try {
+            return rootPath + (rootPath.endsWith("/") ? "" :  "/")
+                    + resource.getURL().toString().substring(baseUrlString.length());
+        } catch (IOException e) {
+            throw new UncheckedIOException(e);
+        }
+    }
+
+}