Add `aclosing()` around asyn gen loop
parent
5db737d368
commit
2ed071c903
|
@ -16,6 +16,7 @@ import time
|
||||||
|
|
||||||
import tractor
|
import tractor
|
||||||
import trio
|
import trio
|
||||||
|
from async_generator import aclosing
|
||||||
|
|
||||||
|
|
||||||
PRIMES = [
|
PRIMES = [
|
||||||
|
@ -103,10 +104,11 @@ async def main():
|
||||||
async with worker_pool() as actor_map:
|
async with worker_pool() as actor_map:
|
||||||
|
|
||||||
start = time.time()
|
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')
|
print(f'processing took {time.time() - start} seconds')
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue