diff --git a/src/main/java/cloud/tianai/captcha/resource/impl/provider/ClassPathResourceProvider.java b/src/main/java/cloud/tianai/captcha/resource/impl/provider/ClassPathResourceProvider.java index 1b99af6..2bcdf44 100644 --- a/src/main/java/cloud/tianai/captcha/resource/impl/provider/ClassPathResourceProvider.java +++ b/src/main/java/cloud/tianai/captcha/resource/impl/provider/ClassPathResourceProvider.java @@ -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; + } }