DATACOUCH-102

Pass event source to onBeforeDelete() and onAfterDelete() event handlers
joelstevenson · Aug 13, 2014 · 275d92a · 275d92a
1 parent 014a755
commit 275d92a
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 16 deletions.
diff --git a/...org/springframework/data/couchbase/core/mapping/event/AbstractCouchbaseEventListener.java b/...org/springframework/data/couchbase/core/mapping/event/AbstractCouchbaseEventListener.java
@@ -41,17 +41,18 @@ public AbstractCouchbaseEventListener() {
 
   @SuppressWarnings("rawtypes")
   public void onApplicationEvent(CouchbaseMappingEvent<?> event) {
+
+	E source = (E) event.getSource();
+	// Check for matching domain type and invoke callbacks
+	if (source != null && !domainClass.isAssignableFrom(source.getClass())) {
+		return;
+	}
+
     if (event instanceof BeforeDeleteEvent) {
-      onBeforeDelete(event.getDocument());
+      onBeforeDelete(event.getSource(), event.getDocument());
       return;
     } else if (event instanceof AfterDeleteEvent) {
-      onAfterDelete(event.getDocument());
-      return;
-    }
-
-    E source = (E) event.getSource();
-    // Check for matching domain type and invoke callbacks
-    if (source != null && !domainClass.isAssignableFrom(source.getClass())) {
+      onAfterDelete(event.getSource(), event.getDocument());
       return;
     }
 
@@ -82,13 +83,13 @@ public void onAfterSave(E source, CouchbaseDocument doc) {
     }
   }
 
-  public void onAfterDelete(CouchbaseDocument doc) {
+  public void onAfterDelete(Object source, CouchbaseDocument doc) {
     if (LOG.isDebugEnabled()) {
       LOG.debug("onAfterConvert({})", doc);
     }
   }
 
-  public void onBeforeDelete(CouchbaseDocument doc) {
+  public void onBeforeDelete(Object source, CouchbaseDocument doc) {
     if (LOG.isDebugEnabled()) {
       LOG.debug("onAfterConvert({})", doc);
     }

diff --git a/...main/java/org/springframework/data/couchbase/core/mapping/event/LoggingEventListener.java b/...main/java/org/springframework/data/couchbase/core/mapping/event/LoggingEventListener.java
@@ -31,14 +31,10 @@ public class LoggingEventListener extends AbstractCouchbaseEventListener<Object>
   private static final Logger LOGGER = LoggerFactory.getLogger(LoggingEventListener.class);
 
   @Override
-  public void onBeforeDelete(CouchbaseDocument doc) {
-    LOGGER.info("onBeforeDelete: {}", doc);
-  }
+  public void onBeforeDelete(Object source, CouchbaseDocument doc) { LOGGER.info("onBeforeDelete: {}, {}", source, doc); }
 
   @Override
-  public void onAfterDelete(CouchbaseDocument doc) {
-    LOGGER.info("onAfterDelete: {}", doc);
-  }
+  public void onAfterDelete(Object source, CouchbaseDocument doc) { LOGGER.info("onAfterDelete: {} {}", source, doc); }
 
   @Override
   public void onAfterSave(Object source, CouchbaseDocument doc) {