Kotlin的ForEach函数狂欢：带你玩转Continue和Break！

用Kotlin的ForEach函数尽情玩耍！

虽然Kotlin的forEach函数相当好用，但它缺少continue和break这两个小捣蛋，让我想尽情玩耍有点困难。但别担心，我有办法解决这个问题，接下来就让我带你见识一下如何整活forEach函数吧！

首先，我们得认识到forEach函数的本质：它是一个扩展函数，作用于集合类之上，用来遍历集合中的每个元素。因此，我们的目标是扩展这个扩展函数，赋予它continue和break的能力。

第一步，我们需要创建一个新的扩展函数。在Kotlin中，我们可以使用infix函数来扩展类，这正是我们需要的。因此，让我们创建一个名为forEachWithContinueAndBreak的infix函数，它的签名如下：

infix fun <T> Iterable<T>.forEachWithContinueAndBreak(action: (T) -> Unit)

在这个函数中，我们接收一个集合和一个需要在集合元素上执行的动作。我们使用infix函数，以便能够像这样调用它：

listOf(1, 2, 3, 4, 5).forEachWithContinueAndBreak {
    if (it % 2 == 0) continue
    println(it)
}

现在，让我们编写这个函数的实现。它将创建一个状态机来跟踪我们当前处于continue还是break状态。我们使用一个布尔变量来表示这一点，并初始化为false。然后，我们遍历集合中的每个元素，执行以下操作：

1. 如果状态机处于continue状态，则跳过当前元素并继续进行下一次迭代。
2. 如果状态机处于break状态，则停止遍历并返回。
3. 否则，执行给定的动作。

以下是代码实现：

infix fun <T> Iterable<T>.forEachWithContinueAndBreak(action: (T) -> Unit) {
    var continueState = false
    var breakState = false
    
    for (item in this) {
        if (continueState) {
            continueState = false
            continue
        }
        if (breakState) break
        
        action(item)
    }
}

现在，我们可以尽情使用continue和break来控制forEach函数的遍历行为。例如，以下代码将跳过所有偶数元素并仅打印奇数元素：

listOf(1, 2, 3, 4, 5).forEachWithContinueAndBreak {
    if (it % 2 == 0) continue
    println(it)
}

而以下代码将遍历集合中的所有元素，直到遇到大于5的元素时停止：

listOf(1, 2, 3, 4, 5, 6, 7).forEachWithContinueAndBreak {
    if (it > 5) break
    println(it)
}

通过扩展forEach函数，我们赋予了它我们一直渴望拥有的continue和break功能。现在，我们可以更灵活地遍历集合，编写更简洁、更具表现力的代码。尽情享受Kotlin的forEach函数带来的乐趣吧！