diff --git a/spring-data-cassandra/src/main/java/org/springframework/data/cassandra/core/mapping/CassandraMappingContext.java b/spring-data-cassandra/src/main/java/org/springframework/data/cassandra/core/mapping/CassandraMappingContext.java index f06d54099..ff7a7d8b6 100644 --- a/spring-data-cassandra/src/main/java/org/springframework/data/cassandra/core/mapping/CassandraMappingContext.java +++ b/spring-data-cassandra/src/main/java/org/springframework/data/cassandra/core/mapping/CassandraMappingContext.java @@ -17,12 +17,11 @@ package org.springframework.data.cassandra.core.mapping; import java.util.Collection; import java.util.Collections; -import java.util.concurrent.ConcurrentHashMap; -import java.util.HashSet; import java.util.Map; import java.util.Objects; import java.util.Optional; import java.util.Set; +import java.util.concurrent.ConcurrentHashMap; import java.util.stream.StreamSupport; import org.springframework.beans.BeansException; @@ -81,8 +80,9 @@ public class CassandraMappingContext // caches private final Map>> entitySetsByTableName = new ConcurrentHashMap<>(); - private final Set> tableEntities = new HashSet<>(); - private final Set> userDefinedTypes = new HashSet<>(); + private final Set> tableEntities = ConcurrentHashMap.newKeySet(); + + private final Set> userDefinedTypes = ConcurrentHashMap.newKeySet(); /** * Create a new {@link CassandraMappingContext}. @@ -345,7 +345,7 @@ public class CassandraMappingContext // now do some caching of the entity Set> entities = this.entitySetsByTableName.computeIfAbsent(entity.getTableName(), - cqlIdentifier -> new HashSet<>()); + cqlIdentifier -> ConcurrentHashMap.newKeySet()); entities.add(entity);