Mercurial Repositories > Vulture Checker / file diff

diff -r ae07d4efb120 -r cf5c3ddc1de3 VultureChecker/VultureCheckerService.py
--- a/VultureChecker/VultureCheckerService.py	Sun Apr 09 17:17:53 2017 +0200
+++ b/VultureChecker/VultureCheckerService.py	Thu Jun 22 18:24:40 2017 +0200
@@ -57,7 +57,7 @@
     return __analyze(file, text)
 
 
-def batchVultureCheck(argumentsList, send, fx, cancelled):
+def batchVultureCheck(argumentsList, send, fx, cancelled, maxProcesses=0):
     """
     Module function to analyze a batch of files.
     
@@ -69,13 +69,19 @@
     @type str
     @param cancelled reference to function checking for a cancellation
     @type function
+    @param maxProcesses number of processes to be used
+    @type int
     """
-    try:
-        NumberOfProcesses = multiprocessing.cpu_count()
-        if NumberOfProcesses >= 1:
-            NumberOfProcesses -= 1
-    except NotImplementedError:
-        NumberOfProcesses = 1
+    if maxProcesses == 0:
+        # determine based on CPU count
+        try:
+            NumberOfProcesses = multiprocessing.cpu_count()
+            if NumberOfProcesses >= 1:
+                NumberOfProcesses -= 1
+        except NotImplementedError:
+            NumberOfProcesses = 1
+    else:
+        NumberOfProcesses = maxProcesses
 
     # Create queues
     taskQueue = multiprocessing.Queue()