Adds grow/shrink methods to eventlet TaskPool

mher · mher · commit 23e99aad1526 · 2014-05-29T16:39:13.000+04:00
diff --git a/celery/concurrency/eventlet.py b/celery/concurrency/eventlet.py
@@ -143,6 +143,16 @@ def on_apply(self, target, args=None, kwargs=None, callback=None,
                         callback, accept_callback,
                         self.getpid)
 
+    def grow(self, n=1):
+        limit = self.limit + n
+        self._pool.resize(limit)
+        self.limit = limit
+
+    def shrink(self, n=1):
+        limit = self.limit - n
+        self._pool.resize(limit)
+        self.limit = limit
+
     def _get_info(self):
         return {
             'max-concurrency': self.limit,
