From a155dde743eea7280e28ca049c70de399a898abf Mon Sep 17 00:00:00 2001
From: Nikita Koksharov <nkoksharov@redisson.pro>
Date: Mon, 28 Jun 2021 10:24:23 +0300
Subject: [PATCH] refactoring

---
 .../org/redisson/cluster/ClusterConnectionManager.java    | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

diff --git a/redisson/src/main/java/org/redisson/cluster/ClusterConnectionManager.java b/redisson/src/main/java/org/redisson/cluster/ClusterConnectionManager.java
index cc5703ce3..56425310a 100644
--- a/redisson/src/main/java/org/redisson/cluster/ClusterConnectionManager.java
+++ b/redisson/src/main/java/org/redisson/cluster/ClusterConnectionManager.java
@@ -661,10 +661,8 @@ public class ClusterConnectionManager extends MasterSlaveConnectionManager {
         Set<ClusterPartition> clusterLastPartitions = getLastPartitions();
 
         // https://github.com/redisson/redisson/issues/3635
-        Map<String, MasterSlaveEntry> nodeEntry = new HashMap<>();
-        clusterLastPartitions.forEach(partition -> {
-            nodeEntry.put(partition.getNodeId(), getEntry(partition.slots().nextSetBit(0)));
-        });
+        Map<String, MasterSlaveEntry> nodeEntries = clusterLastPartitions.stream().collect(Collectors.toMap(p -> p.getNodeId(),
+                                                                                    p -> getEntry(p.slots().nextSetBit(0))));
 
         Set<Integer> changedSlots = new HashSet<>();
         for (ClusterPartition currentPartition : clusterLastPartitions) {
@@ -674,7 +672,7 @@ public class ClusterConnectionManager extends MasterSlaveConnectionManager {
                     continue;
                 }
 
-                MasterSlaveEntry entry = nodeEntry.get(nodeId);
+                MasterSlaveEntry entry = nodeEntries.get(nodeId);
                 BitSet addedSlots = newPartition.copySlots();
                 addedSlots.andNot(currentPartition.slots());
                 currentPartition.addSlots(addedSlots);