Java 8流groupby並計算多個屬性

Question

我有一個具有日期和布爾錯誤指示符的對象Process。 我想獲取每個日期的總進程數和有錯誤的進程數。 例如，Jun 01的計數為2、1； Jun 02將具有1，0和Jun 03 1，1。我能夠做到這一點的唯一方法是兩次流式傳輸以獲取計數。 我嘗試實現自定義收集器，但沒有成功。 是否有一種優雅的解決方案代替我的笨拙方法？

    final SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd");
    final List<Process> processes = new ArrayList<>();
    processes.add(new Process(sdf.parse("2016-06-01"), false));
    processes.add(new Process(sdf.parse("2016-06-01"), true));
    processes.add(new Process(sdf.parse("2016-06-02"), false));
    processes.add(new Process(sdf.parse("2016-06-03"), true));

    System.out.println(processes.stream()
          .collect(
                  Collectors.groupingBy(Process::getDate, Collectors.counting()) ));

    System.out.println(processes.stream().filter(order -> order.isHasError())
              .collect(
                      Collectors.groupingBy(Process::getDate, Collectors.counting()) ));

private class Process  {
    private Date date;
    private boolean hasError;

    public Process(Date date, boolean hasError) {
        this.date = date;
        this.hasError = hasError;
    }

    public Date getDate() {
        return date;
    }

    public boolean isHasError() {
        return hasError;
    }
}

@ glee8e解決方案和@Holger提示后的代碼

Collector<Process, Result, Result> ProcessCollector = Collector.of(
        () -> Result::new,
        (r, p) -> {
            r.increment(0);
            if (p.isHasError()) {
                r.increment(1);
            }
        }, (r1, r2) -> {
            r1.add(0, r2.get(0));
            r1.add(1, r2.get(1));
            return r1;
});

Map<Date, Result> results = Processs.stream().collect(groupingBy(Process::getDate, ProcessCollector));
results.entrySet().stream().sorted(Comparator.comparing(Entry::getKey)).forEach(entry -> System.out
        .println(String.format("date = %s, %s", sdf.format(entry.getKey()), entry.getValue())));



private class Result {

    private AtomicIntegerArray array = new AtomicIntegerArray(2);

    public int get(int index) {
        return array.get(index);
    }

    public void increment(int index) {
        array.getAndIncrement(index);
    }

    public void add(int index, int delta) {
        array.addAndGet(index, delta);
    }

    @Override
    public String toString() {
        return String.format("totalProcesses = %d, totalErrors = %d", array.get(0), array.get(1));
    }
}

Answer 1

最好添加一個POJO來存儲結果，否則合並器功能可能看起來有點晦澀。 我宣布POJO為公開，但是如果您認為隱藏起來更好，則可以對其進行更改。

public class Result {
     public int all, error;
}

主要代碼：

// Add it somewhere in this file.
private static final Set <Characteristics> CH_ID = Collections.unmodifiableSet(EnumSet.of(Collector.Characteristics.IDENTITY_FINISH));

//...
// This is main processing code
processes.stream().collect(collectingAndThen(groupingBy(Process::getDate, new Collector<Process, Result, Result> {
            @Override
            public Supplier<Result> supplier() {
                return Result::new;
            }

            @Override
            public BiConsumer<Process, Result> accumlator() {
                return (p, r) -> {
                    r.total++;
                    if (p.isHasError())
                        r.error++;
                };
            }

            @Override
            public BinaryOperator<Result> combiner() {
                return (r1, r2) -> {
                    r1.total += r2.total;
                    r1.error += r2.error;
                    return r1;
                };
            }

            @Override
            public Function<Result, Result> finisher() {
                return Function.identity();
            }

            @Override
            public Set<Characteristics> characteristics() {
                return CH_ID;
            }
})));

PS：我假設您已import static java.util.stream.Collectors