[infinispan-commits] Infinispan SVN: r995 - trunk/core/src/test/java/org/infinispan/profiling/testinternals.

[infinispan-commits at lists.jboss.org]

Author: manik.surtani at jboss.com
Date: 2009-10-23 12:42:35 -0400 (Fri, 23 Oct 2009)
New Revision: 995

Modified:
   trunk/core/src/test/java/org/infinispan/profiling/testinternals/TaskRunner.java
Log:
Better thread naming

Modified: trunk/core/src/test/java/org/infinispan/profiling/testinternals/TaskRunner.java
===================================================================
--- trunk/core/src/test/java/org/infinispan/profiling/testinternals/TaskRunner.java	2009-10-23 16:42:04 UTC (rev 994)
+++ trunk/core/src/test/java/org/infinispan/profiling/testinternals/TaskRunner.java	2009-10-23 16:42:35 UTC (rev 995)
@@ -2,7 +2,9 @@
 
 import java.util.concurrent.ExecutorService;
 import java.util.concurrent.Executors;
+import java.util.concurrent.ThreadFactory;
 import java.util.concurrent.TimeUnit;
+import java.util.concurrent.atomic.AtomicInteger;
 
 /**
  * Essentially a delegate to an ExecutorService, but a special one that is only used by perf tests so it can be ignored
@@ -12,9 +14,19 @@
    ExecutorService exec;
 
    public TaskRunner(int numThreads) {
-      this.exec = Executors.newFixedThreadPool(numThreads);
+      this(numThreads, false);
    }
+   public TaskRunner(int numThreads, final boolean warmup) {
+      final AtomicInteger counter = new AtomicInteger(0);
+      final ThreadGroup tg = new ThreadGroup(Thread.currentThread().getThreadGroup(), warmup ? "WarmupLoadGenerators" : "LoadGenerators");
+      this.exec = Executors.newFixedThreadPool(numThreads, new ThreadFactory() {
 
+         public Thread newThread(Runnable r) {
+            return new Thread(tg, r, (warmup ? "WarmupLoadGenerator-" : "LoadGenerator-") + counter.incrementAndGet());
+         }
+      });
+   }
+
    public void execute(Runnable r) {
       exec.execute(r);
    }