【Python协程的实现】

"

补充：数据安全问题

进程：





    多个进程操作同一个文件,会出现数据不安全





线程：





    多个线程操作同一个全局变量，会出现数据不安全





    对于共享的数据操作：





        如果是 += *= /= -= 操作，都存在数据不安全问题





        如果是append，extend，pop，remove操作，就不会出现数据不安全问题





协程：





    永远不会出现数据不安全问题





    因为协程是由程序员控制的，而程序员控制的只能是代码

协程示例代码：

# 最简单的协程





 





a = 0





 





def fn1():





    global a





    g = fn2()  # 拿到生成器





    next(g)  # 转向fn2函数执行





    a += 1





    next(g)  # 转向fn2函数执行





 





def fn2():





    global a





    yield





    a += 1





    yield





 





print(fn1())  # None





print(a)  # 2

1. 协程介绍

协程是单线程下的并发，又称微线程，纤程。英文名Coroutine。一句话说明什么是协程：协程是一种用户态的轻量级线程，即协程是由用户程序自己控制调度的.

1. Python的线程属于内核级别的，即由操作系统控制调度（如单线程遇到io或执行时间过长就会被迫交出cpu执行权限，切换其它线程运行）

2. 单线程内开启协程，一旦遇到io，就会从应用程序级别（而非操作系统）控制切换，以此来提升效率（非io操作的切换反而会降低效率！）

• 对比操作系统控制线程的切换，用户在单线程内控制协程的切换

优点如下：

1. 协程的切换开销更小，属于程序级别的切换，操作系统完全感知不到，因而更加轻量级

2. 单线程内就可以实现并发的效果，最大限度地利用cpu

缺点如下：

1. 协程的本质是单线程下实现并发，因而无法利用多核。（可以是一个程序开启多个进程，每个进程内开启多个线程，每个线程内开启协程）

2. 协程指的是单个线程，因而一旦协程出现阻塞，将会阻塞整个线程

• 总结协程特点

1. 必须在一个单线程里实现并发

2. 修改共享数据不需加锁

3. 用户程序里自己保存多个控制流的上下文栈

（附加：一个协程遇到io操作自动切换到其它协程（如何实现检测io？yield、greenlet都无法实现，需要用到gevent模块（select机制）））

---

二、greenlet模块

windows安装命令：pip3 install greenlet

• 单纯的切换（在没有io或没有重复开辟内存空间的操作下）反而会降低程序的执行速度

# 效率对比





 





from greenlet import greenlet





from time import time





 





def func1():





    res = 1





    for i in range(1000000):





        res +=i





 





def func2():





    res = 1





    for i in range(1000000):





        res *=i





 





 





# 顺序执行





start = time()





func1()





func2()





print('run time is', time() - start)





# run time is 0.19996070861816406





 





#  切换





start =time()





g1 = greenlet(func1)





g2 = greenlet(func2)





g1.switch()





print('run time is', time() - start)





# run time is 19.51878547668457

greenlet只是提供了一种比generator更加便捷的切换方式，当切到一个任务执行时，如果遇到io操作，便会原地阻塞，仍然没有解决遇到io自动切换来提升效率的问题.

单线程里的多个任务通常会既有计算操作又有阻塞操作，我们完全可以在执行任务1遇到阻塞时就切换到任务2继续执行。如此才能提高效率，这就需要用到Gevent模块.

---

三、gevent模块

windows安装命令：pip3 install gevent

gevent模块是一个第三方库，可以轻松通过gevent实现并发同步或异步编程，在gevent中用到的主要模式是greenlet，它是以C扩展模块形式接入Python的轻量级协程。greenlet全部运行在主程序操作系统进程的内部，但它们被协作式地调度。

• 基本用法

g = gevent.spawn(func, 1, 2, x=3, y=4)：创建一个协程对象g，spawn括号内的第一个参数是函数名，后面可以有多个参数，可以是位置实参或关键字实参，都是传给函数的func。

g.join()：等待g结束，等价于gevent.joinall([g1, g2])

g.value：拿到func的返回值

# 遇到io主动切换





 





import gevent





 





def eat(name):





    print('%s eat 1' % name)





    gevent.sleep(1)





    print('%s eat 2' % name)





 





def play(name):





    print('%s play 1' % name)





    gevent.sleep(1)





    print('%s play 2' % name)





 





g1 = gevent.spawn(eat, 'egon')





