tiny map() refactoring
authorPiotr Dobrogost <pd@dev.dobrogost.net>
Fri, 20 Jan 2012 20:42:25 +0000 (21:42 +0100)
committerPiotr Dobrogost <pd@dev.dobrogost.net>
Fri, 20 Jan 2012 20:42:25 +0000 (21:42 +0100)
requests/async.py

index fafcb9ed5b32979c8e66a77f30eb374934112e6b..94884478bec7725866d62e0af020c591ea171156 100644 (file)
@@ -78,12 +78,8 @@ def map(requests, prefetch=True, size=None):
 
     requests = list(requests)
 
-    if size:
-        pool = Pool(size)
-        jobs = [pool.spawn(r.send) for r in requests]
-    else:
-        jobs = [gevent.spawn(r.send) for r in requests]
-    
+    pool = Pool(size) if size else None
+    jobs = [send(r, pool) for r in requests]
     gevent.joinall(jobs)
 
     if prefetch: