needed access to entity when use with partial JPA/cross-store persistence

spring-data-commons-core/src/main/java/org/springframework/data/persistence/ChangeSetPersister.java

@@ -1,47 +1,49 @@
-package org.springframework.data.persistence;
-
-import org.springframework.dao.DataAccessException;
-
-/**
- * Interface to be implemented by classes that can synchronize
- * between data stores and ChangeSets.
- * @author Rod Johnson
- *
- * @param <K> entity key
- */
-public interface ChangeSetPersister<K> {
-	
-	String ID_KEY = "_id";
-
-	String CLASS_KEY = "_class";
-	
-	/**
-	 * TODO how to tell when not found? throw exception?
-	 */
-	void getPersistentState(Class<? extends ChangeSetBacked> entityClass, K key, ChangeSet changeSet) throws DataAccessException, NotFoundException;
-	
-	/**
-	 * Return id
-	 * @param cs 
-	 * @return
-	 * @throws DataAccessException
-	 */
-	K getPersistentId(Class<? extends ChangeSetBacked> entityClass, ChangeSet cs) throws DataAccessException;
-	
-	/**
-	 * Return key
-	 * @param cs Key may be null if not persistent
-	 * @return
-	 * @throws DataAccessException
-	 */
-	K persistState(Class<? extends ChangeSetBacked> entityClass, ChangeSet cs) throws DataAccessException;
-	
-	/**
-	 * Exception thrown in alternate control flow if getPersistentState
-	 * finds no entity data.
-	 */
-	class NotFoundException extends Exception {
-		
-	}
-	
-}
+package org.springframework.data.persistence;
+
+import org.springframework.dao.DataAccessException;
+
+/**
+ * Interface to be implemented by classes that can synchronize
+ * between data stores and ChangeSets.
+ * @author Rod Johnson
+ *
+ * @param <K> entity key
+ */
+public interface ChangeSetPersister<K> {
+	
+	String ID_KEY = "_id";
+
+	String CLASS_KEY = "_class";
+	
+	/**
+	 * TODO how to tell when not found? throw exception?
+	 */
+	void getPersistentState(Class<? extends ChangeSetBacked> entityClass, K key, ChangeSet changeSet) throws DataAccessException, NotFoundException;
+	
+	/**
+	 * Return id
+	 * @param entity 
+	 * @param cs 
+	 * @return
+	 * @throws DataAccessException
+	 */
+	K getPersistentId(ChangeSetBacked entity, ChangeSet cs) throws DataAccessException;
+	
+	/**
+	 * Return key
+	 * @param entity 
+	 * @param cs Key may be null if not persistent
+	 * @return
+	 * @throws DataAccessException
+	 */
+	K persistState(ChangeSetBacked entity, ChangeSet cs) throws DataAccessException;
+	
+	/**
+	 * Exception thrown in alternate control flow if getPersistentState
+	 * finds no entity data.
+	 */
+	class NotFoundException extends Exception {
+		
+	}
+	
+}

spring-data-commons-core/src/main/java/org/springframework/data/transaction/ChangeSetBackedTransactionSynchronization.java

@@ -23,7 +23,7 @@ public class ChangeSetBackedTransactionSynchronization implements TransactionSyn
 	
 	public void afterCommit() {
 		log.debug("After Commit called for " + entity);
-		changeSetPersister.persistState(entity.getClass(), entity.getChangeSet());
+		changeSetPersister.persistState(entity, entity.getChangeSet());
 		changeSetTxStatus = 0;
 	}