使用Go语言简单模拟Python的生成器

更新时间：2015年08月12日 12:06:07 作者：taowen

这篇文章主要介绍了使用Go语言简单模拟Python的生成器,Python的generator是非常酷的功能,用Go实现的代码也较为简洁,需要的朋友可以参考下

def demo_input_and_output():
  input = yield 'what is the input?'
  yield 'input is: %s' % input

gen = demo_input_and_output()
print(gen.next())
print(gen.send(42))

这段代码演示了 python generator 的功能。可以看到 yield 同时做了两个操作，一个是往外发数据 "waht is the input"，同时做的操作是往里收数据 input。而且这个接收数据的操作是一个阻塞的操作，如果外部没有调用 next() （也就是往里传递None），或者调用send(42)（也就是往里传递42这个值），那么这个阻塞的操作就会一直等待下去。

也就是说 python 的 generator 自带了一个对外通信的 channel，用于收发消息。用 go 模拟 python 的 generator 的话写起来就是这样的

复制代码代码如下:

package main

import "fmt"

func demoInputAndOutput(channel chan string) {
    channel <- "what is my input?"
    input := <- channel
    channel <- fmt.Sprintf("input is: %s", input)
}

func main() {
    channel := make(chan string)
    go demoInputAndOutput(channel)
    fmt.Println(<- channel)
    channel <- "42"
    fmt.Println(<- channel)
}

这段代码和 python 版本基本上等价。隐含的 channel 在 go 版本里变成显式的了。yield 变成了 channel <- 操作，同时立马做了一个 <- channel 的阻塞读操作。这也就是 yield 的本质吧。

go 的 channel 也可以当成 iterator 被 for 循环使用：

复制代码代码如下:

package main

import "fmt"

func someGenerator() <-chan string {
    channel := make(chan string)
    go func() {
        channel <- "a"
        fmt.Println("after a")
        channel <- "c"
        fmt.Println("after c")
        channel <- "b"
        fmt.Println("after b")
        close(channel)
    }()
    return channel
}

func main() {
    channel := someGenerator()
    for val := range channel {
        fmt.Println(val)
    }
}

和 python 的 yield 不同，这里的 channel <- 不等价于 yield，它会往下执行直到阻塞。效果是

复制代码代码如下:

after a

a

c

after c

after b

b

这和预期的顺序不一样。这里没有把 after a after c after b 都打印出来是因为 channel 默认只有一个元素的buffer，所以写入了一个就阻塞了。如果增大 buffer，那么就有效果了

复制代码代码如下:

make(chan string, 10)

输出变成了：

after a
after c
after b
a
c
b

可见 goroutine 就好象一个独立的线程一样自己和自己玩去了，不用等待被执行。如果要模拟 yield 就要加上显示的同步操作（从 channel 里阻塞读取信号）：

复制代码代码如下:

package main

import "fmt"

func someGenerator() chan string {
    channel := make(chan string)
    go func() {
        channel <- "a"
        <- channel
        fmt.Println("after a")
        channel <- "c"
        <- channel
        fmt.Println("after c")
        channel <- "b"
        <- channel
        fmt.Println("after b")
        close(channel)
    }()
    return channel
}

func main() {
    channel := someGenerator()
    for val := range channel {
        fmt.Println(val)
        channel <- ""
    }
}

输出的结果就是

a
after a
c
after c
b
after b

到这里我们可以看到，python 的 generator 就好象是 golang 的 goroutine 带了一个无buffer的channel。这样导致每次yield一个值，都会产生一次协程上下文切换。虽然协程上下文切换很廉价，但是也不是没有成本。像 goroutine 的 buffered channel 这样的设计，可以让一个 goroutine 一次性多产生一些输出再阻塞等待，而不是产生一个输出就阻塞等待一下，再产生另外一个输出。golang rocks！

您可能感兴趣的文章:

Go常用标准库之fmt的简介与使用详解
fmt 是 Go 语言中的一个常用标准库,它用于格式化输入和输出数据,这篇文章主要为大家介绍了fmt的基本使用,感兴趣的小伙伴可以跟随小编一起学习一下
2023-10-10
GO语言文件的创建与打开实例分析
这篇文章主要介绍了GO语言文件的创建与打开的具体用法,实例分析了GO语言文件创建与打开操作中所涉及的函数具体用法,具有一定的参考借鉴价值,需要的朋友可以参考下
2014-12-12
Go语言指针访问结构体的方法
这篇文章主要介绍了Go语言指针访问结构体的方法,涉及Go语言指针及结构体的使用技巧,具有一定参考借鉴价值,需要的朋友可以参考下
2015-02-02
Go语言常用的打log方式详解
Golang的log包短小精悍,可以非常轻松的实现日志打印转存功能,下面这篇文章主要给大家介绍了关于Go语言常用的打log方式的相关资料,文中通过实例代码介绍的非常详细,需要的朋友可以参考下
2022-10-10
Golang Printf,Sprintf,Fprintf 格式化详解
这篇文章主要介绍了Golang Printf,Sprintf,Fprintf 格式化详解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2021-03-03
Golang中的四个括号示例详解
这篇文章主要介绍了Golang中的四个括号,本文通过实例代码给大家介绍的非常详细,通过实例代码补充介绍了有效的括号golang实现，需要的朋友可以参考下
2024-03-03
一文详解Go中方法接收器的选择
许多 Go 初学者在方法接收器的选择上可能会感到困惑，不知道该选择值接收器还是指针接收器。本文将会对方法接收器进行介绍，并给出如何选择正确方法接收器的指导建议,希望对大家有所帮助
2023-04-04
如何控制Go编码JSON数据时的行为(问题及解决方案)
今天来聊一下我在Go中对数据进行 JSON 编码时遇到次数最多的三个问题以及解决方法，本文通过实例代码给大家介绍的非常详细，具有一定的参考借鉴价值，需要的朋友参考下吧
2020-02-02
详解如何在Go项目中输出版本信息
这篇文章主要介绍了详解如何在Go项目中输出版本信息，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2020-01-01
Golang发送http GET请求的示例代码
这篇文章主要介绍了Golang发送http GET请求的示例代码，帮助大家更好的理解和使用golang，感兴趣的朋友可以了解下
2020-12-12

使用Go语言简单模拟Python的生成器

相关文章

最新评论

大家感兴趣的内容

最近更新的内容

常用在线小工具