Python中使用装饰器装饰或优化异步函数需注意的问题

原创 码农  2020-02-11 10:43:58  阅读 45 次 评论 0 条

在 Python 开发的工程中,我们常常使用装饰器来优化代码,例如一个打印日志的装饰器:

import requests
import datetime


def time_log(func):
    def wrap(*args, **kwargs):
        print(f'现在时间:{datetime.datetime.now()},程序开始运行')
        result = func(*args, **kwargs)
        print(f'现在时间:{datetime.datetime.now()},运行结束')
        return result
    return wrap


@time_log
def get(n):
    resp = requests.get(f'http://httpbin.org/delay/{n}').text
    print(resp)


get(3)

运行效果如下图所示:

Python中使用装饰器装饰或优化异步函数需注意的问题 编程代码 第1张

但如果get是一个异步函数,这个装饰器就会出问题:

import pprint
import aiohttp
import datetime
import asyncio


def time_log(func):
    def wrap(*args, **kwargs):
        print(f'现在时间:{datetime.datetime.now()},程序开始运行')
        result = func(*args, **kwargs)
        print(f'现在时间:{datetime.datetime.now()},运行结束')
        return result
    return wrap

@time_log
async def get(n):
    async with aiohttp.ClientSession() as client:
        resp = await client.get(f'http://httpbin.org/delay/{n}')
        result = await resp.json()
    pprint.pprint(result, indent=2)


asyncio.run(get(3))

运行效果如下图所示:

Python中使用装饰器装饰或优化异步函数需注意的问题 编程代码 第2张

可以看到,print(f'现在时间:{datetime.datetime.now()},程序开始运行')print(f'现在时间:{datetime.datetime.now()},运行结束')几乎同时被打印出来,然后才是请求网络。这并不是我们需要实现的效果。我们想要的是先打印前一行,然后请求网络,再打印后一行。

为了解决这个问题,我们需要把time_log装饰器中的wrap也定义成异步函数:

def time_log(func):
    async def wrap(*args, **kwargs):
        print(f'现在时间:{datetime.datetime.now()},程序开始运行')
        result = await func(*args, **kwargs)
        print(f'现在时间:{datetime.datetime.now()},运行结束')
        return result
    return wrap

同时await func(*args, **kwargs)。这样就能保证代码的执行顺序:

Python中使用装饰器装饰或优化异步函数需注意的问题 编程代码 第3张

但需要注意的是,装饰器本身是一个同步函数,不需要使用async def来定义。只有里面的闭包需要定义为异步函数。

本文地址:https://www.itcodeit.com/post/72.html
版权声明:本文为原创文章,版权归 码农 所有,欢迎分享本文,转载请保留出处!

发表评论


表情

还没有留言,还不快点抢沙发?