Python之并发编程(协程)

文章介绍了如何使用gevent模块、greenlet模块以及asyncio库实现协程,包括协程的概念、gevent的猴子补丁、greenlet的switch切换、asyncio的异步任务和TCP服务端并发处理。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

协程

我们都知道进程是资源单位
	线程是执行单位
	协程是单线程下实现并发(效率极高)
		在代码层面欺骗CPU 让CPU觉得我们的代码里面没有IO操作	
		实际上IO操作被我们自己写的代码检测 一旦有 立刻让代码执行别的
		(该技术完全是程序员自己弄出来的 名字也是程序员自己起的)
		核心:自己写代码完成切换+保存状态

使用gevent 模块实现协程

import time
from gevent import monkey;

monkey.patch_all()  # 固定编写 用于检测所有的IO操作(猴子补丁)
from gevent import spawn


def func1():
    print('func1 running')
    time.sleep(3)
    print('func1 over')


def func2():
    print('func2 running')
    time.sleep(5)
    print('func2 over')


if __name__ == '__main__':
    start_time = time.time()
    # func1()
    # func2()
    s1 = spawn(func1)  # 检测代码 一旦有IO自动切换(执行没有io的操作 变向的等待io结束)
    s2 = spawn(func2)
    s1.join()
    s2.join()
    print(time.time() - start_time)  # 8.01237154006958   协程 5.015487432479858

使用greenlet 模块,通过switch来切换不同的代码块协程运行

from greenlet import greenlet

def foo():
    print("foo")
    bar()

def bar():
    a = 3 + 1
    print(a)
    gr2.switch()
    print("end bar")


gr1 = greenlet(bar)
gr2 = greenlet(foo)
gr1.switch()

执行结果为:
>>>:4
>>>:foo
>>>:4
>>>:end bar

使用asyncio来实现协程(使用这个最好)

import asyncio
import time


async def job(t):             # async 形式的功能
    print('Start job ', t)
    # tieme.sleep() 		#  同步的io操作,即使加了await也不会切换
    await asyncio.sleep(t)          # 等待 "t" 秒, 期间切换其他任务
    print('Job ', t, ' takes ', t, ' s')
    

async def main(loop):          # async 形式的功能
    tasks = [
    loop.create_task(job(t)) for t in range(1, 3)
    ]                                       # 创建任务, 但是不执行
    await asyncio.wait(tasks)               # 执行并等待所有任务完成
    
t1 = time.time()
loop = asyncio.get_event_loop()             # 建立 loop
loop.run_until_complete(main(loop))         # 执行 loop,并且等待所有任务结束
loop.close()                                # 关闭 loop

print("Async total time : ", time.time() - t1)

执行结果:
Start job  1
Start job  2
Job  1  takes  1  s
Job  2  takes  2  s
Async total time :  2.01411509513855

协程实现TCP服务端并发

'服务端'
import socket
from gevent import monkey;

monkey.patch_all()  # 固定编写 用于检测所有的IO操作(猴子补丁)
from gevent import spawn


def communication(sock):
    while True:
        data = sock.recv(1024)  # IO操作接收数据
        print(data.decode('utf8'))
        sock.send(data.upper())


def get_server():
    server = socket.socket()  # 固定代码套路
    server.bind(('127.0.0.1', 8080))
    server.listen(5)
    while True:
        sock, addr = server.accept()  # IO操作
        spawn(communication, sock)  # 一旦有用户进来就启动 如果没有IO操作等着


s1 = spawn(get_server)  # 这样就好了 平凡的切换等着用户进来
s1.join()


'客户端'
import socket
from threading import Thread, current_thread


def get_client():
    client = socket.socket()
    client.connect(('127.0.0.1', 8080))
    while True:
        client.send(f'hello baby {current_thread().name}'.encode('utf8'))
        data = client.recv(1024)
        print(data.decode('utf8'))


for i in range(1000):  # 一人接待1000人!!!
    t = Thread(target=get_client)
    t.start()

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值