相当于uvloop的asyncio事件循环

时间:2018-11-03 09:49:05

标签: python python-3.x async-await python-asyncio uvloop

我有一个使用asyncio的协程方法的事件循环。

我热衷于使用uvloop来查找与以下示例等效的示例。

这是一个简单的asyncio事件循环示例:

import asyncio

async def read(**kwargs):
    oid = kwargs.get('oid', '0.0.0.0.0.0')
    time = kwargs.get('time', 1)
    try:
        print('start: ' + oid)
    except Exception as exc:
        print(exc)
    finally:
        await asyncio.sleep(time)
        print('terminate: ' + oid)


def event_loop(configs):
    loop = asyncio.get_event_loop()

    for conf in configs:
        asyncio.ensure_future(read(oid=conf['oid'], time=conf['time']))

    return loop

if __name__ == '__main__':
    snmp_configurations = [
        {'time': 5, 'oid': '1.3.6.3.2.4'},
        {'time': 6, 'oid': '1.3.6.3.5.8'},
    ]  # TODO :: DUMMY
    loop = event_loop(snmp_configurations)
    try:
        loop.run_forever()
    except KeyboardInterrupt:
        pass
    finally:
        print("Closing Loop")
        loop.close()

问题:

  • 如何使用uvloop重新构造上述代码段代码?

  • 以下更改是否正确使用uvloop并具有更高的性能?

    import uvloop
    
    def event_loop(configs):
        asyncio.set_event_loop_policy(uvloop.EventLoopPolicy())  # TODO  :: uvloop.
        loop = asyncio.get_event_loop()
    
     `   for conf in configs:
            asyncio.ensure_future(read(oid=conf['oid'], time=conf['time']))
    
        return loop
    

[注意]:

  • uvloop声称可以使asyncio加快2-4倍。

1 个答案:

答案 0 :(得分:3)

只需在调用asyncio.get_event_loop()之前设置事件循环策略即可。

import asyncio
import uvloop
asyncio.set_event_loop_policy(uvloop.EventLoopPolicy())

async def read(**kwargs):
    oid = kwargs.get('oid', '0.0.0.0.0.0')
    time = kwargs.get('time', 1)
    try:
        print('start: ' + oid)
    except Exception as exc:
        print(exc)
    finally:
        await asyncio.sleep(time)
        print('terminate: ' + oid)


def event_loop(configs):
    loop = asyncio.get_event_loop()

    for conf in configs:
        asyncio.ensure_future(read(oid=conf['oid'], time=conf['time']))

    return loop

if __name__ == '__main__':
    snmp_configurations = [
        {'time': 5, 'oid': '1.3.6.3.2.4'},
        {'time': 6, 'oid': '1.3.6.3.5.8'},
    ]  # TODO :: DUMMY
    loop = event_loop(snmp_configurations)
    try:
        loop.run_forever()
    except KeyboardInterrupt:
        pass
    finally:
        print("Closing Loop")
        loop.close()

是,此代码正确。您可以在导入后设置事件循环策略。

import uvloop
import asyncio
asyncio.set_event_loop_policy(uvloop.EventLoopPolicy())  # TODO  :: uvloop.

def event_loop(configs):

    loop = asyncio.get_event_loop()

    for conf in configs:
        asyncio.ensure_future(read(oid=conf['oid'], time=conf['time']))

    return loop