diff --git a/CHANGES.txt b/CHANGES.txt index 74c44f8006..0f7cf0ecd2 100644 --- a/CHANGES.txt +++ b/CHANGES.txt @@ -1,4 +1,5 @@ 2.2.8 + * Enable repair -pr and -local together (fix regression of CASSANDRA-7450) (CASSANDRA-12522) * Fail repair on non-existing table (CASSANDRA-12279) * cqlsh copy: fix missing counter values (CASSANDRA-12476) * Move migration tasks to non-periodic queue, assure flush executor shutdown after non-periodic executor (CASSANDRA-12251) diff --git a/src/java/org/apache/cassandra/repair/messages/RepairOption.java b/src/java/org/apache/cassandra/repair/messages/RepairOption.java index d50a2edf66..44a1e57ed6 100644 --- a/src/java/org/apache/cassandra/repair/messages/RepairOption.java +++ b/src/java/org/apache/cassandra/repair/messages/RepairOption.java @@ -208,7 +208,7 @@ public class RepairOption { throw new IllegalArgumentException("Too many job threads. Max is " + MAX_JOB_THREADS); } - if (primaryRange && (!dataCenters.isEmpty() || !hosts.isEmpty())) + if (primaryRange && ((!dataCenters.isEmpty() && !option.isInLocalDCOnly()) || !hosts.isEmpty())) { throw new IllegalArgumentException("You need to run primary range repair on all nodes in the cluster."); } @@ -303,6 +303,10 @@ public class RepairOption return isSubrangeRepair; } + public boolean isInLocalDCOnly() { + return dataCenters.size() == 1 && dataCenters.contains(DatabaseDescriptor.getLocalDataCenter()); + } + @Override public String toString() { diff --git a/src/java/org/apache/cassandra/service/StorageService.java b/src/java/org/apache/cassandra/service/StorageService.java index e8ab54ebe2..48a291be7b 100644 --- a/src/java/org/apache/cassandra/service/StorageService.java +++ b/src/java/org/apache/cassandra/service/StorageService.java @@ -2801,7 +2801,7 @@ public class StorageService extends NotificationBroadcasterSupport implements IE if (option.getDataCenters().isEmpty() && option.getHosts().isEmpty()) option.getRanges().addAll(getPrimaryRanges(keyspace)); // except dataCenters only contain local DC (i.e. -local) - else if (option.getDataCenters().size() == 1 && option.getDataCenters().contains(DatabaseDescriptor.getLocalDataCenter())) + else if (option.isInLocalDCOnly()) option.getRanges().addAll(getPrimaryRangesWithinDC(keyspace)); else throw new IllegalArgumentException("You need to run primary range repair on all nodes in the cluster."); diff --git a/test/unit/org/apache/cassandra/repair/messages/RepairOptionTest.java b/test/unit/org/apache/cassandra/repair/messages/RepairOptionTest.java index cc6f46a8fc..a0eea4e48d 100644 --- a/test/unit/org/apache/cassandra/repair/messages/RepairOptionTest.java +++ b/test/unit/org/apache/cassandra/repair/messages/RepairOptionTest.java @@ -96,6 +96,26 @@ public class RepairOptionTest assertEquals(expectedHosts, option.getHosts()); } + @Test + public void testPrWithLocalParseOptions() + { + DatabaseDescriptor.forceStaticInitialization(); + + Map options = new HashMap<>(); + options.put(RepairOption.PARALLELISM_KEY, "parallel"); + options.put(RepairOption.PRIMARY_RANGE_KEY, "true"); + options.put(RepairOption.INCREMENTAL_KEY, "false"); + options.put(RepairOption.COLUMNFAMILIES_KEY, "cf1,cf2,cf3"); + options.put(RepairOption.DATACENTERS_KEY, "datacenter1"); + + RepairOption option = RepairOption.parse(options, Murmur3Partitioner.instance); + assertTrue(option.isPrimaryRange()); + + Set expectedDCs = new HashSet<>(3); + expectedDCs.add("datacenter1"); + assertEquals(expectedDCs, option.getDataCenters()); + } + @Test public void testIncrementalRepairWithSubrangesIsNotGlobal() throws Exception {