From: Chris Adams <chris@improbable.org>
Date: Sun, 29 Jan 2012 23:42:46 +0000 (-0500)
Subject: async: force prefetch=True on send
X-Git-Tag: v0.10.2~35^2~2
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=495646bb6a89802a60208c2ab09749edba1f8e51;p=services%2Fpython-requests.git

async: force prefetch=True on send

… otherwise we can run out of sockets before any of the responses are ever
processed
---

diff --git a/requests/async.py b/requests/async.py
index 9488447..24f8a61 100644
--- a/requests/async.py
+++ b/requests/async.py
@@ -46,15 +46,15 @@ def patched(f):
     return wrapped
 
 
-def send(r, pool=None):
-    """Sends the request object using the specified pool. If a pool isn't 
+def send(r, pool=None, prefetch=False):
+    """Sends the request object using the specified pool. If a pool isn't
     specified this method blocks. Pools are useful because you can specify size
     and can hence limit concurrency."""
 
     if pool != None:
-        return pool.spawn(r.send)
+        return pool.spawn(r.send, prefetch=prefetch)
 
-    return gevent.spawn(r.send)
+    return gevent.spawn(r.send, prefetch=prefetch)
 
 
 # Patched requests.api functions.
@@ -79,10 +79,7 @@ def map(requests, prefetch=True, size=None):
     requests = list(requests)
 
     pool = Pool(size) if size else None
-    jobs = [send(r, pool) for r in requests]
+    jobs = [send(r, pool, prefetch=prefetch) for r in requests]
     gevent.joinall(jobs)
 
-    if prefetch:
-        [r.response.content for r in requests]
-
     return [r.response for r in requests]
\ No newline at end of file