简单聊聊Python协程

前言

今天我们来聊聊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

用了协程后，程序就只需要8s了。所以协程很有用，协程一般应用在有IO操作的程序中，因为协程可以利用IO等待的时间去执行一些其他的代码，从而提升代码执行效率。

代码解释

最好，我们来解释一下代码。

（1）asyncio.run(main())，进入main函数，事件循环开始。

这里有必要解释一下事件循环，事件循环可以看做成while循环，一直循环着任务，当任务完成时，才会终止循环。

（2）tasks列表里面就创建了两个任务，这两个任务就进入到了事件循环，准备执行。

这里创建的任务就是协程对象，当用async关键字定义的函数就是协程函数，调用这个函数，返回的就是协程对象。

（3）await asyncio.wait(tasks)就会进入任务中，首先进入烧水任务，遇到IO（这里是睡眠），就跳入到另外一个一个任务（刷牙），然后又遇到IO，这里没有其他任务了，所以等待完成刷牙任务，就跳入到烧水任务，完成后，整个事件循环就结束了。

awit关键词就是等待的意思，后面接的是IO等待。

今天的分享就到这了，如果我的文章对你有帮助，别忘了点赞，收藏，转发，这对我有很大的帮助，我们下期再见~