Github user aweisberg commented on a diff in the pull request:
https://github.com/apache/cassandra/pull/224#discussion_r188116936
--- Diff: src/java/org/apache/cassandra/locator/Replica.java ---
@@ -0,0 +1,221 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements. See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership. The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.cassandra.locator;
+
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.Collections;
+import java.util.List;
+import java.util.Objects;
+import java.util.Set;
+
+import com.google.common.base.Preconditions;
+import com.google.common.collect.Iterables;
+import com.google.common.collect.Lists;
+import com.google.common.collect.Sets;
+
+import org.apache.cassandra.db.ConsistencyLevel;
+import org.apache.cassandra.dht.Range;
+import org.apache.cassandra.dht.Token;
+import org.apache.cassandra.exceptions.UnavailableException;
+
+/**
+ * Decorated Endpoint
+ */
+public class Replica
+{
+ private final InetAddressAndPort endpoint;
+ private final Range<Token> range;
+ private final boolean full;
+
+ public Replica(InetAddressAndPort endpoint, Range<Token> range,
boolean full)
+ {
+ Preconditions.checkNotNull(endpoint);
+ this.endpoint = endpoint;
+ this.range = range;
+ this.full = full;
+ }
+
+ public Replica(InetAddressAndPort endpoint, Token start, Token end,
boolean full)
+ {
+ this(endpoint, new Range<>(start, end), full);
+ }
+
+ public boolean equals(Object o)
+ {
+ if (this == o) return true;
+ if (o == null || getClass() != o.getClass()) return false;
+ Replica replica = (Replica) o;
+ return full == replica.full &&
+ Objects.equals(endpoint, replica.endpoint) &&
+ Objects.equals(range, replica.range);
+ }
+
+ public int hashCode()
+ {
+
+ return Objects.hash(endpoint, range, full);
+ }
+
+ @Override
+ public String toString()
+ {
+ StringBuilder sb = new StringBuilder();
+ sb.append(full ? "Full" : "Transient");
+
sb.append('(').append(getEndpoint()).append(',').append(range).append(')');
+ return sb.toString();
+ }
+
+ public final InetAddressAndPort getEndpoint()
+ {
+ return endpoint;
+ }
+
+ public Range<Token> getRange()
+ {
+ return range;
+ }
+
+ public boolean isFull()
+ {
+ return full;
+ }
+
+ public final boolean isTransient()
+ {
+ return !isFull();
+ }
+
+ public ReplicaSet subtract(Replica that)
+ {
+ assert isFull() && that.isFull(); // FIXME: this
+ Set<Range<Token>> ranges = range.subtract(that.range);
+ ReplicaSet replicatedRanges = new ReplicaSet(ranges.size());
+ for (Range<Token> range: ranges)
+ {
+ replicatedRanges.add(new Replica(getEndpoint(), range,
isFull()));
+ }
+ return replicatedRanges;
+ }
+
+ /**
+ * Subtract the ranges of the given replicas from the range of this
replica,
+ * returning a set of replicas with the endpoint and transient
information of
+ * this replica, and the ranges resulting from the subtraction.
+ */
+ public ReplicaSet subtractByRange(Replicas toSubtract)
+ {
+ if (isFull() && Iterables.all(toSubtract, Replica::isFull))
+ {
+ Set<Range<Token>> subtractedRanges =
getRange().subtractAll(toSubtract.asRangeSet());
--- End diff --
Range.subtractAll could take an iterable here and then we could use a lazy
transformation.
---
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
