Python并发编程——多线程与协程

Posted donghe123

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Python并发编程——多线程与协程相关的知识,希望对你有一定的参考价值。

Pythpn并发编程——多线程与协程

1. 进程与线程

1.1 概念上
  • 对于操作系统来说,一个任务就是一个进程Process,在一个进程内部,要同时干很多事情,就需要同时运行多个子任务,进程内的这些子任务就称为线程Thread
  • 操作系统是让各个任务交替执行实现支持多任务的,真正地同时执行多任务需要多核CPU才能实现
  • 线程是最小的执行单元,一个进程至少有一个线程,如何调读进程和线程,完全由操作系统决定,程序自己不能决定什么时候执行,执行多长时间
1.2 多进程与多线程——同时执行多个任务

要实现多任务,设计Master-Worker模式,Master负责分配任务,Worker负责执行任务

  • 多进程模式
    • 启动多个进程,每个进程只有一个线程,多个进程可以一块执行多个任务
    • 最大的优点:稳定性高,一个子进程崩溃了,不会影响主进程和其他子进程
    • 缺点:创建进程的开销大
  • 多线程模式
    • 启动一个进程,在一个进程内启动多个线程,多个线程也可以一块执行多个任务
    • 致命缺点:任何一个线程挂掉都可能造成整个进程崩溃,因为所有线程共享进程的内存
  • 多进程+多线程模式
    • 实际很少采用

2. 并发和并行

并发:不是指同一时刻有多个操作同时进行,实际上,在某个特定时刻,只允许有一个操作发生,线程/任务之间互相切换,直到完成,threadingasyncio

  • 通常应用于I/O操作频繁的场景,例如从网站上下载多个文件

并行:同一时刻,同时发生,multi-processing,m个处理器,开m个进程

3. Python多线程——futures

3.1 多线程用法
  • 1.导入future模块,Python中的future模块,位于concurrent.futures 和 asyncio 中
  • 2.创建线程池,函数ThreadPoolExecutor(max_workers=5),max_workers设置线程个数
  • 3.调用,map()函数
import concurrent.futures
import requests
import threading
import time

def download_one(url):
    resp = requests.get(url)
    print(‘Read {} from {}‘.format(len(resp.content), url))

def download_all(sites):
    # 并发模式,创建了一个线性池,总共有5个线性可以分配使用
    with concurrent.futures.ThreadPoolExecutor(max_workers=5) as executor: 
        # 对sites中的每个元素,并发地调用函数download_one
        executor.map(download_one, sites) 
    # 并行模式,创建进程池,系统自动返回CPU的数量作为可以调用的进程数
    # with concurrent.futures.ProcessPoolExecutor() as executor: # 

# 另一种写法
# def download_all(sites):
#     with concurrent.futures.ThreadPoolExecutor(max_workers=5) as executor: 
#         to_do =[] #
#         for site in sites:
#             future = executor.submit(download_one, site) 
#             to_do.append(future)
#         for future in concurrent.futures.as_completed(to_do):
#             future.result()

def main():
    sites = [
        ‘https://en.wikipedia.org/wiki/Portal:Arts‘,
        ‘https://en.wikipedia.org/wiki/Portal:History‘,
        ‘https://en.wikipedia.org/wiki/Portal:Society‘,
        ‘https://en.wikipedia.org/wiki/Portal:Biography‘
    ]
    start_time = time.perf_counter()
    download_all(sites)
    end_time = time.perf_counter()
    print(f‘Download {len(sites)} sites in {end_time - start_time} seconds‘)

if __name__ == ‘__main__‘:
    main()
Read 182102 from https://en.wikipedia.org/wiki/Portal:Arts
Read 245181 from https://en.wikipedia.org/wiki/Portal:Society
Read 206928 from https://en.wikipedia.org/wiki/Portal:History
Read 336222 from https://en.wikipedia.org/wiki/Portal:Biography
Download 4 sites in 0.22546189799322747 seconds
3.2. 为什么多线程每次只允许只能有一个线程执行?

? 全局解释器锁的存在,GIL(Global Interpreter Lock)

3.3 多线程的缺点
  • 多线程运行过程容易被打断,因此有可能出现 race condition 的情况
  • 线程切换本身存在一定的损耗,线程数不能无限增加

4. python协程——asyncio

4.1 概念

单线程的异步编程模型称为协程。在执行过程中,在子程序内部可中断,然后转而执行别的子程序,在适当的时候再返回来接着执行,注意,这不是函数调用。

Async异步:不同操作间可以相互交替执行,如果其中的某个操作被block了,程序并不会等待,而是找出可执行的操作继续执行
sync同步:指操作一个接一个地执行,下一个操作必须等上一个操作完成后才能执行

4.2 Asyncio原理

