Java 8 GroupingBy与对象

时间:2016-03-22 15:14:50

标签: java java-8 java-stream collectors

我想在对象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对象中有很多操作要做。

这个收藏家怎么了?

1 个答案:

答案 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)
          ));