From dc4f51621c0c965f97459c8c7f19aa3062d41567 Mon Sep 17 00:00:00 2001
From: Jon Marius Venstad <venstad@gmail.com>
Date: Wed, 30 Sep 2020 17:42:03 +0200
Subject: Make ManualClock thread safe

---
 testutil/src/main/java/com/yahoo/test/ManualClock.java | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

(limited to 'testutil')

diff --git a/testutil/src/main/java/com/yahoo/test/ManualClock.java b/testutil/src/main/java/com/yahoo/test/ManualClock.java
index ffef6895c38..ba7d9698d72 100644
--- a/testutil/src/main/java/com/yahoo/test/ManualClock.java
+++ b/testutil/src/main/java/com/yahoo/test/ManualClock.java
@@ -9,6 +9,7 @@ import java.time.ZoneId;
 import java.time.ZoneOffset;
 import java.time.format.DateTimeFormatter;
 import java.time.temporal.TemporalAmount;
+import java.util.concurrent.atomic.AtomicReference;
 
 /** 
  * A clock which initially has the time of its creation but can only be advanced by calling advance
@@ -17,7 +18,7 @@ import java.time.temporal.TemporalAmount;
  */
 public class ManualClock extends Clock {
 
-    private Instant currentTime = Instant.now();
+    private AtomicReference<Instant> currentTime = new AtomicReference<>(Instant.now());
 
     @Inject
     public ManualClock() {}
@@ -27,19 +28,19 @@ public class ManualClock extends Clock {
     }
 
     public ManualClock(Instant currentTime) {
-        this.currentTime = currentTime;
+        setInstant(currentTime);
     }
 
     public void advance(TemporalAmount temporal) {
-        currentTime = currentTime.plus(temporal);
+        currentTime.updateAndGet(time -> time.plus(temporal));
     }
 
     public void setInstant(Instant time) {
-        currentTime = time;
+        currentTime.set(time);
     }
 
     @Override
-    public Instant instant() { return currentTime; }
+    public Instant instant() { return currentTime.get(); }
 
     @Override
     public ZoneId getZone() { return null; }
@@ -48,7 +49,7 @@ public class ManualClock extends Clock {
     public Clock withZone(ZoneId zone) { return null; }
 
     @Override
-    public long millis() { return currentTime.toEpochMilli(); }
+    public long millis() { return instant().toEpochMilli(); }
 
     public static Instant at(String utcIsoTime) {
         return LocalDateTime.parse(utcIsoTime, DateTimeFormatter.ISO_DATE_TIME).atZone(ZoneOffset.UTC).toInstant();
-- 
cgit v1.2.3