问题描述
我有一个带有Seed
元素集合的类。方法的返回类型之一Seed
是Optional<Pair<Boolean,String>>
。
我正在尝试遍历所有seeds
,查找是否有任何boolean
值是true
,并同时创建一个包含所有String
值的集合。例如,我的输入形式为Optional<Pair<Boolean,String>>
,输出应为Optional<Signal>
,其中Signal
类似于:
class Signal {
public boolean exposure;
public Set<String> alarms;
// constructor and getters (can add anything to this class,it's just a bag)
}
这是我目前可以使用的功能:
// Seed::hadExposure yields Optional<Pair<Boolean,String>> where Pair have key/value or left/right
public Optional<Signal> withExposure() {
if (seeds.stream().map(Seed::hadExposure).flatMap(Optional::stream).findAny().isEmpty()) {
return Optional.empty();
}
final var exposure = seeds.stream()
.map(Seed::hadExposure)
.flatMap(Optional::stream)
.anyMatch(Pair::getLeft);
final var alarms = seeds.stream()
.map(Seed::hadExposure)
.flatMap(Optional::stream)
.map(Pair::getRight)
.filter(Objects::nonNull)
.collect(Collectors.toSet());
return Optional.of(new Signal(exposure,alarms));
}
现在我有时间做得更好,因为Seed::hadExposure
可能会成为昂贵的通话,所以我想看看是否可以只通过一次就可以完成所有这些操作。我已经尝试过使用收集器(reduce
,Collectors.collectingAndThen
等)来Collectors.partitioningBy
(使用先前的问题提出了一些建议),但到目前为止还没有。
解决方法
可以使用stream()
在单个map
表达式中执行此操作,以将非空曝光转换为Signal
,然后转换为reduce
以合并信号:
Signal signal = exposures.stream()
.map(exposure ->
new Signal(
exposure.getLeft(),exposure.getRight() == null
? Collections.emptySet()
: Collections.singleton(exposure.getRight())))
.reduce(
new Signal(false,new HashSet<>()),(leftSig,rightSig) -> {
HashSet<String> alarms = new HashSet<>();
alarms.addAll(leftSig.alarms);
alarms.addAll(rightSig.alarms);
return new Signal(
leftSig.exposure || rightSig.exposure,alarms);
});
但是,如果您有很多警报,那会很昂贵,因为它会创建一个新的Set
并将新警报添加到输入中每次曝光的累积警报中。
在从头开始设计以支持功能性编程的语言(例如Scala或Haskell)中,您将拥有Set
数据类型,该数据类型将允许您efficiently create a new set that's identical to an existing set,但要添加一个元素,因此无需担心效率:
filteredSeeds.foldLeft((false,Set[String]())) { (result,exposure) =>
(result._1 || exposure.getLeft,result._2 + exposure.getRight)
}
但是Java并没有开箱即用的东西。
您可以只为结果创建一个Set
并在流的reduce
表达式中对其进行变异,但是有些人会认为这是较差的样式,因为您会混合使用功能性范式(map /减少一个流)和一个过程(变异集合)。
就个人而言,在Java中,我只是放弃了功能性方法,在这种情况下使用了for
循环。这将减少代码,提高效率,并简化IMO。
如果您有足够的空间来存储中间结果,则可以执行以下操作:
List<Pair<Boolean,String>> exposures =
seeds.stream()
.map(Seed::hadExposure)
.flatMap(Optional::stream)
.collect(Collectors.toList());
然后,您只需在输入列表中为每个项目调用一次昂贵的Seed::hadExposure
方法。