来看一个简单的lambda表达式的两种写法
//lambda1
Collections.emptyList().stream().collect(Collectors.counting());
//lambda2
Collections.emptyList().stream().collect(HashMap::new, (map, item) -> map.put(item.getCode(), item.getValue()), HashMap::putAll);
//看一下collect()方法需要传递的参数
//lambda1参数,
<R, A> R collect(Collector<? super T, A, R> collector);
//lambda2参数,
<R> R collect(Supplier<R> supplier,
BiConsumer<R, ? super T> accumulator,
BiConsumer<R, R> combiner);
可以看到, 方法的参数是Collector对象, 因此自定义收集器就是实现Collector接口, 来写自己的逻辑
或者也可以传递三个参数supplier,accumulator,combiner也可以实现
看一下Collector类代码 :
//T : 参数类型
//A : 容器类型
//R : 结果类型
public interface Collector<T, A, R> {
// supplier参数用于生成结果容器,容器类型为A
Supplier<A> supplier();
// accumulator用于消费元素,也就是归纳元素,这里的T就是元素,它会将流中的元素一个一个与结果容器A发生操作
BiConsumer<A, T> accumulator();
// combiner用于两个两个合并并行执行的线程的执行结果,将其合并为一个最终结果A
BinaryOperator<A> combiner();
// finisher用于将之前整合完的结果R转换成为A
Function<A, R> finisher();
// characteristics表示当前Collector的特征值,这是个不可变Set
Set<Characteristics> characteristics();
//构造方法1, 不需要传finisher();, 容器类型和结果类型相同, 例如toList()
public static<T, R> Collector<T, R, R> of(Supplier<R> supplier,
BiConsumer<R, T> accumulator,
BinaryOperator<R> combiner,
Characteristics... characteristics) {
Objects.requireNonNull(supplier);
Objects.requireNonNull(accumulator);
Objects.requireNonNull(combiner);
Objects.requireNonNull(characteristics);
Set<Characteristics> cs = (characteristics.length == 0)
? Collectors.CH_ID
: Collections.unmodifiableSet(EnumSet.of(Collector.Characteristics.IDENTITY_FINISH,
characteristics));
return new Collectors.CollectorImpl<>(supplier, accumulator, combiner, cs);
}
//构造方法2, 需要传finisher();, 容器类型和结果类型不相同, 例如summingDouble()
public static<T, A, R> Collector<T, A, R> of(Supplier<A> supplier,
BiConsumer<A, T> accumulator,
BinaryOperator<A> combiner,
Function<A, R> finisher,
Characteristics... characteristics) {
Objects.requireNonNull(supplier);
Objects.requireNonNull(accumulator);
Objects.requireNonNull(combiner);
Objects.requireNonNull(finisher);
Objects.requireNonNull(characteristics);
Set<Characteristics> cs = Collectors.CH_NOID;
if (characteristics.length > 0) {
cs = EnumSet.noneOf(Characteristics.class);
Collections.addAll(cs, characteristics);
cs = Collections.unmodifiableSet(cs);
}
return new Collectors.CollectorImpl<>(supplier, accumulator, combiner, finisher, cs);
}
enum Characteristics {
CONCURRENT,
UNORDERED,
IDENTITY_FINISH
}
}
在自定义收集器的时候, 通常用Collector.of()来构建Collector对象
知道了自定义收集器需要的所有参数之后, 就可以尝试自己去写自己的收集器
需求 : collectors.summingDouble()方法在求和的时候会丢失经度(里面用的是double运行), 写一个不会丢失经度的自定义收集器
@AllArgsConstructor
@NoArgsConstructor
@Data
public class TestDO implements Serializable {
private String code;
private Double value;
}
//初始化集合
TestDO testDO = new TestDO("001", 75.9989);
TestDO testDO1 = new TestDO("001", 4.7852);
TestDO testDO3 = new TestDO("001", 19.3607);
TestDO testDO4 = new TestDO("001", 1.3191);
List<TestDO> list = Arrays.asList(testDO, testDO1, testDO3, testDO4);
//MyCollectors.summingBigDecimal(TestDO::getValue) 自定义收集器
Map<String, Double> map1 = list.stream().collect(Collectors.groupingBy(TestDO::getCode,MyCollectors.summingBigDecimal(TestDO::getValue)));
public class MyCollectors {
/**
* T : 参数
* ? : 过程中的值 就是new BigDecimal[1]
* Double : 结果
*
* @return java.util.stream.Collector
* @Description
* @Author hc
* @Since 2023/9/5 20:46
* @Param mapper
**/
public static <T> Collector<T, ?, Double> summingBigDecimal(ToDoubleFunction<? super T> mapper) {
return Collector.of(
//初始化容器supplier
() -> new BigDecimal[1],
//消费元素 accumulator
(result, item) -> {
if (result[0] == null) {
result[0] = BigDecimal.valueOf(mapper.applyAsDouble(item));
} else {
result[0] = result[0].add(BigDecimal.valueOf(mapper.applyAsDouble(item)));
}
},
//合并结果combiner(),parallelStream时候使用
(result1, result2) -> {
result1[0] = result1[0].add(result2[0]);
return result1;
},
//转换结果,finisher
total -> total[0].doubleValue()
);
}
}
写自定义收集器只需要理解collector里面那几个参数, 就可以了