add threadCount property to ParallelJUnitTask

netmelody · Mar 7, 2012 · 78b1f79 · 78b1f79
1 parent 19f9e00
commit 78b1f79
Showing 1 changed file with 17 additions and 2 deletions.
diff --git a/src/main/java/org/netmelody/panto/task/ParallelJUnitTask.java b/src/main/java/org/netmelody/panto/task/ParallelJUnitTask.java
@@ -7,6 +7,7 @@
 
 public final class ParallelJUnitTask extends Task {
     private RecordingJUnitTask embeddedJUnitTask;
+    private int threadCount = -1;
     private int stripeCount = -1;
 
     public void setStripeCount(int stripeCount) {
@@ -15,6 +16,13 @@ public void setStripeCount(int stripeCount) {
         }
         this.stripeCount = stripeCount;
     }
+
+    public void setThreadCount(int threadCount) {
+        if (threadCount <= 0) {
+            throw new BuildException("There must be at least one thread");
+        }
+        this.threadCount = threadCount;
+    }
 
     public RecordingJUnitTask createJUnit() {
         try {
@@ -30,10 +38,17 @@ public RecordingJUnitTask createJUnit() {
     public void execute() throws BuildException {
         final Parallel parallelTask = (Parallel) getProject().createTask("parallel");
         parallelTask.init();
-        parallelTask.setThreadsPerProcessor(1);
+
+        if (threadCount > 0) {
+            parallelTask.setThreadCount(threadCount);
+        }
+        else {
+            threadCount = Runtime.getRuntime().availableProcessors();
+            parallelTask.setThreadsPerProcessor(1);
+        }
 
         if (stripeCount <= 0) {
-            stripeCount = Runtime.getRuntime().availableProcessors();
+            stripeCount = threadCount;
         }
         for (int i = 0; i < stripeCount; i++) {
             parallelTask.addTask(embeddedJUnitTask.spawnTaskForStripe(i + 1, stripeCount));