一日一技:在 Python 中实现延迟调用

本文涉及的产品
云数据库 Tair(兼容Redis),内存型 2GB
Redis 开源版,标准版 2GB
推荐场景:
搭建游戏排行榜
简介: 一日一技:在 Python 中实现延迟调用

摄影:产品经理产品经理的生日餐

熟悉 Golang 的同学都知道,Golang 里面有一个关键词叫做defer,它可以实现延迟调用。

实际上在 Python 里面也有相关的语法,那就是contextlib.ExitStack

我们来看这样一个场景:

我有一个函数parse,它的作用是从 Redis 中持续读入数据,并写入到MongoDB 中。示例代码如下:

import json
import redis
import pymongo
client = redis.Redis()
handler = pymongo.MongoClient().test.data
def parse():
    data = client.lpop('test')
    ifnot data:
        return
    handler.insert_one(json.loads(data))

但现在我想增加一个需求,当Redis 读取结束或者读取数据报错的时候,能把当前的时间也写入到MongoDB 中。

那么代码可能变成下面这样:

import json
import redis
import datetime
import pymongo
client = redis.Redis()
handler = pymongo.MongoClient().test.data
def parse():
    whileTrue:
        try:
            data = client.lpop('test')
            ifnot data:
                handler.insert_one({'finished': True, 'ts': datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')})
                return
            handler.insert_one(json.loads(data))
        except Exception:
            handler.insert_one({'finished': True, 'ts': datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S'})

可以看到,代码变得很难看了。

现在,我们可以使用延迟调用来让代码变得更好看。

要实现这个目的,就可以开始使用ExitStack了。它可以注册多个回调函数,在退出上下文缩进时执行。

我们先来看一个简单的例子:

import contextlib
def callback_1():
    print('我是第一个回调函数')
def callback_2(x):
    print(f'我是第二个回调函数,传入参数:{x}')
with contextlib.ExitStack() as stack:
    stack.callback(callback_1)
    stack.callback(callback_2, 100)
    print(12345)
    print('xxxx')
print('退出缩进')

运行效果如下图所示:

可以看出以下特点:

  1. 被添加的回调函数进入了一个栈,所以后添加的回调函数先调用
  2. 回调函数会在结束缩进的时候被调用

现在我们来人工构造一个异常:

可以看到,即使缩进里面出现了报错,回调函数仍然可以正常运行。等所有回调函数运行完成以后,Python 才会退出。

基于以上特点,我们就可以来重构最开始的代码了:

import json
import redis
import datetime
import pymongo
import contextlib
client = redis.Redis()
handler = pymongo.MongoClient().test.data
def add_ts():
    handler.insert_one({'finished': True, 'ts': datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')})
def parse():
    with contextlib.ExitStack() as stack:
        stack.callback(add_ts)
        whileTrue:
            data = client.lpop('test')
            ifnot data:
                return
            handler.insert_one(json.loads(data))

无论是正常运行结束还是运行过程中报错,add_ts函数都会正常运行,确保始终增加一条日期数据。

相关实践学习
基于Redis实现在线游戏积分排行榜
本场景将介绍如何基于Redis数据库实现在线游戏中的游戏玩家积分排行榜功能。
云数据库 Redis 版使用教程
云数据库Redis版是兼容Redis协议标准的、提供持久化的内存数据库服务,基于高可靠双机热备架构及可无缝扩展的集群架构,满足高读写性能场景及容量需弹性变配的业务需求。 产品详情:https://www.aliyun.com/product/kvstore     ------------------------------------------------------------------------- 阿里云数据库体验:数据库上云实战 开发者云会免费提供一台带自建MySQL的源数据库 ECS 实例和一台目标数据库 RDS实例。跟着指引,您可以一步步实现将ECS自建数据库迁移到目标数据库RDS。 点击下方链接,领取免费ECS&RDS资源,30分钟完成数据库上云实战!https://developer.aliyun.com/adc/scenario/51eefbd1894e42f6bb9acacadd3f9121?spm=a2c6h.13788135.J_3257954370.9.4ba85f24utseFl
目录
相关文章
|
3月前
|
网络协议 Python
python requests库如何使用http连接池降低延迟 keepalive复用连接
Python的`requests`库通过内置的连接池机制支持HTTP Keep-Alive特性,允许复用TCP连接以发送多个请求,减少连接开销。默认情况下,`requests`不显式禁用Keep-Alive,其行为取决于底层HTTP库(如urllib3)及服务器的支持。通过创建`Session`对象并自定义`HTTPAdapter`,可以调整连接池大小和重试策略,进一步优化连接复用。测试显示,使用`Session`和定制的`HTTPAdapter`比普通请求方法能显著减少连续请求间的时间消耗,体现了Keep-Alive的优势。
|
4月前
|
Shell 索引 Python
Python的延迟绑定是什么?
Python中的延迟绑定指的是在嵌套函数中,内部函数在被调用时才绑定外部函数的变量,而不是在定义时绑定。这可能导致意外行为,因为变量的值是在函数调用时决定的。
|
数据采集 Shell 开发工具
[oeasy]python0035_ 整合shell编程_循环_延迟_清屏
[oeasy]python0035_ 整合shell编程_循环_延迟_清屏
103 0
 [oeasy]python0035_ 整合shell编程_循环_延迟_清屏
|
数据采集 Shell 开发工具
[oeasy]python0027_整合程序_延迟输出时间_整合两个py程序
[oeasy]python0027_整合程序_延迟输出时间_整合两个py程序
86 0
[oeasy]python0027_整合程序_延迟输出时间_整合两个py程序
|
Python
[oeasy]python0026_刷新时间_延迟时间_time_sleep_死循环_while_True
[oeasy]python0026_刷新时间_延迟时间_time_sleep_死循环_while_True
101 0
[oeasy]python0026_刷新时间_延迟时间_time_sleep_死循环_while_True
|
NoSQL Go MongoDB
一日一技:在 Python 中实现延迟调用
一日一技:在 Python 中实现延迟调用
277 0
一日一技:在 Python 中实现延迟调用
|
Python
python检测网络延迟
#!/usr/bin/env python # coding: utf-8 # coding: cp950 ''' Create Date: 2012-11-06 Version: 1.
2369 0
|
13天前
|
存储 程序员 开发者
Python编程基础:从入门到实践
【10月更文挑战第8天】在本文中,我们将一起探索Python编程的奇妙世界。无论你是初学者还是有一定经验的开发者,这篇文章都将为你提供有价值的信息。我们将从Python的基本概念开始,然后逐步深入到更复杂的主题,如数据结构、函数和类。最后,我们将通过一些实际的代码示例来巩固我们的知识。让我们一起开始这段Python编程之旅吧!
|
1天前
|
弹性计算 安全 小程序
编程之美:Python让你领略浪漫星空下的流星雨奇观
这段代码使用 Python 的 `turtle` 库实现了一个流星雨动画。程序通过创建 `Meteor` 类来生成具有随机属性的流星,包括大小、颜色、位置和速度。在无限循环中,流星不断移动并重新绘制,营造出流星雨的效果。环境需求为 Python 3.11.4 和 PyCharm 2023.2.5。
20 9