Return inferTokenOwnership as soon as a mismatch is there.

Netflix · Jul 10, 2020 · 2881978 · 2881978
1 parent a25e6ef
commit 2881978
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/priam/src/main/java/com/netflix/priam/identity/token/TokenRetrieverUtils.java b/priam/src/main/java/com/netflix/priam/identity/token/TokenRetrieverUtils.java
@@ -78,7 +78,7 @@ public static InferredTokenOwnership inferTokenOwnerFromGossip(
                     if (matchedGossipInstances == noOfInstancesGossipShouldMatch) {
                         inferredTokenOwnership.setTokenInformationStatus(
                                 InferredTokenOwnership.TokenInformationStatus.GOOD);
-                        break;
+                        return inferredTokenOwnership;
                     }
                 } else {
                     // Mismatch in the gossip information from Cassandra.
@@ -92,7 +92,7 @@ public static InferredTokenOwnership inferTokenOwnerFromGossip(
                             inferredTokenOwnership.getTokenInformation().isLive
                                     ? inferredTokenOwnership.getTokenInformation()
                                     : tokenInformation);
-                    break;
+                    return inferredTokenOwnership;
                 }
 
             } catch (GossipParseException e) {