Преглед на файлове

added changes to enable grow and shrink pool to work

Dan преди 11 години
родител
ревизия
1220bc7f65
променени са 1 файла, в които са добавени 4 реда и са изтрити 0 реда
  1. 4 0
      celery/worker/control.py

+ 4 - 0
celery/worker/control.py

@@ -283,6 +283,8 @@ def pool_grow(state, n=1, **kwargs):
         state.consumer.controller.autoscaler.force_scale_up(n)
     else:
         state.consumer.pool.grow(n)
+    state.consumer.qos.increment_eventually(n)
+    state.consumer.initial_prefetch_count = n
     return {'ok': 'spawned worker processes'}
 
 
@@ -292,6 +294,8 @@ def pool_shrink(state, n=1, **kwargs):
         state.consumer.controller.autoscaler.force_scale_down(n)
     else:
         state.consumer.pool.shrink(n)
+    state.consumer.qos.decrement_eventually(n)
+    state.consumer.initial_prefetch_count = n
     return {'ok': 'terminated worker processes'}