kagkarlsson#491 - Eagerly load DBSchedulerStarter

NicklasWallgren · May 20, 2024 · d751b37 · d751b37
1 parent 823c9d2
commit d751b37
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/...ava/com/github/kagkarlsson/scheduler/boot/autoconfigure/DbSchedulerAutoConfiguration.java b/...ava/com/github/kagkarlsson/scheduler/boot/autoconfigure/DbSchedulerAutoConfiguration.java
@@ -40,6 +40,7 @@
 import javax.sql.DataSource;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
+import org.springframework.boot.LazyInitializationExcludeFilter;
 import org.springframework.boot.autoconfigure.AutoConfigurationPackage;
 import org.springframework.boot.autoconfigure.AutoConfigureAfter;
 import org.springframework.boot.autoconfigure.condition.ConditionalOnBean;
@@ -196,6 +197,11 @@ public DbSchedulerStarter dbSchedulerStarter(Scheduler scheduler) {
     return new ImmediateStart(scheduler);
   }
 
+  @Bean
+  public LazyInitializationExcludeFilter eagerDbSchedulerStarter() {
+    return LazyInitializationExcludeFilter.forBeanTypes(DbSchedulerStarter.class);
+  }
+
   private static DataSource configureDataSource(DataSource existingDataSource) {
     if (existingDataSource instanceof TransactionAwareDataSourceProxy) {
       log.debug("Using an already transaction aware DataSource");

diff --git a/...com/github/kagkarlsson/scheduler/boot/autoconfigure/DbSchedulerAutoConfigurationTest.java b/...com/github/kagkarlsson/scheduler/boot/autoconfigure/DbSchedulerAutoConfigurationTest.java
@@ -29,6 +29,7 @@
 import org.junit.jupiter.api.Test;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
+import org.springframework.boot.LazyInitializationExcludeFilter;
 import org.springframework.boot.actuate.autoconfigure.health.HealthContributorAutoConfiguration;
 import org.springframework.boot.actuate.autoconfigure.metrics.CompositeMeterRegistryAutoConfiguration;
 import org.springframework.boot.actuate.autoconfigure.metrics.MetricsAutoConfiguration;
@@ -241,6 +242,17 @@ public void it_should_use_custom_stats_registry_if_present_in_context() {
             });
   }
 
+  @Test
+  void it_should_exclude_db_scheduler_starter_from_lazy_init() {
+    ctxRunner.run(
+        (context) -> {
+          LazyInitializationExcludeFilter filter =
+              context.getBean(LazyInitializationExcludeFilter.class);
+
+          assertThat(filter.isExcluded(null, null, DbSchedulerStarter.class)).isTrue();
+        });
+  }
+
   @Configuration
   static class SingleTaskConfiguration {
     @Bean