mirror of
https://github.com/dromara/tianai-captcha.git
synced 2026-05-06 21:53:10 +08:00
增加classLoader设置
This commit is contained in:
+10
-1
@@ -14,9 +14,14 @@ public class ClassPathResourceProvider extends AbstractResourceProvider {
|
||||
|
||||
public static final String NAME = "classpath";
|
||||
|
||||
public static ClassLoader classLoader;
|
||||
|
||||
@Override
|
||||
public InputStream doGetResourceInputStream(Resource data) {
|
||||
return getClassLoader().getResourceAsStream(data.getData());
|
||||
if (classLoader == null) {
|
||||
return getClassLoader().getResourceAsStream(data.getData());
|
||||
}
|
||||
return classLoader.getResourceAsStream(data.getData());
|
||||
}
|
||||
|
||||
@Override
|
||||
@@ -39,4 +44,8 @@ public class ClassPathResourceProvider extends AbstractResourceProvider {
|
||||
}
|
||||
return classLoader;
|
||||
}
|
||||
|
||||
public static void setClassLoader(ClassLoader classLoader) {
|
||||
ClassPathResourceProvider.classLoader = classLoader;
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user