如何在流上重用filter和map的应用？

Question

我有一组从共享类型继承的域对象（即grouprecord extends record、requestrecord extends record)。子类型具有特定的属性（即GroupRecord::GetCumulativeTime、RequestRecord::GetResponseTime)。

此外，作为解析日志文件的结果，我有一个包含混合子类型的记录列表。

List<Record> records = parseLog(...);

为了计算日志记录上的统计信息，我想只在匹配特定子类型的记录子集上应用数学函数，即只在grouprecords上应用数学函数。因此，我希望有一个特定子类型的过滤流。我知道可以使用以下方法将筛选和映射应用到子类型

records.stream()
       .filter(GroupRecord.class::isInstance)
       .map(GroupRecord.class::cast)
       .collect(...

在流上多次应用这个filter&cast（特别是在对同一子类型进行多次不同计算时）不仅很麻烦，而且会产生大量重复。

class TypeFilter<T>{

    private final Class<T> type;

    public TypeFilter(final Class<T> type) {
        this.type = type;
    }

    public Stream<T> filter(Stream<?> inStream) {
        return inStream.filter(type::isInstance).map(type::cast);
    }
}

TypeFilter<GroupRecord> groupFilter = new TypeFilter(GroupRecord.class); 

SomeStatsResult stats1 = groupFilter.filter(records.stream())
                                      .collect(...)
SomeStatsResult stats2 = groupFilter.filter(records.stream())
                                      .collect(...)

这是可行的，但我发现这种方法对于这样一个简单的任务来说有点太多了。因此，我想知道，有没有更好的方法，或者是什么是最好的方法，以简洁和可读的方式使用流和函数来使这种行为可重用？

Answer 1

这要看你发现什么“更简洁、更可读”。我自己认为，你已经实现的方式是好的，因为它是。

但是，确实有一种方法可以做到这一点，即使用stream.flatmap，从您使用它的地方来看，这种方法稍微短一点：

static <E, T> Function<E, Stream<T>> onlyTypes(Class<T> cls) {
  return el -> cls.isInstance(el) ? Stream.of((T) el) : Stream.empty();
}

它要做的是将每个原始流元素转换为一个元素的stream（如果元素具有预期的类型），或者转换为空的stream（如果元素没有预期的类型）。

其用途是：

records.stream()
  .flatMap(onlyTypes(GroupRecord.class))
  .forEach(...);

这种方法有明显的权衡：

• 您会丢失管道定义中的“filter”字。这可能比原来的更让人困惑，所以可能需要一个比onlytypes更好的名称。
• stream对象相对重量级，创建如此多的对象可能会导致性能下降。但您不应该相信我的话，在重载下对这两个变体进行配置。

static <E, R> Function<E, Stream<R>> filterAndMap(Predicate<? super E> filter, Function<? super E, R> mapper) {
   return e -> filter.test(e) ? Stream.of(mapper.apply(e)) : Stream.empty();
}

原来的onlytypes实现现在变为：

static <E, R> Function<E, Stream<R>> onlyTypes(Class<T> cls) {
  return filterAndMap(cls::isInstance, cls::cast);
}

但是，还有一个折衷：生成的平面映射器函数现在将保存捕获的两个对象（谓词和映射器），而不是上面实现中的单个class对象。这也可能是一种过度抽象的情况，但这取决于您需要代码的地方和原因。