我正在尝试将他的工作代码转换为流:
Map<Month,Integer> data = new HashMap<>();
for (Scooter scooter : scooters) {
scooter.getRentalDays().iterator().forEachRemaining(e -> {
int newVal = scooter.getMileage() * scooter.getMileageRate();
int curVal = data.getOrDefault(e.getMonth(), 0);
data.put(e.getMonth(), curVal + newVal);
});
}
迭代器增加的复杂性给我带来了一个问题,我不知道如何对哈希图内的值求和。下面的示例有点类似于上面的代码,但是不支持该值的总和。
Map<Month,Integer> data = projects.stream()
.flatMap(scooter -> scooter.getRentalDays().stream()
.map(date -> Map.entry(date.getMonth(), scooter.getMileage() * scooter.getMileageRate())))
.collect(Collectors.toMap(
Map.Entry::getKey,
Map.Entry::getValue, // this value needs to be a sum of the previous one.
(k1, k2) -> k1));
关于如何解决此问题的任何提示?这个想法是要有一个完整的流解决方案。
一种可能的解决方案是引入一个中间类来存储中间值(假设Month
是enum
):
private static Map<Month, Integer> sumGroupingByMonth(final Collection<Scooter> scooters) {
@AllArgsConstructor
final class Intermediate {
final int newVal;
final Day rentalDay;
}
return scooters.stream()
.flatMap(scooter -> {
final int newVal = scooter.getMileage() * scooter.getMileageRate();
return scooter.getRentalDays()
.stream()
.map(day -> new Intermediate(newVal, day));
})
.collect(Collectors.groupingBy(intermediate -> intermediate.rentalDay.getMonth(), Collectors.summingInt(intermediate -> intermediate.newVal)));
}
我发现这段代码非常复杂:闭包,显式中间状态,函数作用域,中间对象,以及以下内容可能会更简单,更快?
private static int[] sumGroupingByMonth(final Iterable<Scooter> scooters) {
final int[] data = new int[12];
for ( final Scooter scooter : scooters ) {
final int newVal = scooter.getMileage() * scooter.getMileageRate();
for ( final Day rentalDay : scooter.getRentalDays() ) {
data[rentalDay.getMonth().ordinal()] += newVal;
}
}
return data;
}
谢谢您的回答,我担心使用流将使代码变得太复杂(不易阅读),您的示例证明了这一点。