在 Twisted 回调中使用 async/await 语法

Question

我想将async/await语法与 Twisted Deferred.addCallback方法一起使用。但正如文档中所述，addCallback回调是同步调用的。我见过用于此目的的inlineCallbacks装饰器，但我更喜欢使用async/await语法（如果它甚至可能或有意义的话）。

我从pika 文档中获取了原始代码，但我没有成功尝试将其迁移到 async/await 语法：

import pika
from pika import exceptions
from pika.adapters import twisted_connection
from twisted.internet import defer, reactor, protocol, task


async def run_async(connection):
    channel = await connection.channel()
    exchange = await channel.exchange_declare(exchange='topic_link',type='topic')
    queue = await channel.queue_declare(queue='hello', auto_delete=False, exclusive=False)
    await channel.queue_bind(exchange='topic_link', queue='hello', routing_key='hello.world')
    await channel.basic_qos(prefetch_count=1)
    queue_object, consumer_tag = await channel.basic_consume(queue='hello', no_ack=False)
    l = task.LoopingCall(read_async, queue_object)
    l.start(0.01)


async def read_async(queue_object):
    ch,method,properties,body = await queue_object.get()
    if body:
        print(body)
    await ch.basic_ack(delivery_tag=method.delivery_tag)


parameters = pika.ConnectionParameters()
cc = protocol.ClientCreator(reactor, twisted_connection.TwistedProtocolConnection, parameters)
d = cc.connectTCP('rabbitmq', 5672)
d.addCallback(lambda protocol: protocol.ready)
d.addCallback(run_async)
reactor.run()

这显然不起作用，因为没有人等待run_async功能。

Answer 1

正如notorious.no和Twisted 文档所指出的那样，这ensureDeferred是要走的路。不过，您必须包装回调结果，而不是我不清楚的回调本身。

这是它最终的样子：

def ensure_deferred(f):
    @functools.wraps(f)
    def wrapper(*args, **kwargs):
        result = f(*args, **kwargs)
        return defer.ensureDeferred(result)
    return wrapper


@ensure_deferred
async def run(connection):
    channel = await connection.channel()
    exchange = await channel.exchange_declare(exchange='topic_link', type='topic')
    queue = await channel.queue_declare(queue='hello', auto_delete=False, exclusive=False)
    await channel.queue_bind(exchange='topic_link', queue='hello', routing_key='hello.world')
    await channel.basic_qos(prefetch_count=1)
    queue_object, consumer_tag = await channel.basic_consume(queue='hello', no_ack=False)
    l = task.LoopingCall(read, queue_object)
    l.start(0.01)


@ensure_deferred
async def read(queue_object):
    ch, method, properties, body = await queue_object.get()
    if body:
        print(body)
    await ch.basic_ack(delivery_tag=method.delivery_tag)


parameters = pika.ConnectionParameters()
cc = protocol.ClientCreator(reactor, twisted_connection.TwistedProtocolConnection, parameters)
d = cc.connectTCP('rabbitmq', 5672)
d.addCallback(lambda protocol: protocol.ready)
d.addCallback(run)
reactor.run()

谢谢。