Fix version vector validation when running migration

yorkie-team · Nov 5, 2024 · 9e87b78 · 9e87b78
1 parent 5aaacfb
commit 9e87b78
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/migrations/v0.5.3/add-version-vector.go b/migrations/v0.5.3/add-version-vector.go
@@ -50,16 +50,12 @@ func validateAddVersionVector(ctx context.Context, db *mongo.Client, databaseNam
 		}
 
 		versionVector := info.VersionVector
-		actors, err := versionVector.Keys()
+		actorID, err := info.ActorID.ToActorID()
 		if err != nil {
 			return err
 		}
 
-		if len(actors) > 1 {
-			return fmt.Errorf("found %d actor in version vector", len(actors))
-		}
-
-		if versionVector.VersionOf(actors[0]) != info.Lamport {
+		if versionVector.VersionOf(actorID) != info.Lamport {
 			return fmt.Errorf("wrong lamport in version vector")
 		}