2017-08-09 69 views
0

我疑惑得到以下與ASYNCIO工作:從期貨ASYNCIO

下面的代碼片段被查詢多個設備(通過SNMP),並返回一個字典,它工作正常,但由multiprocessing.cpu_count()

限制
def do_polling(netelement, snmp_comm): 
    msg = {} 
    msg.update({ 
     'bgp'  : do_lookup_bgp(netelement, snmp_comm), 
     'iface'  : do_lookup_iface(netelement, snmp_comm), 
     'ifidx'  : do_lookup_ifindex(netelement, snmp_comm), 
     'agg'  : do_lookup_agg(netelement, snmp_comm), 
    }) 
    return msg 

def save(netelement, job): 
    data[netelement] = job.result() 

with concurrent.futures.ProcessPoolExecutor(max_workers=multiprocessing.cpu_count()) as executor: 
    for k,v in INFO['dev'].items(): 
     job = executor.submit(do_polling, k, v['snmp_comm']) 
     job.add_done_callback(functools.partial(save, k)) 

,所以我想通過改變這樣的遷移到asyncio方法:

@asyncio.coroutine 
def do_polling(netelement, snmp_comm): 
    msg = {} 
    msg['bgp']  = yield from do_lookup_bgp(netelement, snmp_comm) 
    msg['iface'] = yield from do_lookup_iface(netelement, snmp_comm) 
    msg['ifidx'] = yield from do_lookup_ifindex(netelement, snmp_comm) 
    msg['agg']  = yield from do_lookup_agg(netelement, snmp_comm) 

@asyncio.coroutine 
def schedule(INFO): 
    for k,v in INFO['dev'].items(): 
     asyncio.async(do_polling(k, v)) 

asyncio.get_event_loop().run_until_complete(schedule) 

但我得到以下埃羅r:

Traceback (most recent call last): 
    File "/home/app/ip-spotlight/code/ixmac.py", line 60, in <module> 
    main() 
    File "/home/app/ip-spotlight/code/ixmac.py", line 16, in main 
    app.ixmac.initialize.run(INFO) 
    File "/home/app/ip-spotlight/code/app/ixmac/initialize.py", line 191, in run 
    asyncio.get_event_loop().run_until_complete(schedule) 
    File "/usr/lib64/python3.4/asyncio/base_events.py", line 353, in run_until_complete 
    future = tasks.ensure_future(future, loop=self) 
    File "/usr/lib64/python3.4/asyncio/tasks.py", line 553, in ensure_future 
    raise TypeError('A Future, a coroutine or an awaitable is required') 
TypeError: A Future, a coroutine or an awaitable is required 

你能否告訴我我做錯了什麼?

回答

3

您並未將其用作協同程序。你應該改變最後一行:

asyncio.get_event_loop().run_until_complete(schedule(the_info_variable)) 
+0

謝謝:)你介不介意幫我用'edit'以及 – iamsterdam

+0

@iamsterdam你說的'我等待(塊)的意思before' –

+0

的'do_polling'函數返回一個保存在'data [netelement]'中的字典。我看到發生了'do_polling'調用'do_lookup_bgp',然後'do_data_wrangling'被調用。我想要發生的事情是:'do_polling'爲每個'netelement'調用'do_lookup_bgp','do_lookup_iface','do_lookup_ifindex','do_lookup_agg',然後'do_data_wrangling'來調用 – iamsterdam