Java 8 Stream indexOf基于谓词的方法

时间:2018-04-09 14:22:21

标签: java java-8 java-stream

我刚遇到的情况是我需要知道列表中元素的索引(位置),但只有谓词表达式来标识元素。我找了一个像

这样的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索引)将会失败。 / p>

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

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中的某个地方。我只是想念它,还是真的没有这样的功能? (额外的问题:如果没有这样的方法,是否有充分的理由?在函数式编程中使用索引可能是反模式吗?)

4 个答案:

答案 0 :(得分:7)

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

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);
}

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

答案 1 :(得分:1)

我说一个简单的解决方案就是使用IntStream

IntStream.range(0, list.size())
         .filter(i -> Objects.nonNull(list.get(i)))
         .filter(i -> "TESTNAME".equals(list.get(i).getName()))
         .findFirst()
         .orElse(-1);

过滤掉null个元素会阻止它投掷NullPointerException

答案 2 :(得分:1)

    IntPredicate isNotNull = i -> Objects.nonNull(list.get(i));
    IntPredicate isEqualsTestName = i -> list.get(i).getName().equals("TESTNAME");

    int index = IntStream.range(0, list.size())
                .filter(isNotNull.and(isEqualsTestName))
                .findFirst()
                .orElse(-1);

答案 3 :(得分:1)

原因是“索引”的概念仅在“有序集合”中具有意义。 Stream可以来自任何Collection,例如Set;拥有“indexOf”方法本质上意味着给源Set一个get(int index)操作。更不用说Stream也可以来自任何其他来源,而不是集合。

另一个原因是indexOf操作的实现是它基本上需要Stream元素类型的有状态访问者;有状态的,因为它必须计算它已经访问过的成员。有状态操作和Stream不匹配。

至于你的效用函数,我不明白你为什么用你的方式写它而不是这样:

static <T> int indexOf(List<T> findIn, Predicate<? super T> pred) {
    for (int i = 0; i < findIn.size(); i++) {
        if (pred.test(findIn.get(i))) {
            return i;
        }
    }
    return -1;
}