检查流中的 instanceof

Question

我有以下表达式：

scheduleIntervalContainers.stream()
        .filter(sic -> ((ScheduleIntervalContainer) sic).getStartTime() != ((ScheduleIntervalContainer)sic).getEndTime())
        .collect(Collectors.toList());

...其中 scheduleIntervalContainers 具有元素类型 ScheduleContainer:

final List<ScheduleContainer> scheduleIntervalContainers

是否可以在过滤器之前检查类型？

Answer 1

您可以应用另一个 filter 以仅保留 ScheduleIntervalContainer 个实例，添加 map 将为您节省以后的转换：

scheduleIntervalContainers.stream()
    .filter(sc -> sc instanceof ScheduleIntervalContainer)
    .map (sc -> (ScheduleIntervalContainer) sc)
    .filter(sic -> sic.getStartTime() != sic.getEndTime())
    .collect(Collectors.toList());

或者，正如 Holger 评论的那样，如果您喜欢这种风格，可以将 lambda 表达式替换为方法引用：

scheduleIntervalContainers.stream()
    .filter(ScheduleIntervalContainer.class::isInstance)
    .map (ScheduleIntervalContainer.class::cast)
    .filter(sic -> sic.getStartTime() != sic.getEndTime())
    .collect(Collectors.toList());

Answer 2

一个非常优雅的选择是使用 class:

的方法参考

scheduleIntervalContainers
  .stream()
  .filter( ScheduleIntervalContainer.class::isInstance )
  .map( ScheduleIntervalContainer.class::cast )
  .filter( sic -> sic.getStartTime() != sic.getEndTime())
  .collect(Collectors.toList() );

Answer 3

@Eran 解决方案有一个小问题 - 在 filter 和 map 中输入 class 名称容易出错 - 很容易忘记在这两个地方更改 class 的名称。一个改进的解决方案是这样的：

private static <T, R> Function<T, Stream<R>> select(Class<R> clazz) {
    return e -> clazz.isInstance(e) ? Stream.of(clazz.cast(e)) : null;
}

scheduleIntervalContainers
  .stream()
  .flatMap(select(ScheduleIntervalContainer.class))
  .filter( sic -> sic.getStartTime() != sic.getEndTime())
  .collect(Collectors.toList());

但是，为每个匹配元素创建 Stream 可能会降低性能。在庞大的数据集上使用它时要小心。我从@Tagir Vailev

那里学到了这个解决方案

Answer 4

我会推荐这种实用方法，而不是像其他答案建议的那样使用过滤器 + 地图：

public static <Super, Sub extends Super> Function<Super, Stream<Sub>> filterType(Class<Sub> clz) {
  return obj -> clz.isInstance(obj) ? Stream.of(clz.cast(obj)) : Stream.empty();
}

用作：

Stream.of(dog, cat fish)
  .flatMap(filterType(Dog.class));

与filter+map相比有以下优点：

如果 class 没有扩展你的 class 你会得到一个编译错误
一个地方，你永远不会忘记在过滤器或地图

检查流中的 instanceof

Check instanceof in stream

java

instanceof

java-8

java-stream