【generator101】- generator是可以被外部控制执行进度的函数

很多码农终其一生可能在代码上干的事情无非就是追求两件事情:Logic Locality 和 Data Locality。前者决定了开发效率,后者决定了运行效率。协程是用来处理I/O阻塞和流程阻塞这两个普世问题的得力工具,可以达到比较好的Logic Locality。

def demo_before_and_after():
    print('before')
    yield
    print('after')

gen = demo_before_and_after()
gen.next()
gen.next()

这段代码执行的输出是

gen created
before
after
Traceback (most recent call last):
...
StopIteration

通过这个例子,可以了解到Python的generator的两个特性。
首先,demo_before_and_after()的时候,before并没有被打印出来。说明demo_before_and_after这个函数在这个时候还没有真正开始执行。gen.next()被调用一次,函数就往前执行一步。通过控制gen.next(),可以从外面控制函数内部的执行进度。
其次,Python的generator没有类似Java的Iterator.hasNext这样的设计,直接以抛出StopIteration来传达“到尾巴了”这个消息。

利用generator这个特性,Python的标准库提供了一个非常方便的contextmanager的设计

import contextlib

@contextlib.contextmanager
def demo_before_and_after():
    print('before')
    yield
    print('after')

with demo_before_and_after():
    print('inside')

输出结果是

before
inside
after

很多需要获得资源,使用资源,然后确保资源被正确释放的场合(类似C++的RAII),都可以用上面的写法来实现。

    原文作者:taowen
    原文地址: https://segmentfault.com/a/1190000000661293
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