python 3：使用异步任务时如何在异步 function 中等待回调

Question

AWS IoT python Sdk 有一个 function publishAsync
function 的签名是 publishAsync(topic, payload, QoS, ackCallback=None)

我想把它变成一个协程并使用 asyncio。

我想做的是：

1. 连接到 AWS
2. 异步发布 10 条消息
3. 断开

我不知道我应该如何将 function 包装成异步 function。

async def asyncPublish(self,
                     msg:str,
                     topic:str,
                     QoS=1):

        # the publishAckFn
        def internalPubAckHandler(mid):
            print(json.dumps({
                'messageID':mid,
                'acknowledged':True
            }))
            return True
        pass

        # publish to the topic asynchronously
        messageID = self.awsIoTClient.publishAsync(topic,msg,QoS,ackCallback=internalPubAckHandler)
        print(json.dumps({
           'messageID':messageID,
           'topic':topic,
           'payload':msg,
           'QoS':QoS,
           'async':True
        }))

    pass


---- in my main file
tasks = []
for i in range(10):
        tasks += asyncPublish('test','test')
pass

loop = asyncio.get_event_loop()
loop.set_debug(True)  # debug mode
gw1.connect()
loop.run_until_complete(asyncio.wait(tasks))
loop.close()
gw1.disconnect()  

但是当我这样做时，断开连接将比 internalPubAckHandler 更快地调用，我不会得到任何确认。

Answer 1

最后，我找到了解决方案。 我们可以使用期货。

Future 对象用于将基于回调的低级代码与高级异步/等待代码连接起来。 参考

如果 function 的签名是这样的：

def functionWithCallback(args,callback)

我们需要将其包装到等待回调的异步 function 中，我们可以执行以下操作：

async def asyncVersion(args):

    # one way of creating a future
    loop = asyncio.get_running_loop()
    future = loop.create_future()

    def futureCallback():
        # do something here and then signal the future
        ...
        loop.call_soon_threadsafe(future .set_result,time.time())
    pass

    # call your function
    functionWithCallback(args,futureCallback)

    # wait for the callback
    await future


    # do other stuff after the callback is done
    .....

pass