长任务实现

实现代码

长任务

package org.infinite.ark.platform.task;

import java.util.ArrayList;

import org.infinite.ark.framework.collection.Mapx;
import org.infinite.ark.framework.util.StringFormat;
import org.infinite.ark.framework.util.StringUtil;

/**
 * 
 * @author Darkness
 * @date 2014-12-26 上午9:37:33
 * @version V1.0
 */
public abstract class LongTimeTask extends Thread {
	
	private static Mapx<Long, LongTimeTask> map = new Mapx<Long, LongTimeTask>();

	private static long IDBase = System.currentTimeMillis();
	private static final int MaxListSize = 1000;
	
	private long id;
	private ArrayList<String> list = new ArrayList<String>();
	protected int percent;
	protected String currentInfo;
	private String finishedInfo;
	protected ArrayList<String> errors = new ArrayList<String>();
	private boolean stopFlag;
	private String type;
	private long stopTime = System.currentTimeMillis() + (24*60*1000L);

	public static LongTimeTask createEmptyInstance() {
		return new LongTimeTask(false) {
			public void execute() {
			}
		};
	}

	public static LongTimeTask getInstanceById(long id) {
		return map.get(id);
	}

	public static void removeInstanceById(long id) {
		synchronized (LongTimeTask.class) {
			map.remove(id);
		}
	}

	public static String cancelByType(String type) {
		String message = "该任务不存在:" + type;
		LongTimeTask ltt = getInstanceByType(type);
		if (ltt != null) {
			ltt.stopTask();
			message = "任务终止中,请稍等片刻！";
		}
		return message;
	}

	public static LongTimeTask getInstanceByType(String type) {
		if (StringUtil.isNotEmpty(type)) {
			long current = System.currentTimeMillis();
			for (Long key : map.keyArray()) {
				LongTimeTask ltt = map.get(key);
				if (type.equals(ltt.getType())) {
					if (current - ltt.stopTime > 60000L) {
						map.remove(key);
						return null;
					}
					return ltt;
				}
			}
		}
		return null;
	}

	public LongTimeTask() {
		this(true);
	}

	private LongTimeTask(boolean flag) {
		if (flag) {
			setName("LongTimeTask Thread");
			synchronized (LongTimeTask.class) {
				this.id = (IDBase++);
				map.put(this.id, this);
				clearStopedTask();
			}
		}
	}

	private void clearStopedTask() {
		synchronized (LongTimeTask.class) {
			long current = System.currentTimeMillis();
			for (Long k : map.keyArray()) {
				LongTimeTask ltt = map.get(k);
				if (current - ltt.stopTime > 60000L)
					map.remove(k);
			}
		}
	}

	public long getTaskID() {
		return this.id;
	}

	public void info(String message) {

		this.list.add(message);
		if (this.list.size() > MaxListSize)
			this.list.remove(0);
	}

	public String[] getMessages() {
		String[] arr = new String[this.list.size()];
		for (int i = 0; i < arr.length; i++) {
			arr[i] = ((String) this.list.get(i));
		}
		this.list.clear();
		return arr;
	}

	public void run() {
		if (StringUtil.isNotEmpty(this.type)) {
			LongTimeTask ltt = getInstanceByType(this.type);
			if ((ltt != null) && (ltt != this))
				return;
		}
		try {
			execute();
		} catch (Exception ie) {
			interrupt();
			addError(ie.getMessage());
		} finally {
			this.stopTime = System.currentTimeMillis();
		}
	}

	public abstract void execute();

	public boolean checkStop() {
		return this.stopFlag;
	}

	public void stopTask() {
		clearStopedTask();
		this.stopFlag = true;
	}

	public int getPercent() {
		return this.percent;
	}

	public void setPercent(int percent) {
		this.percent = percent;
	}

	public void setCurrentInfo(String currentInfo) {
		this.currentInfo = currentInfo;
	}

	public String getCurrentInfo() {
		return this.currentInfo;
	}

	public void setFinishedInfo(String finishedInfo) {
		this.finishedInfo = finishedInfo;
	}

	public String getFinishedInfo() {
		return this.finishedInfo;
	}

	public void addError(String error) {
		this.errors.add(error);
	}

	public void addError(String[] errorArr) {
		for (int i = 0; i < errorArr.length; i++)
			this.errors.add(errorArr[i]);
	}

	public String getAllErrors() {
		if (this.errors.size() == 0) {
			return null;
		}
		StringBuilder sb = new StringBuilder();
		StringFormat sf = new StringFormat("总计 ? 个错误", this.errors.size());
		sb.append(sf.toString() + ":<br>");
		for (int i = 0; i < this.errors.size(); i++) {
			sb.append(i + 1);
			sb.append(": ");
			sb.append((String) this.errors.get(i));
			sb.append("<br>");
		}
		return sb.toString();
	}

	public String getType() {
		return this.type;
	}

	public void setType(String type) {
		this.type = type;
	}
}

长任务控制

package org.infinite.ark.platform.task;

import java.util.HashMap;
import java.util.Map;

import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody;

import org.infinite.ark.framework.util.StringUtil;
import org.infinite.ark.platform.core.web.ajax.ActionResult;

/**   
 * 
 * @author Darkness
 * @date 2014-12-26 上午9:55:56 
 * @version V1.0   
 */
@Controller
@RequestMapping("/ark/plat/longTimeTask")
public class LongTimeTaskController {

