CollectorTestUtil.java

package jp.terasoluna.fw.collector;

import java.util.concurrent.TimeUnit;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

public class CollectorTestUtil {

    /**
     * Log.
     */
    private static Log logger = LogFactory.getLog(CollectorTestUtil.class);

    /**
     * 全てのコレクタスレッドに割り込みをかける
     */
    public static void allInterrupt() {
        int activeCount = Thread.activeCount();
        Thread[] ta = new Thread[activeCount + 10];
        int threadCount = Thread.enumerate(ta);
        for (int i = 0; i < threadCount; i++) {
            if (ta[i] == null) {
                continue;
            }
            String threadName = ta[i].getName();

            if (threadName != null
                    && threadName
                            .startsWith(CollectorThreadFactory.COLLECTOR_THREAD_NAME_PREFIX)) {
                ta[i].interrupt();
            }
        }
        try {
            TimeUnit.MILLISECONDS.sleep(100);
        } catch (InterruptedException e) {
        }
    }

    /**
     * コレクタスレッド数が指定以下であるかチェックする
     * @param count
     * @return
     */
    public static boolean lessThanCollectorThreadCount(int count) {
        int retry = 0;

        while (count < getCollectorThreadCount()) {
            if (retry++ > 100) {
                if (logger.isWarnEnabled()) {
                    StringBuilder sb = new StringBuilder();
                    sb.append("Thread count error.");
                    sb.append(" expected:[");
                    sb.append(count);
                    sb.append("]");
                    sb.append(" actual:[");
                    sb.append(getCollectorThreadCount());
                    sb.append("]");
                    logger.warn(sb.toString());
                }
                return false;
            }
            try {
                TimeUnit.MILLISECONDS.sleep(100);
            } catch (InterruptedException e) {
            }
        }
        return true;
    }

    /**
     * コレクタスレッドの数を返す
     * @return コレクタスレッドの数
     */
    public static int getCollectorThreadCount() {
        int collectorThreadCount = 0;
        int activeCount = Thread.activeCount();
        Thread[] ta = new Thread[activeCount + 10];
        int threadCount = Thread.enumerate(ta);
        for (int i = 0; i < threadCount; i++) {
            if (ta[i] == null) {
                continue;
            }
            String threadName = ta[i].getName();

            if (threadName != null
                    && threadName
                            .startsWith(CollectorThreadFactory.COLLECTOR_THREAD_NAME_PREFIX)) {
                collectorThreadCount++;
            }
        }
        return collectorThreadCount;
    }
}