Selaa lähdekoodia

Adds grow/shrink methods to eventlet TaskPool

Mher Movsisyan 11 vuotta sitten
vanhempi
commit
23e99aad15
1 muutettua tiedostoa jossa 10 lisäystä ja 0 poistoa
  1. 10 0
      celery/concurrency/eventlet.py

+ 10 - 0
celery/concurrency/eventlet.py

@@ -143,6 +143,16 @@ class TaskPool(base.BasePool):
                         callback, accept_callback,
                         callback, accept_callback,
                         self.getpid)
                         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):
     def _get_info(self):
         return {
         return {
             'max-concurrency': self.limit,
             'max-concurrency': self.limit,