From ebf56b9cf37fdab6a27636b73651b264f98e21b8 Mon Sep 17 00:00:00 2001
From: Bjørn Christian Seime <bjorncs@yahooinc.com>
Date: Thu, 31 Aug 2023 16:06:23 +0200
Subject: Use thread safe hash map

---
 .../src/main/java/ai/vespa/embedding/EmbedderRuntime.java             | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'model-integration')

diff --git a/model-integration/src/main/java/ai/vespa/embedding/EmbedderRuntime.java b/model-integration/src/main/java/ai/vespa/embedding/EmbedderRuntime.java
index 45068db67f4..4e1442e6383 100644
--- a/model-integration/src/main/java/ai/vespa/embedding/EmbedderRuntime.java
+++ b/model-integration/src/main/java/ai/vespa/embedding/EmbedderRuntime.java
@@ -10,8 +10,8 @@ import com.yahoo.metrics.simple.Gauge;
 import com.yahoo.metrics.simple.MetricReceiver;
 import com.yahoo.metrics.simple.Point;
 
-import java.util.HashMap;
 import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
 
 /**
  * @author bjorncs
@@ -20,7 +20,7 @@ public class EmbedderRuntime implements Embedder.Runtime {
 
     private final Gauge embedLatency;
     private final Gauge sequenceLength;
-    private final Map<MetricDimensions, Point> metricPointCache = new HashMap<>();
+    private final Map<MetricDimensions, Point> metricPointCache = new ConcurrentHashMap<>();
 
     @Inject
     public EmbedderRuntime(MetricReceiver metrics) {
-- 
cgit v1.2.3