From 51b662efcf89bf708031365ecd90b272c3cf2ac1 Mon Sep 17 00:00:00 2001
From: "jonghyon.s" <jonghyon.s@linecorp.com>
Date: Thu, 7 Mar 2024 16:19:30 +0900
Subject: [PATCH] fix: support custom repository

---
 .../KotlinJdslJpaRepositoryFactoryBeanPostProcessor.kt         | 3 +--
 .../KotlinJdslJpaRepositoryFactoryBeanPostProcessor.kt         | 3 +--
 2 files changed, 2 insertions(+), 4 deletions(-)

diff --git a/support/spring-data-jpa-javax/src/main/kotlin/com/linecorp/kotlinjdsl/support/spring/data/jpa/javax/autoconfigure/KotlinJdslJpaRepositoryFactoryBeanPostProcessor.kt b/support/spring-data-jpa-javax/src/main/kotlin/com/linecorp/kotlinjdsl/support/spring/data/jpa/javax/autoconfigure/KotlinJdslJpaRepositoryFactoryBeanPostProcessor.kt
index 37bc79f4a..98041697f 100644
--- a/support/spring-data-jpa-javax/src/main/kotlin/com/linecorp/kotlinjdsl/support/spring/data/jpa/javax/autoconfigure/KotlinJdslJpaRepositoryFactoryBeanPostProcessor.kt
+++ b/support/spring-data-jpa-javax/src/main/kotlin/com/linecorp/kotlinjdsl/support/spring/data/jpa/javax/autoconfigure/KotlinJdslJpaRepositoryFactoryBeanPostProcessor.kt
@@ -6,7 +6,6 @@ import org.springframework.beans.factory.annotation.Autowired
 import org.springframework.beans.factory.config.BeanPostProcessor
 import org.springframework.context.annotation.Lazy
 import org.springframework.data.jpa.repository.support.JpaRepositoryFactoryBean
-import org.springframework.data.repository.core.support.RepositoryComposition.RepositoryFragments
 import org.springframework.stereotype.Component
 
 @Component
@@ -18,7 +17,7 @@ open class KotlinJdslJpaRepositoryFactoryBeanPostProcessor : BeanPostProcessor {
 
     override fun postProcessBeforeInitialization(bean: Any, beanName: String): Any? {
         if (bean is JpaRepositoryFactoryBean<*, *, *>) {
-            bean.setRepositoryFragments(RepositoryFragments.just(kotlinJdslJpqlExecutor))
+            bean.setCustomImplementation(kotlinJdslJpqlExecutor)
         }
 
         return super.postProcessAfterInitialization(bean, beanName)
diff --git a/support/spring-data-jpa/src/main/kotlin/com/linecorp/kotlinjdsl/support/spring/data/jpa/autoconfigure/KotlinJdslJpaRepositoryFactoryBeanPostProcessor.kt b/support/spring-data-jpa/src/main/kotlin/com/linecorp/kotlinjdsl/support/spring/data/jpa/autoconfigure/KotlinJdslJpaRepositoryFactoryBeanPostProcessor.kt
index 2647a24fb..d6ea66cc9 100644
--- a/support/spring-data-jpa/src/main/kotlin/com/linecorp/kotlinjdsl/support/spring/data/jpa/autoconfigure/KotlinJdslJpaRepositoryFactoryBeanPostProcessor.kt
+++ b/support/spring-data-jpa/src/main/kotlin/com/linecorp/kotlinjdsl/support/spring/data/jpa/autoconfigure/KotlinJdslJpaRepositoryFactoryBeanPostProcessor.kt
@@ -6,7 +6,6 @@ import org.springframework.beans.factory.annotation.Autowired
 import org.springframework.beans.factory.config.BeanPostProcessor
 import org.springframework.context.annotation.Lazy
 import org.springframework.data.jpa.repository.support.JpaRepositoryFactoryBean
-import org.springframework.data.repository.core.support.RepositoryComposition.RepositoryFragments
 import org.springframework.stereotype.Component
 
 @Component
@@ -18,7 +17,7 @@ open class KotlinJdslJpaRepositoryFactoryBeanPostProcessor : BeanPostProcessor {
 
     override fun postProcessBeforeInitialization(bean: Any, beanName: String): Any? {
         if (bean is JpaRepositoryFactoryBean<*, *, *>) {
-            bean.setRepositoryFragments(RepositoryFragments.just(kotlinJdslJpqlExecutor))
+            bean.setCustomImplementation(kotlinJdslJpqlExecutor)
         }
 
         return super.postProcessAfterInitialization(bean, beanName)