Add `aclosing()` around asyn gen loop

eg_worker_poolz
Tyler Goodlet 2021-01-24 13:14:17 -05:00
parent 07653bc02e
commit f7e1c526c5
2 changed files with 5 additions and 3 deletions

View File

@ -16,6 +16,7 @@ import time
import tractor
import trio
from async_generator import aclosing
PRIMES = [
@ -103,10 +104,11 @@ async def main():
async with worker_pool() as actor_map:
start = time.time()
# for number, prime in zip(PRIMES, executor.map(is_prime, PRIMES)):
async for number, prime in actor_map(is_prime, PRIMES):
print(f'{number} is prime: {prime}')
async with aclosing(actor_map(is_prime, PRIMES)) as results:
async for number, prime in results:
print(f'{number} is prime: {prime}')
print(f'processing took {time.time() - start} seconds')