Make spawn_n and map_n synchronous

These functions are specifically not blocking or awaiting, so they don't need to be coroutines either. Breaking change!

Author: FichteFoll

asyncio_pool/base_pool.py

@@ -182,7 +182,7 @@ async def spawn(self, coro, cb=None, ctx=None):
  self._waiting[future] = self.loop.create_future() # as a placeholder
  return await self._spawn(future, coro, cb=cb, ctx=ctx)
 
- async def spawn_n(self, coro, cb=None, ctx=None):
+ def spawn_n(self, coro, cb=None, ctx=None):
  '''Creates waiting task for given `coro` regardless of pool space. If
  pool is not full, this task will be executed very soon. Main difference
  is that `spawn_n` does not block and returns future very quickly.
@@ -203,15 +203,15 @@ async def exec(self, coro, cb=None, ctx=None):
  '''
  return await (await self.spawn(coro, cb, ctx))
 
- async def map_n(self, fn, iterable, cb=None, ctx=None):
+ def map_n(self, fn, iterable, cb=None, ctx=None):
  '''Creates coroutine with `fn` function for each item in `iterable`,
  spawns each of them with `spawn_n`, returning futures.
 
  Read more about callbacks in `spawn` docstring.
  '''
  futures = []
  for it in iterable:
- fut = await self.spawn_n(fn(it), cb, ctx)
+ fut = self.spawn_n(fn(it), cb, ctx)
  futures.append(fut)
  return futures
 

tests/test_base.py

@@ -26,7 +26,7 @@ async def wrk(n):
 
  pool_size = 5
  async with AioPool(size=pool_size) as pool:
- futures = await pool.map_n(wrk, todo)
+ futures = pool.map_n(wrk, todo)
 
  await aio.sleep(0.01)
 
@@ -70,7 +70,7 @@ async def outer(n, pool):
  loop = aio.get_event_loop()
  pool = AioPool(size=100)
 
- tasks = await pool.map_n(inner, todo)
+ tasks = pool.map_n(inner, todo)
  joined = [loop.create_task(outer(j, pool)) for j in to_release]
  await pool.join()
 
@@ -88,9 +88,9 @@ async def wrk(*arg, **kw):
 
  pool = AioPool(size=2)
 
- f_quick = await pool.spawn_n(aio.sleep(0.15))
- f12 = await pool.spawn(wrk()), await pool.spawn_n(wrk())
- f35 = await pool.map_n(wrk, range(3))
+ f_quick = pool.spawn_n(aio.sleep(0.15))
+ f12 = await pool.spawn(wrk()), pool.spawn_n(wrk())
+ f35 = pool.map_n(wrk, range(3))
 
  # cancel some
  await aio.sleep(0.1)