简单聊聊Python协程

罗罗攀

发布于 2021-12-06 09:56:46

26200

代码可运行

文章被收录于专栏：有趣的Python和你有趣的Python和你

运行总次数：0

代码可运行

前言

今天我们来聊聊Python协程，当Python学习到一定的深度，当你需要对代码进行优化提速时，就避不开异步编程，尤其是现在优秀的第三方库都实现了异步编程，这使得我们不得不学习。

本人也是最近开始学习这个概念，本次分享我们就来简单聊聊Python协程的概念和最基本的语法。

什么是协程

我们先来看看协程的概念。

协程（Coroutine），也可以被称为微线程，是一种用户态内的上下文切换技术。

概念是不是很不好理解，没关系，我们引入一个小学就学过的数学思考题。小明早上起床需要刷牙和烧水，刷牙需要5分钟，烧水需要8分钟，问小明总共需要几分钟完成这两件事情。

答案是8分钟，这你应该是知道的吧。其实这就是协程，现在我再来解释下，刷牙是一个函数，烧水时一个函数，那协程就是通过一个线程实现代码块相互切换执行。

Python实现协程

我们首先来看看不用协程怎么来写。

import time


def brush_teeth():
    print('开始刷牙')
    time.sleep(5)
    print('刷牙完成')


def boil_water():
    print('开始烧水')
    time.sleep(8)
    print('烧水完成')


def main():
    start = time.time()
    brush_teeth()
    boil_water()
    end = time.time()
    print('花费时间{}s'.format(end-start))


main()

开始刷牙
刷牙完成
开始烧水
烧水完成
花费时间13.01053786277771s

看到没，需要13s，那我们来看看协程怎么写。

import asyncio
import time


async def brush_teeth():
    print('开始刷牙')
    await asyncio.sleep(5)
    print('刷牙完成')


async def boil_water():
    print('开始烧水')
    await asyncio.sleep(8)
    print('烧水完成')


async def main():
    start = time.time()
    tasks = [
        asyncio.create_task(boil_water()),
        asyncio.create_task(brush_teeth())
    ]
    await asyncio.wait(tasks)
    end = time.time()
    print('花费时间{}s'.format(end-start))


asyncio.run(main())

开始烧水
开始刷牙
刷牙完成
烧水完成
花费时间8.004523992538452s