如何使用分组通过减少而无需获得可选

时间:2019-03-08 19:55:06

标签: java java-8 java-stream grouping collectors

对于这个大大简化了我的问题的示例,我有一个Stat对象,其中有一个year字段和其他三个统计字段。想象一下,这些结果是兽医链中每种动物类型的患者数量的年度统计数据,而我想按年份获得所有分支的总数。

换句话说,从Stat对象的列表中,我想返回一个Map<Integer, Stat>,其中整数是年份,而Stat对象具有年份和每个对象的总和这四个字段。

public class Stat
{
    int year;
    public int getYear() { return year; }

    long cats;
    public long getCats() { return cats; }

    long dogs;
    public long getDogs() { return dogs; }

    long pigeons;
    public long getPigeons() { return pigeons; }

    public Stat(int year, long cats, long dogs, long pigeons)
    {
        this.year = year;
        this.cats = cats;
        this.dogs = dogs;
        this.pigeons = pigeons;
    }

    public Stat(Stat left, Stat right)
    {
        if (left.year != right.year)
            throw new IllegalArgumentException("Only allow combining for same year.");
        this.year = left.year;
        this.cats = left.cats + right.cats;
        this.dogs = left.dogs + right.dogs ;
        this.pigeons = left.pigeons + right.pigeons;
    }

    @Override
    public String toString()
    {
        return String.format("%d c=%d d=%d p=%d", year, cats, dogs, pigeons);
    }
}
@Test
public void testStat()
{
    List<Stat> items = Arrays.asList(
        new Stat(2017, 5, 8, 12),
        new Stat(2017, 123, 382, 15),
        new Stat(2018, 1, 2, 3)
        );
    Map<Integer, Optional<Stat>> result = items.stream()
        .collect(Collectors.groupingBy(Stat::getYear,
            Collectors.reducing(Stat::new)
        ));
    System.out.println(result);
}

Optional是不必要的,因为如果没有元素,groupingBy将永远不会创建需要List的{​​{1}}。

是否有一种获取reducing的方法,最好不必创建空白的“身份”对象?

如果我不得不向Map<Integer, Stat>创建一个身份创建函数,则Stat对象的合并构造函数必须有一个年份(请参阅构造函数),那么身份构造函数如何将年份传递给它?

1 个答案:

答案 0 :(得分:2)

您宁愿使用Collectors.toMap作为实现方式:

Map<Integer, Stat> result = items.stream()
        .collect(Collectors.toMap(Stat::getYear, 
                Function.identity(), (one, another) -> sumStatsOfSameYear(one, another)));

其中sumAttributes被定义为

// stat from the same year
private static Stat sumStatsOfSameYear(Stat one, Stat another) {
    new Stat(one.getYear(), one.getCats() + another.getCats(),
            one.getDogs() + another.getDogs(), one.getPigeons() + another.getPigeons()))
}