使用python 3.6.12和aioredis 2.0.0, asyncio 3.4.3
尝试使用数据库中的代码片段来测试pub/sub:
import asyncio
import aioredis
async def reader(ch):
while (await ch.wait_message()):
msg = await ch.get_json()
print("Got Message:", msg)
async def main():
pub = await aioredis.create_redis(
'redis://:password@localhost:6379')
sub = await aioredis.create_redis(
'redis://:password@localhost:6379')
res = await sub.subscribe('chan:1')
ch1 = res[0]
tsk = asyncio.ensure_future(reader(ch1))
res = await pub.publish_json('chan:1', ["Hello", "world"])
assert res == 1
await sub.unsubscribe('chan:1')
await tsk
sub.close()
pub.close()
if __name__ == '__main__':
loop = asyncio.get_event_loop()
result = loop.run_until_complete(main())
但是下面的错误一直弹出
Traceback (most recent call last):
File "tests/test_async_redis.py", line 32, in <module>
result = loop.run_until_complete(main())
File "/Users/dustinlee/.pyenv/versions/3.6.12/lib/python3.6/asyncio/base_events.py", line 488, in run_until_complete
return future.result()
File "tests/test_async_redis.py", line 12, in main
pub = await aioredis.create_redis(
AttributeError: module 'aioredis' has no attribute 'create_redis'
谁能告诉我我做错了什么?可能是很明显的事,只是我没发现。谢谢!aioredis
2.0版本现在遵循库redis-py
的公共API实现。
摘自aioredis
文档页
aioredis v2.0现在是一个完全兼容的redis-py的异步本地实现。整个核心和公共API已被重写,以尽可能地遵循redis-py的实现。
因此,aioredis.create_redis
方法不再是2.0版本中可以用来建立连接的公共API。如果你想让create_redis
方法工作,请使用小于2的版本。
您可以参考新的pub子示例。
这里复制的代码,以防将来链接不能工作。
import asyncio
import async_timeout
import aioredis
STOPWORD = "STOP"
async def pubsub():
redis = aioredis.Redis.from_url(
"redis://localhost", max_connections=10, decode_responses=True
)
psub = redis.pubsub()
async def reader(channel: aioredis.client.PubSub):
while True:
try:
async with async_timeout.timeout(1):
message = await channel.get_message(ignore_subscribe_messages=True)
if message is not None:
print(f"(Reader) Message Received: {message}")
if message["data"] == STOPWORD:
print("(Reader) STOP")
break
await asyncio.sleep(0.01)
except asyncio.TimeoutError:
pass
async with psub as p:
await p.subscribe("channel:1")
await reader(p) # wait for reader to complete
await p.unsubscribe("channel:1")
# closing all open connections
await psub.close()
async def main():
tsk = asyncio.create_task(pubsub())
async def publish():
pub = aioredis.Redis.from_url("redis://localhost", decode_responses=True)
while not tsk.done():
# wait for clients to subscribe
while True:
subs = dict(await pub.pubsub_numsub("channel:1"))
if subs["channel:1"] == 1:
break
await asyncio.sleep(0)
# publish some messages
for msg in ["one", "two", "three"]:
print(f"(Publisher) Publishing Message: {msg}")
await pub.publish("channel:1", msg)
# send stop word
await pub.publish("channel:1", STOPWORD)
await pub.close()
await publish()
if __name__ == "__main__":
import os
if "redis_version:2.6" not in os.environ.get("REDIS_VERSION", ""):
asyncio.run(main())
你也可以参考redis-py
文档,因为它应该是aioredis
2.0现在密切遵循的。