aioredis 使用

import asyncio
import aioredis


# 创建连接池
async def create_redis_pools():
    pools = {}
    for db_index in range(8):
        pool = await aioredis.create_redis_pool(
            'redis://localhost',
            db=db_index,
            minsize=5,
            maxsize=10
        )
        pools[db_index] = pool
    return pools


# 使用管道查询多个键
async def get_keys_with_pipeline(pool, *keys):
    async with pool.get() as conn:
        pipeline = conn.pipeline()
        for key in keys:
            pipeline.get(key)
        results = await pipeline.execute()
    return dict(zip(keys, [result.decode() if result else None for result in results]))


# 使用管道增加多个键
async def set_keys_with_pipeline(pool, **key_values):
    async with pool.get() as conn:
        pipeline = conn.pipeline()
        for key, value in key_values.items():
            pipeline.set(key, value)
        await pipeline.execute()


# 关闭所有连接池
async def close_redis_pools(pools):
    for pool in pools.values():
        pool.close()
        await pool.wait_closed()


async def get_keys_by_pattern(redis, pattern):
    keys = []
    async for key in redis.iscan(match=pattern):
        keys.append(key)
    return keys


async def get_values_for_keys(redis, keys):
    values = await redis.mget(*keys)
    return dict(zip(keys, values))


async def find_keys_and_get_values(redis, pattern):
    # 查找所有匹配指定模式的键
    keys = await get_keys_by_pattern(redis, pattern)
    # 获取这些键对应的值
    key_values = await get_values_for_keys(redis, keys)
    return key_values


# 主函数
async def main():
    # 创建连接池
    pools = await create_redis_pools()

    # 假设我们要在数据库0中操作键
    db_index = 0
    pool = pools[db_index]

    # 使用管道增加多个键
    await set_keys_with_pipeline(pool, key1='value1', key2='value2', key3='value3')

    # 使用管道查询多个键
    values = await get_keys_with_pipeline(pool, 'key1', 'key2', 'key3')
    print(f"Got values with pipeline: {values}")


asyncio.run(main())