begin work for youtube stop

lhorizon · Sep 11, 2014 · 175b6d0 · 175b6d0
1 parent 625d50a
commit 175b6d0
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/local/proxy.py b/local/proxy.py
@@ -354,7 +354,11 @@ def __fetchlet(self, range_queue, data_queue, range_delay_size):
                             if self._stopped:
                                 response.close()
                                 return
-                            data = response.read(self.bufsize)
+                            data = None
+                            with gevent.Timeout(1, False):
+                                data = response.read(self.bufsize)
+                            if data is None:
+                                logging.warning('response.read(%r) timeout', self.bufsize)
                             if not data:
                                 break
                             data_queue.put((start, data))
@@ -453,7 +457,11 @@ def handle(self, handler, **kwargs):
             handler.end_headers()
             bufsize = 8192
             while True:
-                data = response.read(bufsize)
+                data = None
+                with gevent.Timeout(2, False):
+                    data = response.read(bufsize)
+                if data is None:
+                    logging.warning('response.read(%r) timeout', bufsize)
                 if data:
                     handler.wfile.write(data)
                 if not data: