Python异步等待条件为真

Question

我正在尝试编写一个异步方法来在 hive 上运行查询（使用 pyhive）。现在，pyhive 确实支持异步查询，我不知道如何等待查询完成而不阻塞。

我可以通过反复检查等待查询完成，但这与阻塞基本相同。

def runQuery():
    cursor = hive.connect('localhost').cursor()
    cursor.execute('select * from mytable', async_ = True)
    status = cursor.poll().operationState
    while status in (TOperationState.INITIALIZED_STATE, TOperationState.RUNNING_STATE):
        status = cursor.poll().operationState
    return cursor.fetchall()

所以我使用异步，但是我不知道如何等待。我尝试了下面的代码，但它抛出了TypeError: object int can't be used in 'await' expression

async def runQueryAsync():
    cursor = hive.connect('localhost').cursor()
    cursor.execute('select * from mytable', async_ = True)
    #THIS DOESN'T WORK 
    await cursor.poll().operationState not in (TOperationState.INITIALIZED_STATE, TOperationState.RUNNING_STATE)
    return cursor.fetchall()

有什么解决方法吗？基本上我想要一种方法，而不是说等待方法调用，我说等待直到这个条件为真

PS：澄清一下，cursor.execute('select * from mytable', async_ = True) 在返回协程/未来的 Python 意义上不是异步的。它只是启动一个查询并立即返回，您必须检查状态才能知道查询是否完成。所以await cursor.execute('select * from mytable', async_ = True) 不起作用。

Answer 1

您必须积极等待：

async def runQueryAsync():
    cursor = hive.connect('localhost').cursor()
    await cursor.execute('select * from mytable', async_ = True)
    while cursor.poll().operationState not in (TOperationState.INITIALIZED_STATE, TOperationState.RUNNING_STATE):
        await asyncio.sleep(1) # try each 1 second
    return cursor.fetchall()

我不确定你是否可以await cursor.execute('select * from mytable', async_ = True)，但如果不只是使用cursor.execute('select * from mytable', async_ = True)，尽管在那里使用它是有意义的。如果它在执行时与await 一起使用，您可能不需要使用while 循环，因为它应该在执行完成后继续：

async def runQueryAsync():
    cursor = hive.connect('localhost').cursor()
    await cursor.execute('select * from mytable', async_ = True)
    return cursor.fetchall()