From 971d62568eb0d4f0eb528fffaf65e08c33ab819b Mon Sep 17 00:00:00 2001
From: alex-semenyuk <alexemenyuk88@gmail.com>
Date: Sat, 15 Aug 2015 16:34:42 +0200
Subject: [PATCH] Added xml support for event listener and made minor cleanup

---
 .../java/org/baeldung/config/MongoConfig.java   | 17 ++++++++---------
 ....java => CascadeSaveMongoEventListener.java} |  4 ++--
 ...RefFieldCallback.java => FieldCallback.java} |  2 +-
 .../java/org/baeldung/model/EmailAddress.java   |  8 ++++----
 .../src/main/java/org/baeldung/model/User.java  |  9 ++++-----
 .../src/main/resources/mongoConfig.xml          | 11 ++++++++---
 6 files changed, 27 insertions(+), 24 deletions(-)
 rename spring-data-mongodb/src/main/java/org/baeldung/event/{CascadingMongoEventListener.java => CascadeSaveMongoEventListener.java} (88%)
 rename spring-data-mongodb/src/main/java/org/baeldung/event/{DbRefFieldCallback.java => FieldCallback.java} (86%)

diff --git a/spring-data-mongodb/src/main/java/org/baeldung/config/MongoConfig.java b/spring-data-mongodb/src/main/java/org/baeldung/config/MongoConfig.java
index 8433e67850..4b776af5b6 100644
--- a/spring-data-mongodb/src/main/java/org/baeldung/config/MongoConfig.java
+++ b/spring-data-mongodb/src/main/java/org/baeldung/config/MongoConfig.java
@@ -3,9 +3,8 @@ package org.baeldung.config;
 import java.util.ArrayList;
 import java.util.List;
 
-
 import org.baeldung.converter.UserWriterConverter;
-import org.baeldung.event.CascadingMongoEventListener;
+import org.baeldung.event.CascadeSaveMongoEventListener;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
 import org.springframework.data.mongodb.config.AbstractMongoConfiguration;
