diff options
author | Jon Marius Venstad <venstad@gmail.com> | 2020-09-30 17:42:03 +0200 |
---|---|---|
committer | Jon Marius Venstad <venstad@gmail.com> | 2020-09-30 17:42:03 +0200 |
commit | dc4f51621c0c965f97459c8c7f19aa3062d41567 (patch) | |
tree | a59d79e187e1cc459ccf72c2d72a13dbfa43a148 /testutil | |
parent | c068007feadd8c3f8332c7e80db9ad6e2e2a8093 (diff) |
Make ManualClock thread safe
Diffstat (limited to 'testutil')
-rw-r--r-- | testutil/src/main/java/com/yahoo/test/ManualClock.java | 13 |
1 files changed, 7 insertions, 6 deletions
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(); |