g2 = gevent.spawn(play, name='egon')





g1.join()





g2.join()





# 或者 gevent.joinall([g1, g2])

上面的gevent.sleep()模拟的是gevent可以识别的io阻塞，而time.sleep()或其它的阻塞，gevent是不能直接识别的，需要用到下面一行代码，打补丁，便可识别：from gevent import monkey; monkey.patch_all()  必须写在被打补丁者之前：

from gevent import spawn, monkey





import time





def eat(name):





    print('%s eat 1' % name)





    time.sleep(1)





    print('%s eat 2' % name)





 





def play(name):





    print('%s play 1' % name)





    time.sleep(1)





    print('%s play 2' % name)





 





monkey.patch_all()  # 打补丁





g1 = spawn(eat, 'egon')





g2 = spawn(play, name='egon')





g1.join()





g2.join()





# 或者 gevent.joinall([g1, g2])

我们可以使用threading.current_thread().getName()来查看每个协程的变量名都会为：DummyThread-n，既假线程。

from gevent import spawn, monkey





from threading import current_thread





 





func1 = lambda :print(current_thread().getName())   # DummyThread-1





 





monkey.patch_all()





spawn(func1).join()

• 同步与异步效率对比

# 同步与异步效率对比





 





from gevent import spawn, joinall, monkey;monkey.patch_all()





from time import sleep





 





def task(pid):





    """Some non-deterministic task"""





    sleep(0.5)





    print('Task %s done' % pid)





 





def synchronous():  # 同步





    [task(i) for i in range(10)]





 





def asynchronous(): # 异步





    gevent_lst = [spawn(task, i) for i in range(10)]





    joinall(gevent_lst)





    print('DONE')





 





if __name__ == '__main__':





    print('Syinchronous:')





    synchronous()





    print('Asynchronous:')





    asynchronous()

• 异步应用爬虫

from gevent import spawn, joinall, monkey; monkey.patch_all()





from requests import get





from time import time





 





def get_page(url):





    print('GET: %s' % url)





    response = get(url)





    if response.status_code == 200:





        print('%d bytes received from %s' %(len(response.text), url))





 





start_time = time()





joinall([





    spawn(get_page, 'https://www.python.org/'),





    spawn(get_page, 'https://www.yahoo.com/'),





    spawn(get_page, 'https://github.com/'),





])





 





print('run time is %s' %(time() - start_time))

• 实例：实现单线程下的socket并发

# Server





 





from gevent import spawn, monkey; monkey.patch_all()





from socket import socket, SOL_SOCKET, SO_REUSEADDR





 





def server(ip='127.0.0.1', port=8080):





    sk = socket()





    sk.setsockopt(SOL_SOCKET, SO_REUSEADDR,1)





    sk.bind((ip, port))





    sk.listen(10)





    while 1:





        conn, addr = sk.accept()





        spawn(task, conn)





        print('Client', addr)





 





def task(conn):





    try:





        while 1:





            res = conn.recv(1472)





            if not res:break





            print(res.decode('UTF-8'))





            conn.send(res.upper())





    except Exception as e:





        print(e)





    finally:





        conn.close()





 





if __name__ == '__main__':





    server()

# Clinet





 





from socket import socket





 





sk = socket()





sk.connect_ex(('127.0.0.1', 8080))





 





while 1:





    ret = input('>>>').strip()





    sk.send(ret.encode('UTF-8'))





    if not ret:break





    print(sk.recv(1472).decode('UTF-8'))

关于yield：

from time import time





 





# 在单线程中，如果存在多个函数，如果有某个函数发生IO操作，你想让程序马上切换到另一个函数去执行





#  以此来实现一个假的并发现象。





# 总结：





#    yield 只能实现单纯的切换函数和保存函数状态的功能





#    不能实现：当某一个函数遇到io阻塞时，自动的切换到另一个函数去执行





#    目标是：当某一个函数中遇到IO阻塞时，程序能自动的切换到另一个函数去执行





#            如果能实现这个功能，那么每个函数都是一个协程





#





#    但是  协程的本质还是主要依靠于yield去实现的。





#





#    如果只是拿yield去单纯的实现一个切换的现象，你会发现，跟本没有程序串行执行效率高





 





def consumer():





    while 1:





        x = yield





        print(x)





 





def producer():





    g = consumer()





    next(g)





    [g.send(i) for i in range(100000)]





 





start = time()





producer()





print('yield:', time() - start)

"