com.clustercontrol.commons.util.MonitoredThreadPoolExecutor.java Source code

Introduction

Here is the source code for com.clustercontrol.commons.util.MonitoredThreadPoolExecutor.java

Source

/*
    
Copyright (C) 2012 NTT DATA Corporation
    
This program is free software; you can redistribute it and/or
Modify it under the terms of the GNU General Public License
as published by the Free Software Foundation, version 2.
    
This program is distributed in the hope that it will be
useful, but WITHOUT ANY WARRANTY; without even the implied
warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
PURPOSE.  See the GNU General Public License for more details.
    
 */

package com.clustercontrol.commons.util;

import java.util.Collections;
import java.util.Map;
import java.util.concurrent.BlockingQueue;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.RejectedExecutionHandler;
import java.util.concurrent.ThreadFactory;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;

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

import com.clustercontrol.commons.bean.ThreadInfo;
import com.clustercontrol.util.HinemosTime;

/**
 * ??????<br/>
 */
public class MonitoredThreadPoolExecutor extends ThreadPoolExecutor {

    private static final Log log = LogFactory.getLog(MonitoredThreadPoolExecutor.class);

    private static Map<Long, ThreadInfo> runningTaskMap = new ConcurrentHashMap<Long, ThreadInfo>();

    public MonitoredThreadPoolExecutor(int corePoolSize, int maximumPoolSize, long keepAliveTime, TimeUnit unit,
            BlockingQueue<Runnable> workQueue) {
        super(corePoolSize, maximumPoolSize, keepAliveTime, unit, workQueue);
    }

    public MonitoredThreadPoolExecutor(int corePoolSize, int maximumPoolSize, long keepAliveTime, TimeUnit unit,
            BlockingQueue<Runnable> workQueue, ThreadFactory threadFactory) {
        super(corePoolSize, maximumPoolSize, keepAliveTime, unit, workQueue, threadFactory);
    }

    public MonitoredThreadPoolExecutor(int corePoolSize, int maximumPoolSize, long keepAliveTime, TimeUnit unit,
            BlockingQueue<Runnable> workQueue, RejectedExecutionHandler handler) {
        super(corePoolSize, maximumPoolSize, keepAliveTime, unit, workQueue, handler);
    }

    public MonitoredThreadPoolExecutor(int corePoolSize, int maximumPoolSize, long keepAliveTime, TimeUnit unit,
            BlockingQueue<Runnable> workQueue, ThreadFactory threadFactory, RejectedExecutionHandler handler) {
        super(corePoolSize, maximumPoolSize, keepAliveTime, unit, workQueue, threadFactory, handler);
    }

    @Override
    protected void beforeExecute(Thread t, Runnable r) {
        try {
            beginTask(t, r.getClass().getName());
        } finally {
            super.beforeExecute(t, r);
        }
    }

    @Override
    protected void afterExecute(Runnable r, Throwable t) {
        try {
            finishTask(Thread.currentThread());
        } finally {
            super.afterExecute(r, t);
        }
    }

    public static void beginTask(Thread t, String className) {
        // ThreadLocal??
        HinemosSessionContext.instance().setProperty(JpaTransactionManager.EM, null);

        // ??
        ThreadInfo threadInfo = new ThreadInfo(t, className, HinemosTime.currentTimeMillis());
        runningTaskMap.put(t.getId(), threadInfo);

        if (log.isDebugEnabled()) {
            log.debug("starting new monitored task : " + threadInfo);
        }
    }

    public static void finishTask(Thread t) {
        // ???
        ThreadInfo threadInfo = runningTaskMap.remove(t.getId());

        if (log.isDebugEnabled()) {
            log.debug("finishing new monitored task : " + threadInfo);
        }
    }

    public static Map<Long, ThreadInfo> getRunningThreadMap() {
        return Collections.unmodifiableMap(runningTaskMap);
    }

}