Fix some dependency issues with NonRootBeanPostProcessor

Author: Quinn-With-Two-Ns

temporal-spring-boot-autoconfigure/src/main/java/io/temporal/spring/boot/autoconfigure/NonRootBeanPostProcessor.java

@@ -33,6 +33,7 @@
 import org.springframework.beans.BeansException;
 import org.springframework.beans.factory.BeanFactory;
 import org.springframework.beans.factory.BeanFactoryAware;
+import org.springframework.beans.factory.BeanNotOfRequiredTypeException;
 import org.springframework.beans.factory.NoSuchBeanDefinitionException;
 import org.springframework.beans.factory.config.BeanPostProcessor;
 import org.springframework.beans.factory.config.ConfigurableListableBeanFactory;
@@ -45,26 +46,30 @@ public class NonRootBeanPostProcessor implements BeanPostProcessor, BeanFactoryA
 
   private final @Nonnull TemporalProperties temporalProperties;
   private final @Nullable List<NonRootNamespaceProperties> namespaceProperties;
-  private final @Nullable Tracer tracer;
-  private final @Nullable TestWorkflowEnvironmentAdapter testWorkflowEnvironment;
-  private final @Nullable Scope metricsScope;
-
-  public NonRootBeanPostProcessor(
-      @Nonnull TemporalProperties temporalProperties,
-      @Nullable Tracer tracer,
-      @Nullable TestWorkflowEnvironmentAdapter testWorkflowEnvironment,
-      @Nullable Scope metricsScope) {
+  private @Nullable Tracer tracer;
+  private @Nullable TestWorkflowEnvironmentAdapter testWorkflowEnvironment;
+  private @Nullable Scope metricsScope;
+
+  public NonRootBeanPostProcessor(@Nonnull TemporalProperties temporalProperties) {
     this.temporalProperties = temporalProperties;
     this.namespaceProperties = temporalProperties.getNamespaces();
-    this.tracer = tracer;
-    this.testWorkflowEnvironment = testWorkflowEnvironment;
-    this.metricsScope = metricsScope;
   }
 
   @Override
-  public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {
+  public Object postProcessAfterInitialization(@Nonnull Object bean, @Nonnull String beanName)
+      throws BeansException {
     if (bean instanceof NamespaceTemplate && beanName.equals("temporalRootNamespaceTemplate")) {
       if (namespaceProperties != null) {
+        // If there are non-root namespaces, we need to inject beans for each of them. Look
+        // up the bean manually instead of using @Autowired to avoid circular dependencies or causing the dependency to
+        // get initialized to early and skip post-processing.
+        //
+        // Note: We don't use @Lazy here because these dependencies are optional and @Lazy doesn't interact well with
+        // optional dependencies.
+        metricsScope = findBean("temporalMetricsScope", Scope.class);
+        tracer = findBean(Tracer.class);
+        testWorkflowEnvironment =
+            findBean("temporalTestWorkflowEnvironment", TestWorkflowEnvironmentAdapter.class);
         namespaceProperties.forEach(this::injectBeanByNonRootNamespace);
       }
     }
@@ -163,6 +168,24 @@ private <T> T findBeanByNamespace(String beanPrefix, Class<T> clazz) {
     return null;
   }
 
+  private <T> @Nullable T findBean(Class<T> clazz) {
+    try {
+      return beanFactory.getBean(clazz);
+    } catch (NoSuchBeanDefinitionException | BeanNotOfRequiredTypeException ignore) {
+      // Ignore if the bean is not found or not of the required type
+    }
+    return null;
+  }
+
+  private <T> @Nullable T findBean(String beanName, Class<T> clazz) {
+    try {
+      return beanFactory.getBean(beanName, clazz);
+    } catch (NoSuchBeanDefinitionException | BeanNotOfRequiredTypeException ignore) {
+      // Ignore if the bean is not found or not of the required type
+    }
+    return null;
+  }
+
   private <T> TemporalOptionsCustomizer<T> findBeanByNameSpaceForTemporalCustomizer(
       String beanPrefix, Class<T> genericOptionsBuilderClass) {
     String beanName =

temporal-spring-boot-autoconfigure/src/main/java/io/temporal/spring/boot/autoconfigure/NonRootNamespaceAutoConfiguration.java

@@ -1,12 +1,9 @@
 package io.temporal.spring.boot.autoconfigure;
 
 import com.google.common.base.MoreObjects;
-import com.uber.m3.tally.Scope;
-import io.opentracing.Tracer;
 import io.temporal.spring.boot.autoconfigure.properties.NamespaceProperties;
 import io.temporal.spring.boot.autoconfigure.properties.NonRootNamespaceProperties;
 import io.temporal.spring.boot.autoconfigure.properties.TemporalProperties;
-import io.temporal.spring.boot.autoconfigure.template.TestWorkflowEnvironmentAdapter;
 import io.temporal.spring.boot.autoconfigure.template.WorkersTemplate;
 import java.util.List;
 import java.util.Optional;
@@ -15,8 +12,6 @@
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.beans.BeansException;
-import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.beans.factory.annotation.Qualifier;
 import org.springframework.boot.autoconfigure.AutoConfigureAfter;
 import org.springframework.boot.autoconfigure.condition.ConditionalOnBean;
 import org.springframework.boot.autoconfigure.condition.ConditionalOnExpression;
@@ -36,27 +31,21 @@
 @AutoConfigureAfter({RootNamespaceAutoConfiguration.class, ServiceStubsAutoConfiguration.class})
 @ConditionalOnBean(ServiceStubsAutoConfiguration.class)
 @ConditionalOnExpression(
-    "${spring.temporal.test-server.enabled:false} || '${spring.temporal.connection.target:}'.length() > 0")
+    "(${spring.temporal.test-server.enabled:false} || '${spring.temporal.connection.target:}'.length() > 0) && ('${spring.temporal.namespaces:}'.length() > 0)")
 public class NonRootNamespaceAutoConfiguration {
 
   protected static final Logger log =
       LoggerFactory.getLogger(NonRootNamespaceAutoConfiguration.class);
 
   @Bean
-  public NonRootBeanPostProcessor nonRootBeanPostProcessor(
-      TemporalProperties properties,
-      @Autowired(required = false) @Nullable Tracer otTracer,
-      @Qualifier("temporalTestWorkflowEnvironmentAdapter") @Autowired(required = false) @Nullable
-          TestWorkflowEnvironmentAdapter testWorkflowEnvironment,
-      @Qualifier("temporalMetricsScope") @Autowired(required = false) @Nullable
-          Scope metricsScope) {
-    return new NonRootBeanPostProcessor(
-        properties, otTracer, testWorkflowEnvironment, metricsScope);
+  public static NonRootBeanPostProcessor nonRootBeanPostProcessor(
+      @Lazy TemporalProperties properties) {
+    return new NonRootBeanPostProcessor(properties);
   }
 
   @Bean
-  public NonRootNamespaceEventListener nonRootNamespaceEventListener(
-      TemporalProperties temporalProperties,
+  public static NonRootNamespaceEventListener nonRootNamespaceEventListener(
+      @Lazy TemporalProperties temporalProperties,
       @Nullable @Lazy List<WorkersTemplate> workersTemplates) {
     return new NonRootNamespaceEventListener(temporalProperties, workersTemplates);
   }