fix(plugin): include ancestors when loading Spring custom plugin (dat…

…ahub-project#10809)
chriscollins3456 · Jul 1, 2024 · ae4ca4b · ae4ca4b
1 parent 8edc94d
commit ae4ca4b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/...service/plugin/src/main/java/com/datahub/plugins/metadata/aspect/SpringPluginFactory.java b/...service/plugin/src/main/java/com/datahub/plugins/metadata/aspect/SpringPluginFactory.java
@@ -14,6 +14,7 @@
 import javax.annotation.Nonnull;
 import javax.annotation.Nullable;
 import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.BeanFactoryUtils;
 import org.springframework.context.ApplicationContext;
 import org.springframework.context.annotation.AnnotationConfigApplicationContext;
 
@@ -109,7 +110,9 @@ protected <T extends PluginSpec> List<T> build(
           final List<T> plugins;
           if (config.getSpring().getName() == null) {
             plugins =
-                springApplicationContext.getBeansOfType(clazz).values().stream()
+                BeanFactoryUtils.beansOfTypeIncludingAncestors(springApplicationContext, clazz)
+                    .values()
+                    .stream()
                     .map(plugin -> (T) plugin)
                     .collect(Collectors.toList());
           } else {