From b50b4f573b7848bcb463584ccc0c0a6b715f1fcc Mon Sep 17 00:00:00 2001
From: Oliver Gierke <info@olivergierke.de>
Date: Wed, 29 Dec 2010 13:36:05 +0100
Subject: [PATCH] Adapt to latest changes in repository factory base classes.

---
 .../repository/support/JpaRepositoryFactory.java    | 13 ++++++++-----
 .../custom/CustomGenericJpaRepositoryFactory.java   |  5 +++--
 2 files changed, 11 insertions(+), 7 deletions(-)

diff --git a/src/main/java/org/springframework/data/jpa/repository/support/JpaRepositoryFactory.java b/src/main/java/org/springframework/data/jpa/repository/support/JpaRepositoryFactory.java
index d8bb9125f..392c582b7 100644
--- a/src/main/java/org/springframework/data/jpa/repository/support/JpaRepositoryFactory.java
+++ b/src/main/java/org/springframework/data/jpa/repository/support/JpaRepositoryFactory.java
@@ -46,9 +46,10 @@ public class JpaRepositoryFactory extends RepositoryFactorySupport {
      */
     @Override
     protected <T, ID extends Serializable> RepositorySupport<T, ID> getTargetRepository(
-            Class<T> domainClass) {
+            Class<T> domainClass, Class<?> repositoryInterface) {
 
-        return getTargetRepository(domainClass, entityManager);
+        return getTargetRepository(domainClass, repositoryInterface,
+                entityManager);
     }
 
 
@@ -64,7 +65,8 @@ public class JpaRepositoryFactory extends RepositoryFactorySupport {
      * @return
      */
     protected <T, ID extends Serializable> RepositorySupport<T, ID> getTargetRepository(
-            Class<T> domainClass, EntityManager entityManager) {
+            Class<T> domainClass, Class<?> repositoryInterface,
+            EntityManager entityManager) {
 
         return new SimpleJpaRepository<T, ID>(domainClass, entityManager);
     }
@@ -75,11 +77,12 @@ public class JpaRepositoryFactory extends RepositoryFactorySupport {
      * 
      * @see
      * org.springframework.data.repository.support.RepositoryFactorySupport#
-     * getRepositoryClass()
+     * getRepositoryClass(java.lang.Class)
      */
     @Override
     @SuppressWarnings("rawtypes")
-    protected Class<? extends RepositorySupport> getRepositoryClass() {
+    protected Class<? extends RepositorySupport> getRepositoryClass(
+            Class<?> repositoryInterface) {
 
         return SimpleJpaRepository.class;
     }
diff --git a/src/test/java/org/springframework/data/jpa/repository/custom/CustomGenericJpaRepositoryFactory.java b/src/test/java/org/springframework/data/jpa/repository/custom/CustomGenericJpaRepositoryFactory.java
index a81fa4349..5cd2cc32d 100644
--- a/src/test/java/org/springframework/data/jpa/repository/custom/CustomGenericJpaRepositoryFactory.java
+++ b/src/test/java/org/springframework/data/jpa/repository/custom/CustomGenericJpaRepositoryFactory.java
@@ -50,7 +50,7 @@ public class CustomGenericJpaRepositoryFactory extends JpaRepositoryFactory {
      */
     @Override
     protected <T, ID extends Serializable> RepositorySupport<T, ID> getTargetRepository(
-            Class<T> domainClass, EntityManager em) {
+            Class<T> domainClass, Class<?> repositoryInterface, EntityManager em) {
 
         return new CustomGenericJpaRepository<T, ID>(domainClass, em);
     }
@@ -65,7 +65,8 @@ public class CustomGenericJpaRepositoryFactory extends JpaRepositoryFactory {
      */
     @Override
     @SuppressWarnings("rawtypes")
-    protected Class<? extends RepositorySupport> getRepositoryClass() {
+    protected Class<? extends RepositorySupport> getRepositoryClass(
+            Class<?> repositoryInterface) {
 
         return CustomGenericJpaRepository.class;
     }