首页 > 解决方案 > 仅当流不是空后缀时,自定义收集器才在分隔符、后缀和前缀上加入流

问题描述

我有一个字符串流:

Stream<String> stream = ...;

我想使用创建一个字符串

stream.collect(Collectors.joining(',', '[', ']'))

如果流不包含任何元素,我只想返回“No Strings”。

我注意到该String java.util.stream.Stream.collect(Collector<? super String, ?, String> collector) 方法采用 java.util.stream.Collector<T, A, R> 类型的参数

对于我的项目,我在很多地方都需要这个功能,所以我需要一个实现Collector接口的类。

我知道这可以通过 Stream to a List 来完成,然后检查 List.size() == 0?然后在需要时再次将列表转换为流。

List<String> list = stream.collect(Collectors.toList());

if (list.size() == 0) {
    return "No Strings";
}
return list.stream().collect(Collectors.joining(",", "[", "]"));`

这就是我想要发生的事情

List emptyList<String> = new ArrayList<>; System.out.println(emptyList.stream().collect(Collectors.joining(",", "[", "]")));

输出

[]

我想要的输出

No Strings

标签: javajava-streamcollectors

解决方案


老实说,我会采用您目前的方法(测试空虚)。

但是如果你真的想使用直接的收集器,你可以使用的源代码Collections.joiningJavadocStringJoiner作为指南来制作自定义收集器:

Collector.of(
    () -> new StringJoiner(",", "[", "]").setEmptyValue("No strings"),
    StringJoiner::add,
    StringJoiner::merge,
    StringJoiner::toString)

推荐阅读