diff --git a/spring-data-mongodb/src/main/java/org/springframework/data/mongodb/core/index/MongoPersistentEntityIndexCreator.java b/spring-data-mongodb/src/main/java/org/springframework/data/mongodb/core/index/MongoPersistentEntityIndexCreator.java index a0d8d2bbd..3400eef99 100644 --- a/spring-data-mongodb/src/main/java/org/springframework/data/mongodb/core/index/MongoPersistentEntityIndexCreator.java +++ b/spring-data-mongodb/src/main/java/org/springframework/data/mongodb/core/index/MongoPersistentEntityIndexCreator.java @@ -18,6 +18,7 @@ package org.springframework.data.mongodb.core.index; import java.lang.reflect.Field; import java.util.Collections; +import java.util.Map; import java.util.Set; import java.util.concurrent.ConcurrentHashMap; @@ -49,7 +50,7 @@ public class MongoPersistentEntityIndexCreator implements private static final Log log = LogFactory.getLog(MongoPersistentEntityIndexCreator.class); - private final Set> classesSeen = Collections.newSetFromMap(new ConcurrentHashMap, Boolean>()); + private final Map, Boolean> classesSeen = new ConcurrentHashMap, Boolean>(); private final MongoDbFactory mongoDbFactory; /** @@ -81,7 +82,7 @@ public class MongoPersistentEntityIndexCreator implements protected void checkForIndexes(final MongoPersistentEntity entity) { final Class type = entity.getType(); - if (!classesSeen.contains(type)) { + if (!classesSeen.containsKey(type)) { if (log.isDebugEnabled()) { log.debug("Analyzing class " + type + " for index information."); } @@ -144,7 +145,7 @@ public class MongoPersistentEntityIndexCreator implements } }); - classesSeen.add(type); + classesSeen.put(type, true); } }