org.testng.internal.thread.TestNGThreadFactory Maven / Gradle / Ivy
package org.testng.internal.thread;
import java.util.Collection;
import java.util.Collections;
import java.util.Set;
import java.util.WeakHashMap;
import java.util.concurrent.ThreadFactory;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.stream.Collectors;
public class TestNGThreadFactory implements ThreadFactory {
private final AtomicInteger threadNumber = new AtomicInteger(1);
private final String name;
private final Set threads = Collections.newSetFromMap(new WeakHashMap<>());
public TestNGThreadFactory(String name) {
this.name = ThreadUtil.THREAD_NAME + "-" + name + "-";
}
public Collection getRunningThreads() {
return threads.stream().filter(Thread::isAlive).collect(Collectors.toList());
}
@Override
public Thread newThread(Runnable r) {
Thread thread = new Thread(r, name + threadNumber.getAndIncrement());
threads.add(thread);
return thread;
}
}