自定义classloader问题

当项目中使用了自定义classloader的时候，可以通过设置classloader上下文的方式来使得自己的mapper class能够被找到
abel533 · Jan 22, 2018 · c15579a · c15579a
1 parent 7afd35d
commit c15579a
Showing 1 changed file with 21 additions and 4 deletions.
diff --git a/src/main/java/tk/mybatis/mapper/util/MsUtil.java b/src/main/java/tk/mybatis/mapper/util/MsUtil.java
@@ -44,11 +44,28 @@ public static Class<?> getMapperClass(String msId) {
             throw new MapperException("当前MappedStatement的id=" + msId + ",不符合MappedStatement的规则!");
         }
         String mapperClassStr = msId.substring(0, msId.lastIndexOf("."));
-        try {
-            return Class.forName(mapperClassStr);
-        } catch (ClassNotFoundException e) {
-            return null;
+        ClassLoader[] classLoader = getClassLoaders();
+        Class<?> mapperClass = null;
+        for (ClassLoader cl : classLoader) {
+            if (null != cl) {
+                try {
+                    mapperClass = Class.forName(mapperClassStr, true, cl);
+                    if (mapperClass != null) {
+                        break;
+                    }
+                } catch (ClassNotFoundException e) {
+                    // we'll ignore this until all class loaders fail to locate the class
+                }
+            }
         }
+        if (mapperClass == null) {
+            throw new MapperException("class loaders failed to locate the class " + mapperClassStr);
+        }
+        return mapperClass;
+    }
+
+    private static ClassLoader[] getClassLoaders() {
+        return new ClassLoader[]{Thread.currentThread().getContextClassLoader(), MsUtil.class.getClassLoader()};
     }
 
     /**