Merge pull request #379 from b9r5/AddToInfected2

Adds to infected when gossip received from 2nd and following senders
scalecube · Dec 1, 2022 · b7d8714 · b7d8714
2 parents 170d23b + b2ebaa9
commit b7d8714
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/cluster/src/main/java/io/scalecube/cluster/gossip/GossipProtocolImpl.java b/cluster/src/main/java/io/scalecube/cluster/gossip/GossipProtocolImpl.java
@@ -203,13 +203,15 @@ private void onGossipRequest(Message message) {
     final long period = this.currentPeriod;
     final GossipRequest gossipRequest = message.data();
     for (Gossip gossip : gossipRequest.gossips()) {
+      GossipState gossipState = gossips.get(gossip.gossipId());
       if (ensureSequence(gossip.gossiperId()).add(gossip.sequenceId())) {
-        GossipState gossipState = gossips.get(gossip.gossipId());
         if (gossipState == null) { // new gossip
           gossipState = new GossipState(gossip, period);
           gossips.put(gossip.gossipId(), gossipState);
           sink.emitNext(gossip.message(), RETRY_NON_SERIALIZED);
         }
+      }
+      if (gossipState != null) {
         gossipState.addToInfected(gossipRequest.from());
       }
     }