asyncio 将循环附加在进程上

一只仟纸鹤

于 2022-05-08 22:51:43 发布

阅读量371

点赞数

分类专栏： asyncio 文章标签： python 开发语言

本文链接：https://blog.youkuaiyun.com/qq_36057956/article/details/124656546

版权

asyncio 专栏收录该内容

3 篇文章

订阅专栏

此博客展示了如何在Python中使用asyncio库和multiprocessing模块进行异步任务处理。通过定义async worker函数模拟业务，并在process_main函数中启动多个进程执行这些任务。在主程序中，使用KeyboardInterrupt信号来优雅地停止进程，并通过cleanup函数清理已结束的进程。博客主要探讨了Python中并发执行和进程管理的实现。

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

import asyncio
import os
import random
from multiprocessing import Process

process=[]

def cleanup():
    global process
    while process:
        proc=process.pop()
        try:
            proc.join()
        except KeyboardInterrupt:
            proc.terminate()

async def worker():
    rand=random.randint(0,3)
    result=await asyncio.sleep(rand,result="模拟业务时间为{}s".format(rand))
    print(result)

def process_main(worker,num):
    loop=asyncio.new_event_loop()
    try:
        workers=[worker() for _ in range(num)]
        loop.run_until_complete(asyncio.gather(*workers,loop=loop))
    except KeyboardInterrupt:
        print(f"stopping {os.getpid()}")
        loop.stop()
    finally:
        loop.close()

def main(process,nums_prop,w_num, process_main):
    for _ in range(nums_prop):
        proc=Process(target=process_main,args=(worker,w_num))
        process.append(proc)
        proc.start()

if __name__ == '__main__':
    try:
        main(process,10,2,process_main)
    except KeyboardInterrupt:
        print("crtl+c 执行成功，停止进程...")
    finally:
        cleanup()