org.voovan.tools.threadpool.ThreadPoolTask Maven / Gradle / Ivy

Go to download
package org.voovan.tools.threadpool;

import org.voovan.tools.TEnv;
import org.voovan.tools.TPerformance;
import org.voovan.tools.log.Logger;

import java.util.TimerTask;
import java.util.concurrent.ThreadPoolExecutor;

/**
 * 线程池监控类
 *
 * @author helyho
 *
 * Voovan Framework.
 * WebSite: https://github.com/helyho/Voovan
 * Licence: Apache v2 License
 */
public class ThreadPoolTask extends TimerTask {
	private ThreadPoolExecutor	threadPoolInstance;
	private int					cpuCoreCount;
	private int 				count = 0;

	public ThreadPoolTask(ThreadPoolExecutor threadPoolInstance) {
		cpuCoreCount = Runtime.getRuntime().availableProcessors();
		this.threadPoolInstance = threadPoolInstance;
	}

	@Override
	public void run() {


		try {
			if (threadPoolInstance.isShutdown()) {
				this.cancel();
			}


			if(count*1000 % ThreadPool.getStatusInterval() == 0) {

				String threadPoolInfo = "[PoolInfo]: A:" + threadPoolInstance.getActiveCount() + "/ S:" + threadPoolInstance.getCorePoolSize() + "/ L:"
						+ threadPoolInstance.getLargestPoolSize() + "/ M:" + threadPoolInstance.getMaximumPoolSize() + " \t[TaskCount]: "
						+ threadPoolInstance.getCompletedTaskCount() + "/"
						+ threadPoolInstance.getTaskCount() + " \t[QueueSize]: " + threadPoolInstance.getQueue().size() + " \t[PerCoreLoadAvg]: "
						+ TPerformance.cpuPerCoreLoadAvg();
				if (threadPoolInstance.getActiveCount() != 0) {
					Logger.fremawork("[ShutDown]: " + threadPoolInstance.isShutdown() + " \t" + threadPoolInfo);
				}
			}

			count++;

			int poolSize = threadPoolInstance.getPoolSize();
			// 动态调整线程数,且系统CPU负载值要小于1
			if (threadPoolInstance.getQueue().size() > 0) {

				int newPoolSize = (int) (threadPoolInstance.getCorePoolSize() * 1.20);

				if (newPoolSize > ThreadPool.MAX_POOL_SIZE) {
					newPoolSize = ThreadPool.MAX_POOL_SIZE;
				}

				if (newPoolSize != poolSize) {
					threadPoolInstance.setCorePoolSize(newPoolSize);
					Logger.fremawork("PoolSizeChange: " + poolSize + "->" + threadPoolInstance.getCorePoolSize());
				}
			} else if (threadPoolInstance.getActiveCount() <= threadPoolInstance.getPoolSize() / 2 &&
					threadPoolInstance.getCorePoolSize() > ThreadPool.MIN_POOL_SIZE) {

				int newPoolsize = (int) (threadPoolInstance.getCorePoolSize() * 0.8);

				if (newPoolsize < ThreadPool.MIN_POOL_SIZE) {
					newPoolsize = ThreadPool.MIN_POOL_SIZE;
				}

				if (newPoolsize != poolSize) {
					threadPoolInstance.setCorePoolSize(newPoolsize);
					Logger.fremawork("PoolSizeChange: " + poolSize + "->" + threadPoolInstance.getCorePoolSize());
				}
			}

			//如果主线程结束,则线程池也关闭
			if (TEnv.IS_SHUTDOWN) {
				threadPoolInstance.shutdown();
			}
		} catch (Exception e){
			Logger.error("Threadpooltask error: ", e);
		}
	}
}