Add WORKER_POOL environmental option to facilitate pysawsim.manager benchmarking.

[sawsim.git] / pysawsim / manager / subproc.py

diff --git a/pysawsim/manager/subproc.py b/pysawsim/manager/subproc.py
index d767da4011cac35dc66c544a78eff99bac7e6775..6a17078ab4cb2a9978eec68b7116da7ca99441a2 100644 (file)
--- a/pysawsim/manager/subproc.py
+++ b/pysawsim/manager/subproc.py
@@ -30,6 +30,7 @@ except ImportError, _DISABLING_ERROR:
     Process = object
     _SKIP = '  # doctest: +SKIP'
 
+import os
 
 from .. import log
 from . import Job
@@ -116,8 +117,8 @@ class SubprocessManager (ThreadManager):
         self._receive_queue = Queue()
 
     def _spawn_workers(self, worker_pool=None):
-        if worker_pool == None:
-            worker_pool = cpu_count() + 1
+        if worker_pool is None:
+            worker_pool = int(os.environ.get('WORKER_POOL', cpu_count() + 1))
         self._manager = Manager()
         self._workers = []
         for i in range(worker_pool):