	/**
	 * 获取长任务信息
	 * 
	 * @author Darkness
	 * @date 2014-12-26 上午10:17:52 
	 * @version V1.0
	 */
	@ResponseBody
	@RequestMapping("getInfo")
	public ActionResult getInfo(Long taskId) {
		if(taskId == null) {
			return new ActionResult("请指定任务Id");
		}
		
		Map<String, Object> data = new HashMap<String, Object>();
		
		LongTimeTask ltt = LongTimeTask.getInstanceById(taskId);
		if ((ltt != null) && (ltt.isAlive())) {
			data.put("currentInfo", StringUtil.isNotEmpty(ltt.getCurrentInfo()) ? ltt.getCurrentInfo() + "..." : "");
			data.put("messages", ltt.getMessages());
			data.put("percent", ltt.getPercent());
		} else {
			data.put("isCompleted", true);
			String finishInfo = "任务己完成";
			if (ltt != null) {
				String errors = ltt.getAllErrors();
				if (StringUtil.isNotEmpty(errors)) {
					data.put("currentInfo", errors);
					data.put("hasError", true);
				} else {
					finishInfo = StringUtil.isNotEmpty(ltt.getFinishedInfo()) ? ltt.getFinishedInfo() : finishInfo;
					data.put("currentInfo", finishInfo);
				}
			} else {
				data.put("currentInfo", finishInfo);
			}
			LongTimeTask.removeInstanceById(taskId);
		}
		
		return new ActionResult("任务信息", data);
	}

	@ResponseBody
	@RequestMapping("stop")
	public ActionResult stop(Long taskId) {
		if(taskId == null) {
			return new ActionResult("请指定任务Id");
		}
		
		LongTimeTask ltt = LongTimeTask.getInstanceById(taskId);
		if (ltt != null) {
			ltt.stopTask();
		}
		
		return new ActionResult("任务停止成功");
	}

	@ResponseBody
	@RequestMapping("stopComplete")
	public ActionResult stopComplete(Long taskId) {
		if(taskId == null) {
			return new ActionResult("请指定任务Id");
		}
		
		LongTimeTask ltt = LongTimeTask.getInstanceById(taskId);
		if ((ltt == null) || (!ltt.isAlive())) {
			LongTimeTask.removeInstanceById(taskId);
			return new ActionResult("任务停止成功");
		} else {
			return new ActionResult("任务停止失败");
		}
	}
	
}

长任务示例

后端调用示例

package org.infinite.ark.platform.task;

import java.util.HashMap;
import java.util.Map;

import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody;

import org.infinite.ark.platform.core.ServiceException;
import org.infinite.ark.platform.core.web.ajax.ActionResult;

/**   
 * 
 * @author Darkness
 * @date 2014-12-26 上午10:23:44 
 * @version V1.0   
 */
public class LongTimeTaskDemoController {

	/**
	 * 示例方法
	 * 
	 * @author Darkness
	 * @date 2014-12-26 上午10:08:19 
	 * @version V1.0
	 */
	@ResponseBody
	@RequestMapping("export")
	public ActionResult exportDatabase() {
		LongTimeTask ltt = LongTimeTask.getInstanceByType("exportDatabase");
		if (ltt != null) {
			throw new ServiceException("数据库正在导出中...");
		}
		ltt = new LongTimeTask() {
			public void execute() {
				DBExporter di = new DBExporter();
				di.setTask(this);
				di.importDB("export file path");
				setPercent(100);
			}
		};
		ltt.setType("exportDatabase");
		ltt.start();
		
		Map<String, Object> data = new HashMap<String, Object>();
		data.put("taskId", ltt.getTaskID());
		
		return new ActionResult("导出数据库", data);
	}
	
}


class DBExporter {
	
	private LongTimeTask task;
	
	public void setTask(LongTimeTask task) {
		this.task = task;
	}

	public void importDB(String filePath) {
		String[] tables = new String[]{};
		for (int i=0;i<tables.length; i++) {
			this.task.setPercent(new Double(i * 100.0D /tables.length).intValue());
			this.task.setCurrentInfo("Importing table " + tables[i]);
		}
	}
	
}

前端调用示例

Server.sendRequest("/Database/export.do",{},function(response){ 
	var taskId = response.get("taskId"); 
	var p = new Progress(taskId,"数据导出中...",500,150); 
	p.show();  
}); 

/**
 * 进度条示例
 */
function Progress(taskId, msg, width, height) {
	this.taskId = taskId;
	this.refreshTaskInfo();
}

Progress.prototype = {
	refreshTaskInfo: function() {
		var me = this;
		var runner = new TaskRunner();
		runner.start({
			run: function(task) {
				Server.sendRequest(
					"/ark/plat/longTimeTask/getInfo.do", 
					{"taskId": me.taskId}, 
					function(response, data) {
						if (response.Status == 1) {
							if(data.isCompleted) {// 长任务执行完毕
								task.stop();// 停止前端同步任务，不再执行
								
								if(data.hasError) {// 有错误
									alert(data.currentInfo);
								} else {// 没错误
									alert(data.currentInfo);
								}
							} else {
								// 设置任务不在运行中，下次才会继续定时执行
								task.running = false;
								
								me.updateProgressInfo(data.percent, data.currentInfo);
							}
						} else {
							alert(response.Message);
						}
					}
				);
			},
			interval: 500 
		});
	},
	updateProgressInfo: function(percent, progressInfo){
		// 更新进度条的进度+文本
	}
	show: function() {
		// 显示进度条
	}
}

参考

一个简单的阻塞式AjaxTaskRunner实现