com.alogic.doer.local.LocalTaskQueue Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of alogic-doer Show documentation
后台执行器，定时器框架
There is a newer version: 1.6.17
package com.alogic.doer.local;

import java.util.ArrayList;
import java.util.Hashtable;
import java.util.List;
import java.util.Map;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.TimeUnit;

import org.w3c.dom.Element;

import com.alogic.doer.core.TaskDispatcher;
import com.alogic.doer.core.TaskRobber;
import com.alogic.doer.core.TaskQueue;
import com.alogic.doer.core.TaskReport;
import com.alogic.timer.core.Task;
import com.alogic.timer.core.Task.State;
import com.anysoft.util.BaseException;
import com.anysoft.util.Factory;
import com.anysoft.util.JsonTools;
import com.anysoft.util.Properties;
import com.anysoft.util.PropertiesConstants;
import com.anysoft.util.XmlElementProperties;

/**
 * JVM本地的任务队列
 * 
 * @author duanyy
 * @since 1.6.3.4
 * 
 */
public class LocalTaskQueue implements TaskQueue {
	
	public TaskReport getTaskReport(String id) {
		return reports.get(id);
	}
	
	public void reportState(Task task, State state,
			int percent) {
		TaskReport.Default report = reports.get(task.id());
		if (report != null){
			report.reportState(state, percent);
		}
	}

	public void reportState(String id, State state, int percent) {
		TaskReport.Default report = reports.get(id);
		if (report != null){
			report.reportState(state, percent);
		}
	}	
	
	synchronized public void dispatch(Task task) throws BaseException {
		TaskReport.Default report = new TaskReport.Default(task);
		reports.put(task.id(), report);
		
		if (!queue.offer(task)){
			report.reportState(State.Queued, -1);
			throw new BaseException("core.queue_is_full","The queue is full.queue:" + id());
		}
	} 

	public void configure(Properties p) throws BaseException {
		id = PropertiesConstants.getString(p, "id", "");
		
		doersCnt = PropertiesConstants.getInt(p,"cnt", doersCnt);
		doersCnt = doersCnt <= 0 ? 30:doersCnt;
	}
	
	public void configure(Element _e, Properties _properties)
			throws BaseException {
		Properties p = new XmlElementProperties(_e,_properties);
		
		configure(p);
		
		Factory factory = new Factory();
		
		doers = new ArrayList(doersCnt);
		for (int i = 0 ;i < doersCnt; i ++){
			TaskRobber doer = factory.newInstance(_e, _properties, "module", TaskRobber.Default.class.getName());
			doer.setTaskQueue(this);
			doer.start();
			doers.add(doer);
		}
		
		queue = new LinkedBlockingQueue();
		
		reports = new Hashtable();
	}

	public void report(Element xml) {
		if (xml != null){
			xml.setAttribute("module", getClass().getName());
		}
	}

	public void report(Map json) {
		if (json != null){
			JsonTools.setString(json,"module",getClass().getName());
		}
	}

	public String id() {
		return id;
	}

	public void askForTask(TaskDispatcher doer, long timeout) {
		try {
			Task task = queue.poll(timeout, TimeUnit.MILLISECONDS);
			if (task != null){
				reportState(task.id(),State.Polled,-1);
				doer.dispatch(task);
			}
		}catch (Exception ex){
			
		}
	}

	/**
	 * 队列的ID
	 */
	protected String id;
	
	/**
	 * 任务队列
	 */
	protected LinkedBlockingQueue queue = null;
	
	/**
	 * 任务处理者列表
	 */
	protected List doers = null;
	
	/**
	 * 任务处理者个数
	 */
	protected int doersCnt = 5;

	/**
	 * 报告列表
	 */
	protected Hashtable reports = null;
}