Kotlin序列中如何实现首个匹配元素查找并转换的函数式写法？

我完全懂你的痛点——想要用Kotlin函数式风格处理序列，找到第一个能转换出非空结果的元素后立刻终止遍历，避免做无用功。你当前用fold实现的方案确实达不到这个效果，因为fold作为终端操作，必须遍历完整个序列才能完成折叠，哪怕你早就找到了目标结果。

问题根源 #

你的mapFirst函数用fold配合原子引用，看起来是在提前返回，但实际上fold会强制处理序列中的每一个元素。哪怕第一个元素就返回了非null值，后续的所有元素还是会被传入block执行，完全浪费了序列的惰性求值优势。

正确实现方式 #

方式一：直接用标准库函数（推荐）

从Kotlin 1.5版本开始，标准库已经提供了完美匹配你需求的函数：firstNotNullOfOrNull。它的作用就是：对序列中的每个元素应用转换函数，返回第一个非null的结果；如果所有元素都返回null，则返回null。最重要的是，一旦找到第一个非null结果，它会立刻停止遍历后续元素。

用它改写你的代码：

fun main() {
    Files.list(someDir).asSequence().map { it.toFile() }.firstNotNullOfOrNull { file ->
        file.useLines { lines ->
            lines.firstNotNullOfOrNull { line -> if (line == "123") line.toInt() else null }
        }
    }?.let { num ->
        println("num is $num") // 找到第一个包含"123"的行就会打印并终止
    } ?: println("not a single file had a line eq to '123'")
}

这里嵌套使用firstNotNullOfOrNull：

• 外层处理文件序列：找到第一个能返回非null值的文件（即包含"123"行的文件）后，立刻停止遍历后续文件
• 内层处理文件的行序列：找到第一个"123"行后，立刻停止读取该文件的后续行

完全符合你想要的尽早终止+函数式风格的需求。

方式二：自定义兼容低版本的实现

如果你需要兼容Kotlin 1.5之前的版本，或者想要自定义函数名，可以直接操作序列的迭代器来手动控制遍历流程：

fun <F, T> Sequence<F>.mapFirst(block: (F) -> T?): T? {
    val iterator = iterator()
    while (iterator.hasNext()) {
        val result = block(iterator.next())
        if (result != null) {
            return result // 找到结果立刻返回，终止遍历
        }
    }
    return null
}

这个实现直接利用了序列的惰性迭代器，每处理一个元素就检查结果，一旦得到非null值就立即终止，完全不会处理后续元素。

为什么原方案不行？ #

fold是一个贪婪的终端操作，它的设计目标是遍历整个序列来完成累加计算。哪怕你在lambda里提前返回当前的累加器，fold仍然会继续把后续所有元素传入lambda执行，这就导致了不必要的遍历和计算，完全没用到序列的惰性优势。

内容的提问来源于stack exchange，提问作者Yuri Geinish