event loop对象维护两个任务列表,预备状态和等待状态,选取预备状态的一个任务,使其运行,一直到这个任务把控制权交还给event loop为止,当任务把控制权交还给event loop时,如果任务完成,它则将其放到预备状态的列表,否则,放在等待状态的列表,然后遍历等待状态的列表,查看它们是否完成,而原先在预备状态列表的任务位置仍旧不变,因为它们还未运行。当所有任务被重新放置在合适的列表后,新一轮的循环又开始了。

4.3 如何使用?
  1. 导入内置库 asyncio

  2. async 修饰词声明异步函数,调用异步函数,便可得到一个协程对象

  3. 协程的执行:

    • 通过await来调用,执行效果和正常执行一样,会阻塞在这里,进入被调用的协程函数,执行完毕返回后再继续
  • asynio.creat_task(调用异步函数) 创建任务
    • asynio.run(main())作为主程序的入口函数,触发运行
import asyncio

async def crawl_page(url):
    print(‘crawling {}‘.format(url))
    sleep_time = int(url.split(‘_‘)[-1])
    await asyncio.sleep(sleep_time) # 从当前任务切出,事件调读器开始调度
    print(‘OK {}‘.format(url)) # 任务完成后,从事件循环中退出

async def main(urls):
    tasks = [asyncio.create_task(crawl_page(url)) for url in urls] # 列表生成式
    for task in tasks: # 多个任务被创建,进入事件循环等待运行
        await task # 执行,用户选择从当前主任务中切出,事件调度器开始调度
   # await asyncio.gather(*tasks)

asyncio.run(main([‘url_1‘, ‘url_2‘, ‘url_3‘, ‘url_4‘]))

########## 输出 ##########

crawling url_1
crawling url_2
crawling url_3
crawling url_4
OK url_1
OK url_2
OK url_3
OK url_4
import asyncio

async def worker_1():
    await asyncio.sleep(1) # 4.从当前任务切出,事件调读器开始调度任务2
    return 1 # 7.1秒后,事件调读器将控制权重新传给任务1,返回1,任务1完成,从事件循环中退出,并把控制器传给主任务

async def worker_2():  # 协程运行时出现错误
    await asyncio.sleep(2) #5.从当前任务切出,事件调读器开始调度任务3,
    return 2 / 0 # 8.2秒后,事件调读器将控制器重新传给任务2,运行出错,从事件循环中退出,控制器传给主任务

async def worker_3():
    await asyncio.sleep(3) # 6.从当前任务中切出,事件调读器暂停调度
    return 3 # 9.触发限定运行规则,任务3被取消,退出事件循环

async def main():
    task_1 = asyncio.create_task(worker_1()) # 2.任务123被创建,并进入事件循环等待运行
    task_2 = asyncio.create_task(worker_2())
    task_3 = asyncio.create_task(worker_3())

    await asyncio.sleep(2)  # 给任务3限定运行时间,一旦超时就取消
    task_3.cancel()

    res = await asyncio.gather(task_1, task_2, task_3, return_exceptions=True) # 3.执行任务,用户选择从当前的主任务中切出,事件调读器开始调度任务1
    print(res) # 10.主任务输出res,协程任务结束,事件循环结束

asyncio.run(main())  # 1.程序进入main()函数,事件循环开启

########## 输出 ##########

[1, ZeroDivisionError(‘division by zero‘), CancelledError()]
4.4. 协程的优点
  • 协程是单线程的,但其内部 event loop 的机制,可以让它并发地运行多个不同的任务,并且比多线程享有更大的自主控制权。
  • Asyncio 中的任务,在运行过程中不会被打断,因此不会出现 race condition 的情况
  • 子程序切换不是线程切换,而是由程序自身控制,在哪些地方交出控制权,切换到下一个任务,因此没有线程切换的开销,具有极高的执行效率
  • 协程的写法更加清晰简洁,把async/await 语法和 create_task结合来用,对于中小级别的并发需求已经毫无压力
  • 很多情况下,使用 Asyncio 需要特定第三方库的支持(缺点),例如不支持requests,但可以用aiohttp

6. 选择多线程还是协程

  • 如果是 I/O bound,并且 I/O 操作很慢,需要很多任务 / 线程协同实现,那么使用 Asyncio 更合适。
  • 如果是 I/O bound,但是 I/O 操作很快,只需要有限数量的任务 / 线程,那么使用多线程就可以了。
  • 如果是 CPU bound,则需要使用多进程来提高程序运行效率

以上是关于Python并发编程——多线程与协程的主要内容,如果未能解决你的问题,请参考以下文章

Python:线程进程与协程——线程池

Python:线程进程与协程——概念

帮你搞懂Python进程,线程与协程

写给Python社群的第11课:Python线程,进程,协程,3个毫无关系的兄弟

写给Python社群的第11课:Python线程,进程,协程,3个毫无关系的兄弟

Python:线程进程与协程——multiprocessing模块