JAX-RS Application classloader set as context classloader (#214)

vojtechhabarta · Feb 2, 2018 · d08d86e · d08d86e
1 parent 7d99973
commit d08d86e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/...generator-core/src/main/java/cz/habarta/typescript/generator/JaxrsApplicationScanner.java b/...generator-core/src/main/java/cz/habarta/typescript/generator/JaxrsApplicationScanner.java
@@ -13,7 +13,9 @@
 public class JaxrsApplicationScanner {
 
     public static List<SourceType<Type>> scanJaxrsApplication(Class<?> jaxrsApplicationClass, Predicate<String> isClassNameExcluded) {
+        final ClassLoader originalContextClassLoader = Thread.currentThread().getContextClassLoader();
         try {
+            Thread.currentThread().setContextClassLoader(jaxrsApplicationClass.getClassLoader());
             System.out.println("Scanning JAX-RS application: " + jaxrsApplicationClass.getName());
             final Constructor<?> constructor = jaxrsApplicationClass.getDeclaredConstructor();
             constructor.setAccessible(true);
@@ -27,6 +29,8 @@ public static List<SourceType<Type>> scanJaxrsApplication(Class<?> jaxrsApplicat
             return new JaxrsApplicationScanner().scanJaxrsApplication(jaxrsApplicationClass, resourceClasses, isClassNameExcluded);
         } catch (ReflectiveOperationException e) {
             throw reportError(e);
+        } finally {
+            Thread.currentThread().setContextClassLoader(originalContextClassLoader);
         }
     }