我想在对象myClass
的集合上流式传输,以便使用Collectors.groupingBy()
对其进行分组。但是,我不想检索Map<String, List<myClass>>
,而是将其分组到对象myOutput
上并检索Map<String, myOutput>
。我试图创建一个自定义收集器:
List<myClass> myList = new ArrayList<myClass>();
myList.add(new myClass("a", 1));
myList.add(new myClass("a", 2));
myList.add(new myClass("b", 3));
myList.add(new myClass("b", 4));
Map<String,myOutput> myMap = myList.stream().collect(Collectors.groupingBy(myClass::getA, Collectors.of(myOutput::new, myOutput::accept, myOutput::combine)));
myClass:
protected String a;
protected int b;
public myClass(String aA, int aB)
{
a = aA;
b = aB;
}
public String getA()
{
return a;
}
public int getB()
{
return b;
}
myOutput:
protected int i;
public myOutput()
{
i = 0;
}
public void accept(myClass aMyClass)
{
i += aMyClass.getB();
}
public myOutput combine(myOutput aMyOutput)
{
i += aMyOutput.getI();
return this;
}
public int getI()
{
return i;
}
但是使用此代码时,收集器存在问题:
Collectors.of(myOutput::new, myOutput::accept, myOutput::combine)
我知道在这种情况下减少会容易得多,但我们假设在myOutput对象中有很多操作要做。
这个收藏家怎么了?
答案 0 :(得分:6)
你的收藏家很好。您只需拥有Collector.of
静态工厂(而不是Collectors.of
)。
这个编译很好并且有你想要的输出
Map<String,myOutput> myMap =
myList.stream()
.collect(Collectors.groupingBy(
myClass::getA,
Collector.of(myOutput::new, myOutput::accept, myOutput::combine)
));
但请注意,您不需要这样的收藏家。您可以重复使用现有的。在这种情况下,您希望按a
值进行分组,对于分组到同一a
的每个元素,您希望将其b
值相加。您可以使用映射器返回b
值的内置Collectors.summingInt(mapper)
:
Map<String,Integer> myMap =
myList.stream()
.collect(Collectors.groupingBy(
myClass::getA,
Collectors.summingInt(myClass::getB)
));