docs: private Map<Integer, Integer> getNodeIdToPrimaryCount(Cluster cluster) {
          Map<Integer, Integer> nodeIdToPrimaryCount = Maps.newHashMap();
          for(Node node: cluster.getNodes()) {
              nodeIdToPrimaryCount.put(node.getId(), node.getPartitionIds().size());
          }
  
          return nodeIdToPrimaryCount;
      }