@@ -20,7 +19,7 @@ import com.mongodb.MongoClient;
 @EnableMongoRepositories(basePackages = "org.baeldung.repository")
 public class MongoConfig extends AbstractMongoConfiguration {
 
-    private List<Converter<?,?>> converters = new ArrayList<Converter<?,?>>();
+    private List<Converter<?, ?>> converters = new ArrayList<Converter<?, ?>>();
 
     @Override
     protected String getDatabaseName() {
@@ -36,15 +35,15 @@ public class MongoConfig extends AbstractMongoConfiguration {
     public String getMappingBasePackage() {
         return "org.baeldung";
     }
-    
+
     @Bean
-    public CascadingMongoEventListener cascadingMongoEventListener(){
-        return new CascadingMongoEventListener();
+    public CascadeSaveMongoEventListener cascadingMongoEventListener() {
+        return new CascadeSaveMongoEventListener();
     }
-    
+
     @Override
     public CustomConversions customConversions() {
-       converters.add(new UserWriterConverter());
-       return new CustomConversions(converters);
+        converters.add(new UserWriterConverter());
+        return new CustomConversions(converters);
     }
 }
diff --git a/spring-data-mongodb/src/main/java/org/baeldung/event/CascadingMongoEventListener.java b/spring-data-mongodb/src/main/java/org/baeldung/event/CascadeSaveMongoEventListener.java
similarity index 88%
rename from spring-data-mongodb/src/main/java/org/baeldung/event/CascadingMongoEventListener.java
rename to spring-data-mongodb/src/main/java/org/baeldung/event/CascadeSaveMongoEventListener.java
index e461160eb4..ad09f1ac04 100644
--- a/spring-data-mongodb/src/main/java/org/baeldung/event/CascadingMongoEventListener.java
+++ b/spring-data-mongodb/src/main/java/org/baeldung/event/CascadeSaveMongoEventListener.java
@@ -9,7 +9,7 @@ import org.springframework.util.ReflectionUtils;
 
 import java.lang.reflect.Field;
 
-public class CascadingMongoEventListener extends AbstractMongoEventListener<Object> {
+public class CascadeSaveMongoEventListener extends AbstractMongoEventListener<Object> {
     @Autowired
     private MongoOperations mongoOperations;
 
@@ -24,7 +24,7 @@ public class CascadingMongoEventListener extends AbstractMongoEventListener<Obje
                     final Object fieldValue = field.get(source);
 
                     if (fieldValue != null) {
-                        DbRefFieldCallback callback = new DbRefFieldCallback();
+                        FieldCallback callback = new FieldCallback();
 
                         ReflectionUtils.doWithFields(fieldValue.getClass(), callback);
 
diff --git a/spring-data-mongodb/src/main/java/org/baeldung/event/DbRefFieldCallback.java b/spring-data-mongodb/src/main/java/org/baeldung/event/FieldCallback.java
similarity index 86%
rename from spring-data-mongodb/src/main/java/org/baeldung/event/DbRefFieldCallback.java
rename to spring-data-mongodb/src/main/java/org/baeldung/event/FieldCallback.java
index 66f797042a..868bbc31f0 100644
--- a/spring-data-mongodb/src/main/java/org/baeldung/event/DbRefFieldCallback.java
+++ b/spring-data-mongodb/src/main/java/org/baeldung/event/FieldCallback.java
@@ -5,7 +5,7 @@ import java.lang.reflect.Field;
 import org.springframework.data.annotation.Id;
 import org.springframework.util.ReflectionUtils;
 
-public class DbRefFieldCallback implements ReflectionUtils.FieldCallback {
+public class FieldCallback implements ReflectionUtils.FieldCallback {
     private boolean idFound;
 
     public void doWith(Field field) throws IllegalArgumentException, IllegalAccessException {
diff --git a/spring-data-mongodb/src/main/java/org/baeldung/model/EmailAddress.java b/spring-data-mongodb/src/main/java/org/baeldung/model/EmailAddress.java
index f54aea0e78..095ce40c01 100644
--- a/spring-data-mongodb/src/main/java/org/baeldung/model/EmailAddress.java
+++ b/spring-data-mongodb/src/main/java/org/baeldung/model/EmailAddress.java
@@ -8,12 +8,12 @@ public class EmailAddress {
     @Id
     private String id;
     private String value;
-    
-    public EmailAddress(){
+
+    public EmailAddress() {
     }
-    
+
     public EmailAddress(String value) {
-        this.value =  value;
+        this.value = value;
     }
 
     public String getId() {
diff --git a/spring-data-mongodb/src/main/java/org/baeldung/model/User.java b/spring-data-mongodb/src/main/java/org/baeldung/model/User.java
index ff4eacd0c2..70d97d3417 100644
--- a/spring-data-mongodb/src/main/java/org/baeldung/model/User.java
+++ b/spring-data-mongodb/src/main/java/org/baeldung/model/User.java
@@ -20,17 +20,16 @@ public class User {
     private String name;
 
     private Integer age;
-    
+
     @DBRef
     @Field("email")
     @CascadeSave
     private EmailAddress emailAddress;
-    
-    public User(){
+
+    public User() {
     }
 
-    
-    public User(String name, Integer age, String value){
+    public User(String name, Integer age, String value) {
         this.name = name;
         this.age = age;
         this.emailAddress = new EmailAddress(value);
diff --git a/spring-data-mongodb/src/main/resources/mongoConfig.xml b/spring-data-mongodb/src/main/resources/mongoConfig.xml
index 1bda9d0303..7772b351e5 100644
--- a/spring-data-mongodb/src/main/resources/mongoConfig.xml
+++ b/spring-data-mongodb/src/main/resources/mongoConfig.xml
@@ -10,21 +10,26 @@
        http://www.springframework.org/schema/context/spring-context-3.2.xsd"
 >
     <context:annotation-config/>
-    
+
     <bean id="mongo" class="org.springframework.data.mongodb.core.MongoFactoryBean">
         <property name="host" value="localhost"/>
     </bean>
 
     <bean id="mongoTemplate" class="org.springframework.data.mongodb.core.MongoTemplate">
         <constructor-arg name="mongo" ref="mongo"/>
+        <constructor-arg ref="mongoConverter" />
         <constructor-arg name="databaseName" value="test"/>
     </bean>
 
     <bean class="org.springframework.dao.annotation.PersistenceExceptionTranslationPostProcessor"/>
 
     <mongo:repositories base-package="org.baeldung.repository" mongo-template-ref="mongoTemplate"/>
-    
-    <bean class="org.baeldung.event.BeforeConvertListener">
+
+    <bean class="org.baeldung.event.CascadingMongoEventListener">
     </bean>
 
+    <mongo:mapping-converter id="mongoConverter" base-package="org.baeldung.converter">
+        <mongo:custom-converters base-package="org.baeldung.converter"/>
+    </mongo:mapping-converter>
+
 </beans>
\ No newline at end of file