com.jeesuite.scheduler.JobContext Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jeesuite-scheduler Show documentation
There is a newer version: 1.4.0
/**
 * 
 */
package com.jeesuite.scheduler;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;

import com.jeesuite.common.util.NodeNameHolder;
import com.jeesuite.scheduler.helper.ConsistencyHash;
import com.jeesuite.scheduler.registry.NullJobRegistry;

/**
 * @description 

 * @author vakin
 * @date 2016年8月19日
 */
public class JobContext {

	private static JobContext context = new JobContext();
	
	private Set activeNodes = new HashSet();
	
	private ConsistencyHash hash = new ConsistencyHash();
	
	private Map allJobs = new HashMap<>();
	
	private ConfigPersistHandler configPersistHandler;
	
	private TaskRetryProcessor retryProcessor;
	
	private JobLogPersistHandler jobLogPersistHandler;
	
	private JobRegistry registry;
	
	public void startRetryProcessor(){
		if(retryProcessor == null){
			synchronized (context) {
				if(retryProcessor != null)return;
				retryProcessor = new TaskRetryProcessor(1);
			}
		}
	}

	public static JobContext getContext() {
		return context;
	}

	public String getNodeId() {
		return NodeNameHolder.getNodeId();
	}
	
	public ConfigPersistHandler getConfigPersistHandler() {
		return configPersistHandler;
	}

	public void setConfigPersistHandler(ConfigPersistHandler configPersistHandler) {
		this.configPersistHandler = configPersistHandler;
	}

	public JobLogPersistHandler getJobLogPersistHandler() {
		return jobLogPersistHandler;
	}

	public void setJobLogPersistHandler(JobLogPersistHandler jobLogPersistHandler) {
		this.jobLogPersistHandler = jobLogPersistHandler;
	}

	public JobRegistry getRegistry() {
		if(registry == null){
			registry = new NullJobRegistry();
		}
		return registry;
	}

	public void setRegistry(JobRegistry registry) {
		this.registry = registry;
	}

	public TaskRetryProcessor getRetryProcessor() {
		return retryProcessor;
	}

	public void refreshNodes(List nodes){
		activeNodes.clear();
		activeNodes.addAll(nodes);
		if(activeNodes.isEmpty())return;
		hash.refresh(nodes);
	}
	
	public void addNode(String node){
		activeNodes.add(node);
		hash.refresh(new ArrayList<>(activeNodes));
	}
	
	public void removeNode(String node){
		activeNodes.remove(node);
		if(activeNodes.isEmpty())return;
		hash.refresh(new ArrayList<>(activeNodes));
	}
	
	public boolean matchCurrentNode(Object shardFactor){
		if(activeNodes.size() == 1)return true;
		String expectNodeId = hash.getAssignedRealNode(shardFactor);
		return expectNodeId.equals(getNodeId());
	}
	
	public void addJob(AbstractJob job){
		String key = job.group + ":" + job.jobName;
		allJobs.put(key, job);
	}

	public Map getAllJobs() {
		return allJobs;
	}

	public Set getActiveNodes() {
		return activeNodes;
	}
	
	public void close(){
		if(retryProcessor != null){
			retryProcessor.close();
		}
	}

}