web-dev-qa-db-ja.com

asyncioタスクとして実行される関数から値を取得する

私は次のコードを試していました:

import asyncio

@asyncio.coroutine
def func_normal():
        print("A")
        yield from asyncio.sleep(5)
        print("B")
        return 'saad'

@asyncio.coroutine
def func_infinite():
    i = 0
    while i<10:
        print("--"+str(i))
        i = i+1
    return('saad2')

loop = asyncio.get_event_loop()

tasks = [
    asyncio.async(func_normal()),
    asyncio.async(func_infinite())]

loop.run_until_complete(asyncio.wait(tasks))
loop.close()

これらの関数から変数の値を取得する方法がわかりません。これはできません。

asyncio.async(a = func_infinite())

これがキーワード引数になるためです。これを達成するにはどうすればよいですか?

37
Saad Aleem

コルーチンはそのまま機能します。 loop.run_until_complete()call asyncio.gather()から返された値を使用して、複数の結果を収集します

#!/usr/bin/env python3
import asyncio

@asyncio.coroutine
def func_normal():
    print('A')
    yield from asyncio.sleep(5)
    print('B')
    return 'saad'

@asyncio.coroutine
def func_infinite():
    for i in range(10):
        print("--%d" % i)
    return 'saad2'

loop = asyncio.get_event_loop()
tasks = func_normal(), func_infinite()
a, b = loop.run_until_complete(asyncio.gather(*tasks))
print("func_normal()={a}, func_infinite()={b}".format(**vars()))
loop.close()

出力

--0
--1
--2
--3
--4
--5
--6
--7
--8
--9
A
B
func_normal()=saad, func_infinite()=saad2
37
jfs

loop.run_until_completeは、渡した関数によって返された値を返します。したがって、asyncio.waitの出力を返します。

import asyncio

@asyncio.coroutine
def func_normal():
    print("A")
    yield from asyncio.sleep(5)
    print("B")
    return 'saad'

@asyncio.coroutine
def func_infinite():
    i = 0
    while i<10:
        print("--"+str(i))
        i = i+1
    return('saad2')

loop = asyncio.get_event_loop()

tasks = [
    asyncio.async(func_normal()),
    asyncio.async(func_infinite())]

done, _ = loop.run_until_complete(asyncio.wait(tasks))
for fut in done:
    print("return value is {}".format(fut.result()))
loop.close()

出力:

A
--0
--1
--2
--3
--4
--5
--6
--7
--8
--9
B
return value is saad2
return value is saad

tasks配列から結果に直接アクセスすることもできます。

print(tasks[0].result())
print(tasks[1].result())
6
dano

コルーチンが終了するとすぐにコルーチンによって返された値を使用する場合は、将来のオブジェクトをコロに渡し、計算された値によってこの将来を更新できます。 futureが更新されるとすぐに、そのfuture.result()が、指定されたfutureにバインドされたコールバック関数に渡されます。以下のコードを参照してください:

import asyncio


async def func_normal(future):
    print("A")
    await asyncio.sleep(5)
    print("B")
    # return 'saad'
    future.set_result('saad')


async def func_infinite(future):
    i = 0
    while i<10:
        print("--"+str(i))
        i = i+1
    # return('saad2')
    future.set_result('saad2')

def got_result(future):
    print(future.result())

loop = asyncio.get_event_loop()
future1 = asyncio.Future()
future2 = asyncio.Future()

future1.add_done_callback(got_result)
future2.add_done_callback(got_result)

# Coros are automatically wrapped in Tasks by asyncio.wait() 
coros = [
    func_normal(future1),
    func_infinite(future2)]

loop.run_until_complete(asyncio.wait(coros))
loop.close()

コールバック関数は、単一の引数で呼び出されます-バインドされる未来のオブジェクト。さらにコールバックに引数を渡す必要がある場合は、functoolsパッケージのpartialを使用します。

future1.add_done_callback(functools.partial(print, "future:", argin))

電話します

print("future:", argin)
2
Petr Javorik