diff --git a/src/java/org/apache/cassandra/service/Rebuild.java b/src/java/org/apache/cassandra/service/Rebuild.java index 70da6c031e..732a0a3b7f 100644 --- a/src/java/org/apache/cassandra/service/Rebuild.java +++ b/src/java/org/apache/cassandra/service/Rebuild.java @@ -233,10 +233,9 @@ public class Rebuild private static MovementMap movementMap(ClusterMetadata metadata, String keyspace, String tokens) { MovementMap.Builder movementMapBuilder = MovementMap.builder(); - DataPlacements placements = metadata.placements(); if (keyspace == null) { - placements.forEach((params, placement) -> movementMapBuilder.put(params, addMovementsForParams(placement, null))); + metadata.placements().forEach((params, placement) -> movementMapBuilder.put(params, addMovementsForParams(placement, null))); // Special case to include the system metadata keyspace ReplicationParams metaParams = Keyspace.open(METADATA_KEYSPACE_NAME).getMetadata().params.replication; movementMapBuilder.put(metaParams, addMovementsForParams(metadata.placement(metaParams), null)); @@ -244,13 +243,13 @@ public class Rebuild else if (tokens == null) { ReplicationParams params = Keyspace.open(keyspace).getMetadata().params.replication; - movementMapBuilder.put(params, addMovementsForParams(placements.get(params), null)); + movementMapBuilder.put(params, addMovementsForParams(metadata.placement(params), null)); } else { ReplicationParams params = Keyspace.open(keyspace).getMetadata().params.replication; RangesAtEndpoint ranges = rangesForRebuildWithTokens(tokens, keyspace); - movementMapBuilder.put(params, addMovementsForParams(placements.get(params), ranges)); + movementMapBuilder.put(params, addMovementsForParams(metadata.placement(params), ranges)); } return movementMapBuilder.build(); }