frobnitzem / aiowire

aiowire - A simple event loop using asyncio

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

aiowire - A simple event loop using asyncio

This package implements a EventLoop class that manages concurrent coroutines.

It is based on the principles of functional reactive programming and draws inspiration from Haskell's Control.Wire library.

In particular, every co-routine started by the event loop is a Wire.

Wire-s either return None, indicating they're done, or another Wire.

An example helps explain the idea:

from aiowire import EventLoop

event = 0
async def show_event(ev) \
        -> Optional[Callable[[EventLoop],Awaitable]]:
    print("Running...")
    event += 1
    await asyncio.sleep(event*0.15)
    print(f"Event {event}")
    if event < 5:
        return show_event

async with EventLoop(timeout=1) as event:
    event.start(show_event)
    event.start(show_event)

We start up an event loop and drop in two wires. Each runs, then returns the show_event function. The event loop runs those functions next... and so on.

But this isn't functional programming. The wires have access to the event loop, and can start more tasks. Easy, right?

What can I do with it?

What if you have a server that's spawning programs, working with sockets, and managing timeouts? Drop in one wire for each program, one polling on socket I/O, and another acting as a timer (as above).

The canonical task types are thus:

asyncio.create_subprocess_exec # run a process

asyncio.sleep # awake the loop after a given time lapse

zmq.asyncio.Poller.poll # awake the loop after I/O on socket/file
# Note: see aiowire.Poller for a Wire-y interface.

Now your sockets can launch programs, and your program results can start/stop sockets, and everyone can start background tasks.

Poller?

The Poller class lets you schedule callbacks in response to socket or file-descriptor activity. Of course, the callbacks are wires, and run concurrently.

Poller is also a Wire, created as, Poller(dictionary mapping sockets / fd-s to callback wires).

You add it to your event loop as usual:

# ... create sock from zmq.asyncio.Context

async def echo(ev):
    await sock.send( await sock.recv() )

todo = { 0:  Call(print, "received input on sys.stdin"),
         sock: echo
       }
async with EventLoop() as ev:
    ev.start( Poller(todo) )

Tell me more

Yes, you could just send async functions taking one argument to EventLoop.start, but where's the fun in writing closures everywhere?

To take it to the next level, aiowire comes with a Wire convenience class that lets you write Wire-s expressively. The following class extensions help you make Wire-s out of common programming idioms:

  • Wire(w): acts like an identity over "async func(ev):" functions
  • Repeat(w, n): repeat wire w n times in a row
  • Forever(w): repeat forever -- like Repeat(w) * infinity
  • Call(fn, *args): call fn (normal or async), ignore the return, and exit

Consider, for example, printing 4 alarms separated by some time interval:

from aiowire import EventLoop, Call

prog = ( Call(asyncio.sleep, 0.1) >> Call(print, 'beep\a') ) * 4

async with EventLoop() as ev:
    ev.start(prog)

References

About

aiowire - A simple event loop using asyncio

License:GNU Lesser General Public License v3.0


Languages

Language:Python 100.0%