这似乎是自定义收集器的理想用例.但在此之前,我认为您可以简化金额的总和,如下所示:
BigDecimal amount1 = dummy.stream()
.map(Dummy::getAmount1)
.filter(Objects::nonNull)
.reduce(BigDecimal::add).orElse(BigDecimal.ZERO);
现在,自定义收藏家.您可以在静态实用程序方法中将Dummy的实例累积到专用本地类的实例中:
static Collector toDummyObject(
Function mapper,
boolean someCondition) {
class Accumulator {
List newDummyList = new ArrayList<>();
BigDecimal amount1 = BigDecimal.ZERO;
BigDecimal amount2 = BigDecimal.ZERO;
public void add(Dummy dummy) {
newDummyList.add(mapper.apply(dummy));
}
public void addAndSum(Dummy dummy) {
if (dummy.getAmount1() != null) amount1 = amount1.add(dummy.getAmount1());
if (dummy.getAmount2() != null) amount2 = amount2.add(dummy.getAmount2());
add(dummy);
}
public Accumulator merge(Accumulator another) {
newDummyList.addAll(another.newDummyList);
return this;
}
public Accumulator mergeAndSum(Accumulator another) {
amount1 = amount1.add(another.amount1);
amount2 = amount2.add(another.amount2);
return merge(another);
}
public DummyObject finish() {
return someCondition ?
new DummyObject(newDummyList, amount1, amount2) :
new DummyObject(newDummyList);
}
}
return Collector.of(
Accumulator::new,
someCondition ? Accumulator::addAndSum : Accumulator::add,
someCondition ? Accumulator::mergeAndSum : Accumulator::merge,
Accumulator::finish);
}
现在我们准备好了:
dummy.stream().collect(toDummyObject(
eachDummy -> mapper.map(eachDummy, NewDummy.class),
someCondition));