有没有办法拥有生产者成员功能？

Question

使用协程 1.3-RC2，我想做以下事情：

class Foo {
    fun getPrimes() = produce {
        var i = 0
        while (true) {
            if (i.isPrime()) {
                send(i)
            }

            i++
        }
    }
}

但它抱怨produce 由于接收器不匹配而无法使用。我可以将produce{} 包装在runBlocking 中，它可以编译，但它会阻塞。

那么，如何实现这种生产者模式，使得客户端代码可以运行myFoo.getPrimes().consumeEach(...)？

Answer 1

produce 需要一个协程作用域来运行。你可以传递一个作用域：

class Foo {
    fun getPrimes(scope: CoroutineScope) = scope.produce {
        var i = 0
        while (true) {
            if (i.isPrime()) {
                send(i)
            }

            i++
        }
    }
}

或者，例如，标记getPrimes 暂停并创建一个新范围：

class Foo {
    suspend fun getPrimes() = coroutineScope {
        produce {
            var i = 0
            while (true) {
                if (i.isPrime()) {
                    send(i)
                }

                i++
            }
        }
    }
}