Avoid thread-unsafe use of ProgressToken

2 files changed, 7 insertions, 7 deletions

diff --git a/clustercontroller-reindexer/src/main/java/ai/vespa/reindexing/Reindexing.java b/clustercontroller-reindexer/src/main/java/ai/vespa/reindexing/Reindexing.java
index 1b5a685b69c..896b9dfc26e 100644
--- a/clustercontroller-reindexer/src/main/java/ai/vespa/reindexing/Reindexing.java
+++ b/clustercontroller-reindexer/src/main/java/ai/vespa/reindexing/Reindexing.java
@@ -72,11 +72,11 @@ public class Reindexing {
 
         private final Instant startedAt;
         private final Instant endedAt;
-        private final ProgressToken progress;
+        private final String progress;
         private final State state;
         private final String message;
 
-        Status(Instant startedAt, Instant endedAt, ProgressToken progress, State state, String message) {
+        Status(Instant startedAt, Instant endedAt, String progress, State state, String message) {
             this.startedAt = startedAt;
             this.endedAt = endedAt;
             this.progress = progress;
@@ -100,7 +100,9 @@ public class Reindexing {
         public Status progressed(ProgressToken progress) {
             if (state != State.RUNNING)
                 throw new IllegalStateException("Current state must be RUNNING when updating progress");
-            return new Status(startedAt, null, requireNonNull(progress), state, null);
+            synchronized (progress) {
+                return new Status(startedAt, null, progress.serializeToString(), state, null);
+            }
         }
 
         /** Returns a copy of this in state HALTED. */
@@ -133,7 +135,7 @@ public class Reindexing {
         }
 
         public Optional<ProgressToken> progress() {
-            return Optional.ofNullable(progress);
+            return Optional.ofNullable(progress).map(ProgressToken::fromSerializedString);
         }
 
         public State state() {
diff --git a/clustercontroller-reindexer/src/main/java/ai/vespa/reindexing/ReindexingCurator.java b/clustercontroller-reindexer/src/main/java/ai/vespa/reindexing/ReindexingCurator.java
index d46ab812aca..6074ce99101 100644
--- a/clustercontroller-reindexer/src/main/java/ai/vespa/reindexing/ReindexingCurator.java
+++ b/clustercontroller-reindexer/src/main/java/ai/vespa/reindexing/ReindexingCurator.java
@@ -5,7 +5,6 @@ import ai.vespa.reindexing.Reindexing.Status;
 import com.google.common.util.concurrent.UncheckedTimeoutException;
 import com.yahoo.document.DocumentType;
 import com.yahoo.document.DocumentTypeManager;
-import com.yahoo.documentapi.ProgressToken;
 import com.yahoo.path.Path;
 import com.yahoo.slime.Cursor;
 import com.yahoo.slime.Inspector;
@@ -17,7 +16,6 @@ import com.yahoo.yolean.Exceptions;
 
 import java.time.Duration;
 import java.time.Instant;
-import java.util.Collection;
 import java.util.Map;
 import java.util.function.Function;
 
@@ -126,7 +124,7 @@ public class ReindexingCurator {
                                             .collect(toUnmodifiableMap(object -> require(TYPE, object, field -> types.getDocumentType(field.asString())),
                                                                        object -> new Status(require(STARTED_MILLIS, object, field -> Instant.ofEpochMilli(field.asLong())),
                                                                                             get(ENDED_MILLIS, object, field -> Instant.ofEpochMilli(field.asLong())),
-                                                                                            get(PROGRESS, object, field -> ProgressToken.fromSerializedString(field.asString())),
+                                                                                            get(PROGRESS, object, field -> field.asString()),
                                                                                             require(STATE, object, field -> toState(field.asString())),
                                                                                             get(MESSAGE, object, field -> field.asString())))));
         }