我有以下课程:
public class Offer {
private final OfferType type;
private final BigDecimal price;
// constructor, getters and setters
}
和枚举类型:
public enum OfferType {
STANDARD, BONUS;
}
我的用例是将要约列表作为输入,我想过滤掉除最便宜之外的所有标准.所以对于以下输入数据
List offers = Arrays.asList(new Offer(OfferType.STANDARD, BigDecimal.valueOf(10.0)),
new Offer(OfferType.STANDARD, BigDecimal.valueOf(20.0)),
new Offer(OfferType.STANDARD, BigDecimal.valueOf(30.0)),
new Offer(OfferType.BONUS, BigDecimal.valueOf(5.0)),
new Offer(OfferType.BONUS, BigDecimal.valueOf(5.0)));
我期待以下结果
[Offer [type=STANDARD, price=10.0], Offer [type=BONUS, price=5.0], Offer [type=BONUS, price=5.0]]
是否有单行语句(使用流或任何第三方库)允许这样做?
解决方法:
虽然不是单流操作:
List some = offers.stream()
.filter(x -> x.getType() != OfferType.STANDARD)
.collect(Collectors.toCollection(ArrayList::new));
offers.stream()
.filter(x -> x.getType() == OfferType.STANDARD)
.min(Comparator.comparing(Offer::getPrice))
.ifPresent(some::add);
如果你发现自己做了很多,可能会旋转自定义收藏家:
public static Collector> minCollector() {
class Acc {
Offer min = null;
List result = new ArrayList<>();
void add(Offer offer) {
if (offer.getType() == OfferType.STANDARD) {
if (min == null) {
min = offer;
} else {
min = offer.getPrice()
.compareTo(min.getPrice()) > 0 ? min : offer;
}
} else {
result.add(offer);
}
}
Acc combine(Acc another) {
this.min = reduceMin(this.min, another.min);
result.addAll(another.result);
return this;
}
List finisher() {
result.add(min);
return result;
}
private Offer reduceMin(Offer left, Offer right) {
return Collections.min(Arrays.asList(left, right),
Comparator.nullsLast(Comparator.comparing(Offer::getPrice)));
}
}
return Collector.of(Acc::new, Acc::add, Acc::combine, Acc::finisher);
}
用法是:
List result = offers.stream()
.collect(minCollector());
标签:java,java-8,java-stream
来源: https://codeday.me/bug/20190724/1521095.html