【Python】使用 Python 中的 aioredis 实现后进先出队列

引言

在现代的分布式系统中,使用消息队列是一种常见的方式来实现异步通信和任务处理。Redis 是一种流行的内存数据库,它的列表数据结构非常适合用作消息队列。本文将介绍如何使用 Python 和 aioredis 库来连接 Redis 数据库,并实现一个基于 Redis 的连接池和一个后进先出队列。

python 复制代码
import aioredis


class RedisPool:
    _instances = {}
    def __new__(cls, redisURL):
        if redisURL not in cls._instances:
            cls._instances[redisURL] = super().__new__(cls)
            cls._instances[redisURL]._pool = None
            return cls._instances[redisURL]
        else:
            raise cls._instances[redisURL]

    def __init__(self, redisURL):
        self._pool = None
        self.redisURL = redisURL

    async def get_pool(self):
        if self._pool is None:
            self._pool = await aioredis.from_url(self.redisURL)
        return self._pool

class RedisQueue:
    def __init__(self, redisPool, queueName):
        self._pool = redisPool
        self._queueName = queueName

    async def push(self, data):
        pool = await self._pool.get_pool()
        await pool.lpush(self._queueName, data)

    async def pop(self):
        pool = await self._pool.get_pool()
        return await pool.lpop(self._queueName)

    async def size(self):
        pool = await self._pool.get_pool()
        return await pool.llen(self._queueName)


async def main():
    # 带密码的redis redis://:password@localhost:6379/0
    redisPool = RedisPool('redis://:password@233.233.233.233:33333/0')
    queue = RedisQueue(redisPool, 'testQueue')

    for i in range(10):
        print(f"push {i}")
        await queue.push(str(i))
    print(f"size {await queue.size()}")
    for i in range(10):
        print(await queue.pop())
    print(f"size {await queue.size()}")


if __name__ == '__main__':
    import asyncio
    asyncio.run(main())
相关推荐
coding随想2 分钟前
掌控网页的魔法之书:JavaScript DOM的奇幻之旅
开发语言·javascript·ecmascript
Norvyn_711 分钟前
LeetCode|Day18|20. 有效的括号|Python刷题笔记
笔记·python·leetcode
爱吃烤鸡翅的酸菜鱼21 分钟前
IDEA高效开发:Database Navigator插件安装与核心使用指南
java·开发语言·数据库·编辑器·intellij-idea·database
chao_78929 分钟前
更灵活方便的初始化、清除方法——fixture【pytest】
服务器·自动化测试·python·pytest
心情好的小球藻1 小时前
Python应用进阶DAY9--类型注解Type Hinting
开发语言·python
都叫我大帅哥1 小时前
LangChain加载HTML内容全攻略:从入门到精通
python·langchain
惜.己1 小时前
使用python读取json数据,简单的处理成元组数组
开发语言·python·测试工具·json
Y4090011 小时前
C语言转Java语言,相同与相异之处
java·c语言·开发语言·笔记
都叫我大帅哥3 小时前
Python的Optional:让你的代码优雅处理“空值”危机
python
曾几何时`5 小时前
基于python和neo4j构建知识图谱医药问答系统
python·知识图谱·neo4j