基于谓词的Java 8 Stream indexOf方法

Question

这个问题在这里已经有了答案:

Stream Way to get index of first element matching boolean (6 个回答)

关闭 4年前。

我刚刚遇到这样的情况，我需要知道一个元素在列表中的索引(位置)，但只有一个谓词表达式来标识该元素。我找了一个类似的 Stream 函数

int index = list.stream().indexOf(e -> "TESTNAME".equals(e.getName()));

但无济于事。当然，我可以这样写:

int index = list.indexOf(list.stream().filter(e -> "TESTNAME".equals(e.getName()))
    .findFirst().get());

但这会 a) 对列表进行两次迭代(在最坏的情况下，该元素将是最后一个元素)并且 b) 如果没有元素与谓词匹配(我更喜欢 -1 索引)，则会失败。

我为此功能编写了一个实用方法:

public static <T> int indexOf(List<T> list, Predicate<? super T> predicate) {
    int idx = 0;
    for (Iterator<T> iter = list.iterator(); iter.hasNext(); idx++) {
        if (predicate.test(iter.next())) {
            return idx;
        }
    }

    return -1;
}

但由于这似乎是一个非常微不足道的算法，我本来希望它在 Java 8 Stream API 中的某个地方。是我错过了，还是真的没有这个功能？ (额外问题:如果没有这样的方法，是否有充分的理由？在函数式编程中使用索引可能是一种反模式吗？)

Answer 1

你的循环还不错，但你可以简化它:

public static <T> int indexOf(List<T> list, Predicate<? super T> predicate) {
    for(ListIterator<T> iter = list.listIterator(); iter.hasNext(); )
        if(predicate.test(iter.next())) return iter.previousIndex();
    return -1;
}

您可以使用类似的流

public static <T> int indexOf(List<T> list, Predicate<? super T> predicate) {
    return IntStream.range(0, list.size())
        .filter(ix -> predicate.test(list.get(ix)))
        .findFirst().orElse(-1);
}

但是如果列表很大并且不是随机访问，这将变得非常低效。我会留在循环中。

---

从 Java 9 开始，还有其他选择

public static <T> int indexOf(List<T> list, Predicate<? super T> predicate) {
    long noMatchPrefix = list.stream().takeWhile(predicate.negate()).count();
    return noMatchPrefix == list.size()? -1: (int) noMatchPrefix;
}

这对于“计算第一个匹配的元素”这个任务非常有表现力，但与“获取第一个匹配元素的索引”并不完全相同，它显示了没有匹配的情况，所以我们需要然后用 -1 替换结果。