fix: Support other custom repository for javax

line · Jun 10, 2024 · 5346774 · 5346774
1 parent b5dadd3
commit 5346774
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/...rt/spring/data/jpa/javax/autoconfigure/KotlinJdslJpaRepositoryFactoryBeanPostProcessor.kt b/...rt/spring/data/jpa/javax/autoconfigure/KotlinJdslJpaRepositoryFactoryBeanPostProcessor.kt
@@ -16,10 +16,15 @@ open class KotlinJdslJpaRepositoryFactoryBeanPostProcessor : BeanPostProcessor {
     lateinit var kotlinJdslJpqlExecutor: KotlinJdslJpqlExecutor
 
     override fun postProcessBeforeInitialization(bean: Any, beanName: String): Any? {
-        if (bean is JpaRepositoryFactoryBean<*, *, *>) {
+        if (bean is JpaRepositoryFactoryBean<*, *, *> && bean.hasJdsl()) {
             bean.setCustomImplementation(kotlinJdslJpqlExecutor)
         }
 
         return super.postProcessAfterInitialization(bean, beanName)
     }
+
+    private fun JpaRepositoryFactoryBean<*, *, *>.hasJdsl(): Boolean {
+        return this.objectType.interfaces
+            .any { it == KotlinJdslJpqlExecutor::class.java }
+    }
 }