揭秘 Kotlin Flow 操作符源码，提升 Kotlin 协程编程技巧

2024-02-14 05:41:11







## 前言

Kotlin Flow 是一个强大的库，用于在 Kotlin 协程中处理数据流。它提供了一系列操作符，可以轻松地转换、过滤和组合数据流。在本文中，我们将深入探讨 Flow 操作符的源码，帮助您理解 Flow 的工作原理和使用技巧，从而提升 Kotlin 协程编程能力。

## drop 操作符

drop 操作符用于忽略数据流的前 [count] 个元素。其源码如下：

```kotlin
fun <T> Flow<T>.drop(count: Long): Flow<T> = flow {
    var skipped = 0L
    collect { value ->
        if (skipped < count) {
            skipped++
        } else {
            emit(value)
        }
    }
}

从源码中可以看出，drop 操作符通过一个名为 skipped 的变量来记录已忽略的元素数量。在 collect 函数中，它首先检查 skipped 是否小于 count。如果小于，则将 skipped 加 1 并继续忽略下一个元素。否则，它将通过 emit 函数发出该元素。

dropWhile 操作符

dropWhile 操作符用于忽略满足条件的元素，并返回一个包含之后所有元素的流。其源码如下：

fun <T> Flow<T>.dropWhile(predicate: suspend (T) -> Boolean): Flow<T> = flow {
    var dropping = true
    collect { value ->
        if (dropping && predicate(value)) {
            // Continue dropping
        } else {
            dropping = false
            emit(value)
        }
    }
}

从源码中可以看出，dropWhile 操作符通过一个名为 dropping 的布尔变量来记录当前是否正在忽略元素。在 collect 函数中，它首先检查 dropping 是否为 true。如果为 true，则检查当前元素是否满足谓词 predicate。如果满足，则继续忽略该元素。否则，将 dropping 设置为 false 并通过 emit 函数发出该元素。

take 操作符

take 操作符用于获取数据流的前 [count] 个元素。其源码如下：

fun <T> Flow<T>.take(count: Long): Flow<T> = flow {
    var taken = 0L
    collect { value ->
        if (taken < count) {
            taken++
            emit(value)
        } else {
            complete()
        }
    }
}

从源码中可以看出，take 操作符通过一个名为 taken 的变量来记录已获取的元素数量。在 collect 函数中，它首先检查 taken 是否小于 count。如果小于，则将 taken 加 1 并通过 emit 函数发出该元素。否则，它将通过 complete 函数完成流。