Pix*_*tix 5 java grouping java-8 java-stream collectors
对于我的问题的这个大大简化的示例,我有一个Stat包含一个year字段和三个其他统计字段的对象。想象一下,结果是来自兽医连锁店分支机构的每种动物类型的患者数量的年度统计数据,我想按年获得所有分支机构的总和。
换句话说,从Stat对象列表中,我想返回 a 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);
}
Run Code Online (Sandbox Code Playgroud)
该Optional是不必要的,因为groupingBy永远不会创建一个List需要reducing,如果没有的元素。
有没有办法得到Map<Integer, Stat>,最好不必创建一个空白的“身份”对象?
如果我不得不求助于创建一个身份创建函数 to reducing,那么 Stat 对象的组合构造函数必须有一年(请参阅构造函数),那么身份构造函数如何将年份传递给它?
您可以使用Collectors.toMapas实现这一点:
Map<Integer, Stat> result = items.stream()
.collect(Collectors.toMap(Stat::getYear,
Function.identity(), (one, another) -> sumStatsOfSameYear(one, another)));
Run Code Online (Sandbox Code Playgroud)
其中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()))
}
Run Code Online (Sandbox Code Playgroud)
| 归档时间: |
|
| 查看次数: |
718 次 |
| 最近记录: |