code: private Map<Integer, Integer> getNodeIdToNaryCount(Cluster cluster,
                                                         StoreRoutingPlan storeRoutingPlan) {
          Map<Integer, Integer> nodeIdToNaryCount = Maps.newHashMap();
  
          for(int nodeId: cluster.getNodeIds()) {
              nodeIdToNaryCount.put(nodeId, storeRoutingPlan.getZoneNAryPartitionIds(nodeId).size());
          }
  
          return nodeIdToNaryCount;
      }
docs: Go through all node IDs and determine which node
  
  @param cluster
  @param storeRoutingPlan
  @return