[samza] Use concurrent hashmap for key schema str

majisourav99 · Mar 3, 2025 · 0d0b1f5 · 0d0b1f5
1 parent 1243dfe
commit 0d0b1f5
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/integrations/venice-samza/src/main/java/com/linkedin/venice/samza/VeniceSystemProducer.java b/integrations/venice-samza/src/main/java/com/linkedin/venice/samza/VeniceSystemProducer.java
@@ -38,7 +38,6 @@
 import com.linkedin.venice.serialization.avro.SchemaPresenceChecker;
 import com.linkedin.venice.serializer.FastSerializerDeserializerFactory;
 import com.linkedin.venice.serializer.RecordSerializer;
-import com.linkedin.venice.utils.BoundedHashMap;
 import com.linkedin.venice.utils.Pair;
 import com.linkedin.venice.utils.PartitionUtils;
 import com.linkedin.venice.utils.SystemTime;
@@ -136,7 +135,7 @@ public class VeniceSystemProducer implements SystemProducer, Closeable {
   private Schema keySchema;
   private String canonicalKeySchemaStr;
   // To avoid the excessive usage of the cache in case each message is using a unique key schema
-  private final Map<Schema, String> canonicalSchemaStrCache = new BoundedHashMap<>(10, true);
+  private final Map<Schema, String> canonicalSchemaStrCache = new VeniceConcurrentHashMap<>(10);
 
   private D2Client primaryControllerColoD2Client;
   private D2Client childColoD2Client;