com.jeesuite.common2.task.GlobalInternalScheduleService Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jeesuite-common2 Show documentation
The newest version!
package com.jeesuite.common2.task;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.TimeUnit;

import org.apache.commons.lang3.exception.ExceptionUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.DisposableBean;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.core.Ordered;
import org.springframework.core.annotation.Order;

import com.jeesuite.common.async.AsyncInitializer;
import com.jeesuite.common.async.StandardThreadExecutor.StandardThreadFactory;
import com.jeesuite.spring.InstanceFactory;
import com.jeesuite.spring.helper.SpringAopHelper;

/**
 * 
 * 
 * 

 * Class Name   : GlobalInternalScheduleService
 *
 * @author vakin
 * @version 1.0.0
 * @date Jan 8, 2022
 */
@Order(Ordered.LOWEST_PRECEDENCE)
public class GlobalInternalScheduleService implements InitializingBean, DisposableBean ,AsyncInitializer {

	private static Logger log = LoggerFactory.getLogger("global.internal.task");

	private List taskStats = new ArrayList<>();
	private ScheduledExecutorService executor;
	
	public GlobalInternalScheduleService() {}
	
	public GlobalInternalScheduleService(List tasks) {
		for (SubTimerTask task : tasks) {
			taskStats.add(new SubTimerTaskStat(task));
		}
	}



	@Override
	public void destroy() throws Exception {
		if (executor == null)
			return;
		executor.shutdown();
	}

	@Override
	public void afterPropertiesSet() throws Exception {

		Map taskMap = InstanceFactory.getBeansOfType(SubTimerTask.class);
		if (taskMap != null) {
			for (SubTimerTask task : taskMap.values()) {
				taskStats.add(new SubTimerTaskStat(task));
			}
		}
		
		if(taskStats.isEmpty())return;

		executor = Executors.newScheduledThreadPool(1, new StandardThreadFactory("globalInternalScheduler"));
		executor.scheduleWithFixedDelay(new Runnable() {
			@Override
			public void run() {
				for (SubTimerTaskStat taskStat : taskStats) {
					try {
						Thread.sleep(10);
					} catch (Exception e) {
					}
					execSubTimerTask(taskStat);
				}
			}
		}, 1000, 5000, TimeUnit.MILLISECONDS);

		log.info("init GlobalInternalScheduleService finish -> subtaskNums:{}", taskStats.size());
	}

	protected void execSubTimerTask(SubTimerTaskStat taskStat) {
		if (taskStat.running)
			return;
		long currentTime = System.currentTimeMillis();
		if (currentTime - taskStat.lastFireTime < taskStat.task.periodMillis()) {
			return;
		}
		taskStat.running = true;
		try {
			if (log.isDebugEnabled())
				log.debug("InternalSchedule[{}] execute Begin..", taskStat.taskName);
			taskStat.task.doSchedule();
			taskStat.lastFireTime = currentTime;
		} catch (Exception e) {
			log.error("InternalSchedule[{}] execute Error:{}", taskStat.taskName,ExceptionUtils.getMessage(e));
		} finally {
			taskStat.running = false;
		}
	}

	private class SubTimerTaskStat {
		SubTimerTask task;
		String taskName;
		boolean running = false;
		long lastFireTime;

		public SubTimerTaskStat(SubTimerTask task) {
			this.task = task;
			try {
				this.taskName = SpringAopHelper.getTarget(task).getClass().getSimpleName();
			} catch (Exception e) {
				this.taskName = task.getClass().getSimpleName();
			}
			// 确保启动执行
			this.lastFireTime = System.currentTimeMillis() - 3600 * 24 * 1000;
		}
	}


	@Override
	public void doInitialize() {
		for (SubTimerTaskStat taskStat : taskStats) {
			execSubTimerTask(taskStat);
		}
	}

}