From 39d792de07dd6fbc85c20b71905d7f6d45640d14 Mon Sep 17 00:00:00 2001
From: Nicklas Wallgren <nicklas.wallgren@gmail.com>
Date: Mon, 20 May 2024 07:53:50 +0200
Subject: [PATCH] #491 - Eagerly load DBSchedulerStarter

---
 .../autoconfigure/DbSchedulerAutoConfiguration.java   |  6 ++++++
 .../DbSchedulerAutoConfigurationTest.java             | 11 +++++++++++
 2 files changed, 17 insertions(+)

diff --git a/db-scheduler-boot-starter/src/main/java/com/github/kagkarlsson/scheduler/boot/autoconfigure/DbSchedulerAutoConfiguration.java b/db-scheduler-boot-starter/src/main/java/com/github/kagkarlsson/scheduler/boot/autoconfigure/DbSchedulerAutoConfiguration.java
index 3162dc7a..9dc6f138 100644
--- a/db-scheduler-boot-starter/src/main/java/com/github/kagkarlsson/scheduler/boot/autoconfigure/DbSchedulerAutoConfiguration.java
+++ b/db-scheduler-boot-starter/src/main/java/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/db-scheduler-boot-starter/src/test/java/com/github/kagkarlsson/scheduler/boot/autoconfigure/DbSchedulerAutoConfigurationTest.java b/db-scheduler-boot-starter/src/test/java/com/github/kagkarlsson/scheduler/boot/autoconfigure/DbSchedulerAutoConfigurationTest.java
index 6487db4a..93338dad 100644
--- a/db-scheduler-boot-starter/src/test/java/com/github/kagkarlsson/scheduler/boot/autoconfigure/DbSchedulerAutoConfigurationTest.java
+++ b/db-scheduler-boot-starter/src/test/java/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,16 @@ 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