Merge pull request #1 from admintertar/master

兼容spring-native环境下protocol返回的是resource

Author: abel533

src/main/java/io/mybatis/config/defaults/VersionConfig.java

@@ -18,10 +18,16 @@
 
 import io.mybatis.config.Config;
 import io.mybatis.config.ConfigHelper;
+import org.springframework.core.io.Resource;
+import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
+import org.springframework.core.io.support.ResourcePatternResolver;
 
 import java.io.*;
 import java.net.JarURLConnection;
+import java.net.URI;
 import java.net.URL;
+import java.nio.file.Path;
+import java.nio.file.Paths;
 import java.util.*;
 import java.util.function.Function;
 import java.util.jar.JarEntry;
@@ -230,6 +236,42 @@ private Properties chooseFromFile(File file, String version) throws IOException
     });
   }
 
+  /**
+   * 选择版本
+   */
+  private Properties chooseFromResource(String version) throws IOException {
+
+    Map<String, URI> fileMap = new HashMap<>();
+
+    ResourcePatternResolver resolver = new PathMatchingResourcePatternResolver();
+    Resource[] resources = resolver.getResources("classpath*:io/mybatis/provider/config/*.properties");
+    for (Resource resource : resources) {
+      URI uri = resource.getURI();
+
+      String urlString = uri.toString();
+      if (urlString.endsWith(FILE_TYPE)) {
+        Path path = Paths.get(uri);
+        // 获取文件名
+        Path fileName = path.getFileName();
+        fileMap.put(fileName.toString(), uri);
+      }
+    }
+
+    List<ConfigVersion> versions = sortVersions(new ArrayList<>(fileMap.keySet()));
+    ConfigVersion chooseVersion = chooseVersion(versions, version);
+
+    return build(versions, chooseVersion, configVersion -> {
+      URI url = fileMap.get(chooseVersion.getFileName());
+      InputStream inputStream = null;
+      try {
+        inputStream = url.toURL().openStream();
+      } catch (IOException e) {
+        return null;
+      }
+      return inputStream;
+    });
+  }
+
   /**
    * 获取版本配置
    */
@@ -263,6 +305,12 @@ protected Properties buildVersionProperties() {
       } catch (IOException e) {
         throw new RuntimeException(e);
       }
+    } else if (resource.getProtocol().equals("resource")) {
+      try {
+        return chooseFromResource(version);
+      } catch (Exception e) {
+        throw new RuntimeException(e);
+      }
     }
     return null;
   }