--- old/src/java.base/share/classes/java/util/stream/Collectors.java 2018-09-24 17:22:35.419009000 +0700 +++ new/src/java.base/share/classes/java/util/stream/Collectors.java 2018-09-24 17:22:35.043009000 +0700 @@ -1,5 +1,5 @@ /* - * Copyright (c) 2012, 2017, Oracle and/or its affiliates. All rights reserved. + * Copyright (c) 2012, 2018, Oracle and/or its affiliates. All rights reserved. * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. * * This code is free software; you can redistribute it and/or modify it @@ -1885,6 +1885,102 @@ } /** + * Returns a {@code Collector} that is a composite of two downstream collectors. + * Every element passed to the resulting collector is processed by both downstream + * collectors, then their results are merged using the specified merge function + * into the final result. + * + *

The resulting collector functions do the following: + * + *

+ * + *

The resulting collector is {@link Collector.Characteristics#UNORDERED} if both downstream + * collectors are unordered and {@link Collector.Characteristics#CONCURRENT} if both downstream + * collectors are concurrent. + * + * @param the type of the input elements + * @param the result type of the first collector + * @param the result type of the second collector + * @param the final result type + * @param downstream1 the first downstream collector + * @param downstream2 the second downstream collector + * @param merger the function which merges two results into the single one + * @return a {@code Collector} which aggregates the results of two supplied collectors. + * @since 12 + */ + public static + Collector teeing(Collector downstream1, + Collector downstream2, + BiFunction merger) { + return teeing0(downstream1, downstream2, merger); + } + + private static + Collector teeing0(Collector downstream1, + Collector downstream2, + BiFunction merger) { + Objects.requireNonNull(downstream1, "downstream1"); + Objects.requireNonNull(downstream2, "downstream2"); + Objects.requireNonNull(merger, "merger"); + + Supplier c1Supplier = Objects.requireNonNull(downstream1.supplier(), "downstream1 supplier"); + Supplier c2Supplier = Objects.requireNonNull(downstream2.supplier(), "downstream2 supplier"); + BiConsumer c1Accumulator = + Objects.requireNonNull(downstream1.accumulator(), "downstream1 accumulator"); + BiConsumer c2Accumulator = + Objects.requireNonNull(downstream2.accumulator(), "downstream2 accumulator"); + BinaryOperator c1Combiner = Objects.requireNonNull(downstream1.combiner(), "downstream1 combiner"); + BinaryOperator c2Combiner = Objects.requireNonNull(downstream2.combiner(), "downstream2 combiner"); + Function c1Finisher = Objects.requireNonNull(downstream1.finisher(), "downstream1 finisher"); + Function c2Finisher = Objects.requireNonNull(downstream2.finisher(), "downstream2 finisher"); + + Set characteristics; + Set c1Characteristics = downstream1.characteristics(); + Set c2Characteristics = downstream2.characteristics(); + if (CH_ID.containsAll(c1Characteristics) || CH_ID.containsAll(c2Characteristics)) { + characteristics = CH_NOID; + } else { + EnumSet c = EnumSet.noneOf(Collector.Characteristics.class); + c.addAll(c1Characteristics); + c.retainAll(c2Characteristics); + c.remove(Collector.Characteristics.IDENTITY_FINISH); + characteristics = Collections.unmodifiableSet(c); + } + + class PairBox { + A1 left = c1Supplier.get(); + A2 right = c2Supplier.get(); + + void add(T t) { + c1Accumulator.accept(left, t); + c2Accumulator.accept(right, t); + } + + PairBox combine(PairBox other) { + left = c1Combiner.apply(left, other.left); + right = c2Combiner.apply(right, other.right); + return this; + } + + R get() { + R1 r1 = c1Finisher.apply(left); + R2 r2 = c2Finisher.apply(right); + return merger.apply(r1, r2); + } + } + + return new CollectorImpl<>(PairBox::new, PairBox::add, PairBox::combine, PairBox::get, characteristics); + } + + /** * Implementation class used by partitioningBy. */ private static final class Partition