Fix #2 remove IdentitySet in order to reuse existing objet

arey · Mar 23, 2016 · 7cd175e · 7cd175e
1 parent 4713c23
commit 7cd175e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 71 deletions.
diff --git a/src/main/java/com/javaetmoi/javabean/JavaBeanMarshaller.java b/src/main/java/com/javaetmoi/javabean/JavaBeanMarshaller.java
@@ -3,7 +3,6 @@
 import com.javaetmoi.javabean.bean.Item;
 import com.javaetmoi.javabean.bean.SetterParam;
 import com.javaetmoi.javabean.generator.*;
-import com.javaetmoi.javabean.util.IdentitySet;
 import com.squareup.javapoet.JavaFile;
 import com.squareup.javapoet.MethodSpec;
 import com.squareup.javapoet.TypeSpec;
@@ -38,13 +37,13 @@
  */
 public class JavaBeanMarshaller {
 
-    private Map<Object, String> variables = new HashMap<Object, String>();
+    private Map<Object, String> variables = new HashMap<>();
 
     private final TypeSpec.Builder clazz;
 
     private final MethodSpec.Builder method;
 
-    private Set<Object> processedBeans = new IdentitySet<Object>();
+    private Set<Object> processedBeans = new HashSet<>();
 
     private final static Logger LOG = LoggerFactory.getLogger(JavaBeanMarshaller.class);
 

diff --git a/src/main/java/com/javaetmoi/javabean/util/IdentitySet.java b/src/main/java/com/javaetmoi/javabean/util/IdentitySet.java