浅谈Ruby里的生成器特性

在老东家写了一段时间Python后,我被告知Python里面有个很经典的被称之为生成器的特性,而这个特性至今让我难以忘怀。后来我接触了Ruby,我就纳闷了Ruby怎么就没看到这个特性?

直到最近在读《松本行弘的程序员世界》的时候我才意识到我错了,这个特性早在Ruby1.9就已经具备,只不过是我资历尚浅,察觉不到它的存在罢了。

松本行弘

1. Python的yield语句可用于构造生成器

下面是一个简单的Python生成器例子,它的功能是遍历0-max范围内的所有数字

def range(max):
    n = 0
    while n < max:
        yield n
        n += 1

if __name__ == '__main__':
    for i in range(100):
        print(i)

PS: 结果就是打印0~99这里就不贴出来了。

有没有感觉这个函数很有梯度感?(这是我放弃Python的理由之一 )T_T。如果我们直接在REPL环境里面运行range函数,会得到下面的结果

In [8]: range(100)
Out[8]: <generator object range at 0x102f41460>

从字面上可以看出它是一个generator,也就是我们所说的生成器。

当然,我们也可以直接构造一个数组,通过遍历数组来完成以上的打印过程。但生成器的好处就是它是惰性的,也就是说当你访问某个元素的时候对应元素才会被生成,而不是先生成一个完整的数组,然后我们再去遍历它。

2. Ruby中的yield

现在我们看看Ruby的yield语句是用来做什么的,举个简单的例子就能够看出来

def handler(a)
  yield(a)
end

result = handler(100) { |i| i * 20 }

puts result

程序的运行结果是

> 2000

简单来说,Ruby里面的yield可以用来接收方法块,调用它就相当于调用方法块。上面的例子里我们的方法块是{ |i| i * 20 },它接收一个参数i并返回i * 20的值,结果就是 100 * 20 = 2000

3. Ruby中的生成器

那问题来了,yield这个语句已经被用作方法块的代理了,那我们在Ruby里面要怎么实现如Python般可以返回生成器的函数?还是说Ruby中压根没有生成器这种机制?

怎么可能呢?其实生成器就在我们身边,只不过我们很多时候都察觉不到它的存在,具体我们可以参考一下这篇文章。在Ruby1.9之后就实现了Enumerator这个类,它的机制跟Python里面的生成器类似,可以用作惰性求值。使用方式如下

> a = (1..100000000000000000).each
=> #<Enumerator: 1..100000000000000000:each>
> a.next
=> 1
> a.next
=> 2
> a.next
=> 3
> a.next
=> 4
> a.next
=> 5

有点编程经验的朋友都知道,我们不怎么可能活着看到我们的计算机生成长度为100000000000000000的数组,所以这里面一定有什么黑魔法T_T。

本质上它是通过Range#each方法返回了Enumerator的对象,我们可以通过依次调用Enumerator#next方法来获取下一个元素,而这个Enumerator的对象就相当于一个生成器。

4. 用Ruby实现可以返回生成器的函数


Ruby1.9之后Ruby提供了名为Fiber的类,可以用于构建轻量级的协程,它让我们可以手动地去控制我们的线程是否被唤醒。那么这个玩意到底有什么用呢?毕竟,我们现在来谈协程这些多线程的东西似乎有点早。

下面是一个简单的例子

fiber = Fiber.new do
  Fiber.yield 1
  2
end

puts fiber.resume
puts fiber.resume
puts fiber.resume

可见Fiber::yield的作用跟Python中的yield语法有点类似。这里我们通过Fiber#new语句来创建Fiber实例,并传递一个代码块。我们可以把Fiber::yield想象成类似于byebug或者binding.pry等调试语句,程序运行到这里就会暂停,然后把控制权让出去。

当我们第一次运行Fiber#resume方法的时候就会返回Fiber::yield语句后面的数值,而第二次执行Fiber#resume就会继续运行代码块余下的代码并返回2,第三次运行Fiber#resume的时候由于我们块已经执行完毕所以会报错

> fiber.resume
=> 1
> fiber.resume
=> 2
> fiber.resume
FiberError: dead fiber called
    from (irb):7:in `resume'
    from (irb):7

感觉这个Fiber类就是我们要找的生成器的类。下面是我YY出来的用Ruby的生成器特性来实现的函数,它的功能类似文章开头用Python实现的函数。

def range(n)
  Fiber.new do
    i = 0
    while i < n
      Fiber.yield i
      i += 1
    end
  end
end

fiber = range(10)

p fiber.resume
p fiber.resume
p fiber.resume
p fiber.resume

上面脚本的输出是

0
1
2
3

在Python里面含有yield语句的方法被调用之后会返回一个生成器对象,然后我们可以利用这个生成器来进行惰性求值(个人感觉这种语法不够清爽,没有一个显式的返回语句,把生成器对象返回)。另外,Python的版本里使用了for语句来迭代生成器,我这里只是简单地手动去迭代。

Ruby的range方法返回了一个Fiber的对象,它就相当于一个生成器(或者也可以说是一个外部的迭代器),通过反复调用Fiber#resume方法就可以获取下一个值,如果反复调用超出限制就会报错。我Fiber#resume10次之后就会得到下面的结果。

dead fiber called (FiberError)

5. 总结

瞎扯了一下Ruby的生成器特性。作为一个特性,它并不像Python那么明显,但是它确实存在着。得益于Fiber这种类库的出现,已经更多Ruby多线程相关的库被开发出来了。据说Fiber性能方面的表现比Ruby原来内置的Thread要好一些,具体的比较我想放在后面的文章再详细讨论,很感谢你能够看到这里。

Happy Coding and Writing !!

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 158,425评论 4 361
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 67,058评论 1 291
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 108,186评论 0 243
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 43,848评论 0 204
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,249评论 3 286
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,554评论 1 216
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,830评论 2 312
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,536评论 0 197
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,239评论 1 241
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,505评论 2 244
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 32,004评论 1 258
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,346评论 2 253
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 32,999评论 3 235
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,060评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,821评论 0 194
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,574评论 2 271
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,480评论 2 267

推荐阅读更多精彩内容