Avoid startup bean processing warnings

wolfiabot · Jan 2, 2025 · f019ad7 · f019ad7
1 parent b7eaefe
commit f019ad7
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/src/main/java/space/npstr/wolfia/db/DataSourceProxyPostProcessor.kt b/src/main/java/space/npstr/wolfia/db/DataSourceProxyPostProcessor.kt
@@ -7,19 +7,23 @@ import net.ttddyy.dsproxy.listener.logging.DefaultQueryLogEntryCreator
 import net.ttddyy.dsproxy.listener.logging.SLF4JLogLevel.WARN
 import net.ttddyy.dsproxy.listener.logging.SLF4JSlowQueryListener
 import net.ttddyy.dsproxy.support.ProxyDataSourceBuilder
+import org.springframework.beans.factory.ObjectProvider
 import org.springframework.beans.factory.config.BeanPostProcessor
 import org.springframework.stereotype.Component
 
 @Component
 class DataSourceProxyPostProcessor(
-	queryCountHolder: SingleQueryCountHolder,
+	// Use ObjectProvider to avoid warnings about this bean being initialized too early for post processing
+	queryCountHolder: ObjectProvider<SingleQueryCountHolder>,
 ) : BeanPostProcessor {
 
-	private val proxyDataSourceBuilder = ProxyDataSourceBuilder()
-		.listener(buildSlf4jSlowQueryListener())
-		.multiline()
-		.name("postgres")
-		.countQuery(queryCountHolder)
+	private val proxyDataSourceBuilder by lazy {
+		ProxyDataSourceBuilder()
+			.listener(buildSlf4jSlowQueryListener())
+			.multiline()
+			.name("postgres")
+			.countQuery(queryCountHolder.getObject())
+	}
 
 	override fun postProcessAfterInitialization(bean: Any, beanName: String): Any? {
 //		if (bean is HikariDataSource) {