JedisByteMapAggregator.java

package redis.clients.jedis.executors.aggregators;

import redis.clients.jedis.util.JedisByteMap;

import java.util.ArrayList;
import java.util.List;

class JedisByteMapAggregator<T> implements Aggregator<JedisByteMap<T>, JedisByteMap<T>> {

  // Parts stores references to the maps added to the aggregator.
  // Defines the initial capacity of the list holding the parts.
  // Hard to come up with a reasonable default.
  // Start with 3 as min redis cluster has 3 masters.
  private static final int INITIAL_CAPACITY = 3;

  private List<JedisByteMap<T>> parts;

  @Override
  public void add(JedisByteMap<T> map) {
    if (map == null) {
      return;
    }

    if (parts == null) {
      parts = new ArrayList<>(INITIAL_CAPACITY);
    }

    parts.add(map);
  }

  @Override
  public JedisByteMap<T> getResult() {
    if (parts == null) {
      return null;
    }

    // Fast path: only one non-null map
    if (parts.size() == 1) {
      return parts.get(0);
    }

    JedisByteMap<T> result = new JedisByteMap<>();

    for (JedisByteMap<T> part : parts) {
      result.putAll(part);
    }

    return result;
  }
}