浏览代码

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'}