python中生成器的应用

生成器的应用:实现多任务

import time

def task_1():
    while True:
        print('-----1-------')
        time.sleep(0.1)
        yield


def task_2():
    while True:
        print('-------2---------')
        time.sleep(0.1)
        yield

# task_1()
# task_2()

def main():
    t1 = task_1()
    t2 = task_2()
    """
    类似于两个while True一起执行
    先让t1运行一会,当t1遇到yield的时候,再返回到18行
    然后执行t2,当它遇到yield的时候,再次切换到t1中
    这样t1/t2/t1/t2的交替运行,最终实现了多任务---->协程
    """
    while True:
        next(t1)
        next(t2)
main()

并行(真的):有两个任务,两个cpu,一个任务占一个cpu
并发(假的):有四个任务,两个cpu,四个任务交替占有cpu执行

yield实现单线程并发

import time

def consumer(name):
    print('%s 准备学习了~' %(name))
    while True:
        lesson = yield
        print('开始[%s]了,[%s]老师来讲课了~' %(lesson,name))

def producer(name):
    c1 = consumer('A')
    c2 = consumer('B')
    c1.__next__()
    c2.__next__()
    print('同学们开始上课了~')
    for i in range(10):
        time.sleep(1)
        print('到了两个同学')
        c1.send(i)
        c2.send(i)

producer('westos')

利用了关键字yield一次性返回一个结果,阻塞,重新开始
send 唤醒

使用greenlet完成多任务
为了更好的使用协程来完成多任务,python中的greeblet模块
对其进行的封装
pip3 install greenlet
“”"

from greenlet import greenlet
import time

def test1():
    while True:
        print('---A----')
        gr2.switch()
        time.sleep(0.5)

def test2():
    while True:
        print('----B----')
        gr1.switch()
        time.sleep(0.5)

"""
greenlet这个类对yield进行的封装
"""
gr1= greenlet(test1)
gr2 = greenlet(test2)

gr1.switch()

同样的还有gevent模块

import gevent

def f1(n):
    for i in range(n):
        print(gevent.getcurrent(),i)
        gevent.sleep(0.5)

def f2(n):
    for i in range(n):
        print(gevent.getcurrent(),i)
        gevent.sleep(0.5)
def f3(n):
    for i in range(n):
        print(gevent.getcurrent(),i)
        #gevent.sleep(0.5)

g1 = gevent.spawn(f1,5)
g2 = gevent.spawn(f2,5)
g3 = gevent.spawn(f3,5)

g1.join()
g2.join()
g3.join()

你可能感兴趣的:(python中生成器的应用)