Fun with classloaders

openrewrite · Sep 7, 2020 · 344f1ff · 344f1ff
1 parent 563bf12
commit 344f1ff
Showing 1 changed file with 28 additions and 4 deletions.
diff --git a/rewrite-core/src/main/java/org/openrewrite/config/AutoConfigureRefactorVisitorLoader.java b/rewrite-core/src/main/java/org/openrewrite/config/AutoConfigureRefactorVisitorLoader.java
@@ -24,16 +24,22 @@
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
+import java.io.UncheckedIOException;
 import java.lang.reflect.Constructor;
+import java.net.MalformedURLException;
+import java.net.URL;
+import java.net.URLClassLoader;
 import java.nio.file.Path;
 import java.util.ArrayList;
 import java.util.Collection;
 import java.util.List;
 import java.util.Map;
+import java.util.stream.Collectors;
 
 import static java.util.Collections.emptyList;
 import static java.util.Collections.emptyMap;
 import static java.util.stream.Collectors.toList;
+import static java.util.stream.StreamSupport.stream;
 
 public class AutoConfigureRefactorVisitorLoader implements ResourceLoader {
     private static final Logger logger = LoggerFactory.getLogger(AutoConfigureRefactorVisitorLoader.class);
@@ -49,20 +55,38 @@ public AutoConfigureRefactorVisitorLoader(Iterable<Path> compileClasspath, Strin
     public Collection<? extends RefactorVisitor<?>> loadVisitors() {
         List<RefactorVisitor<?>> visitors = new ArrayList<>(loadVisitors(new ClassGraph()));
 
-        if(compileClasspath.iterator().hasNext()) {
-            visitors.addAll(loadVisitors(new ClassGraph().overrideClasspath(compileClasspath)));
+        if (compileClasspath.iterator().hasNext()) {
+            URLClassLoader classpathLoader = new URLClassLoader(
+                    stream(compileClasspath.spliterator(), false)
+                            .map(cc -> {
+                                try {
+                                    return cc.toUri().toURL();
+                                } catch (MalformedURLException e) {
+                                    throw new UncheckedIOException(e);
+                                }
+                            })
+                            .toArray(URL[]::new),
+                    getClass().getClassLoader()
+            );
+
+            visitors.addAll(loadVisitors(new ClassGraph()
+                    .ignoreParentClassLoaders()
+                    .overrideClassLoaders(
+                            classpathLoader
+                    )
+            ));
         }
 
         return visitors;
     }
 
     @NotNull
     private List<RefactorVisitor<?>> loadVisitors(ClassGraph classGraph) {
-        if(acceptVisitorPackages != null && acceptVisitorPackages.length > 0) {
+        if (acceptVisitorPackages != null && acceptVisitorPackages.length > 0) {
             classGraph = classGraph.acceptPackages(acceptVisitorPackages);
         }
 
-        try(ScanResult scanResult = classGraph
+        try (ScanResult scanResult = classGraph
                 .enableMemoryMapping()
                 .enableAnnotationInfo()
                 .ignoreClassVisibility()