A menudo me tropiezo con este problema, pero no veo una implementación común: ¿cómo encuentro idiomáticamente (funcionalmente) un elemento, detengo la búsqueda después de la coincidencia y también devuelvo un tipo diferente (es decir, mapeo lo que coincida con otro tipo)?
He sido capaz de hacer una solución con
fun <F,T> Sequence<F>.mapFirst(block: (F) -> T?): T? = fold(AtomicReference<T>()) { ref, from -> if (ref.get() != null) return@fold ref ref.set(block(from)) ref }.get() fun main() { Files.list(someDir).asSequence().map { it.toFile() }.mapFirst { file -> file.useLines { lines -> lines.mapFirst { line -> if (line == "123") line.toInt() else null } } }?.let { num -> println("num is $num") // will print 123 as an Int } ?: println("not a single file had a line eq to '123'") } Pero eso no se detiene en la coincidencia (cuando block() devuelve un valor no nulo) y consume todos los archivos y todas sus líneas.
Un bucle for simple es suficiente para implementar mapFirst :
fun <F,T> Sequence<F>.mapFirst(block: (F) -> T?): T? { for (e in this) { block(e)?.let { return it } } return null } Si necesita una solución sin introducir sus propias extensiones (aunque no tiene nada de malo), puede usar la combinación mapNotNull + firstOrNull :
files.asSequence() .mapNotNull { /* read the first line and return not null if it's ok */ } .firstOrNull()No mapearía los valores que descarta entonces, sino que lo haría así:
sequenceOf(1, 2, 3) .firstOrNull() { it == 2 } ?.let { it * 2 } ?: 6Primero encuentras el valor que coincide con tu condición, luego lo transformas también como quieras. En caso de que no encuentre un elemento coincidente, asigne un valor predeterminado (en este caso, 6).