package com.github.ltsopensource.admin.web.api;
import com.github.ltsopensource.admin.cluster.BackendAppContext;
import com.github.ltsopensource.admin.request.JobQueueReq;
import com.github.ltsopensource.admin.response.PaginationRsp;
import com.github.ltsopensource.admin.support.AppConfigurer;
import com.github.ltsopensource.admin.support.I18nManager;
import com.github.ltsopensource.admin.web.AbstractMVC;
import com.github.ltsopensource.admin.web.support.Builder;
import com.github.ltsopensource.admin.web.vo.RestfulResponse;
import com.github.ltsopensource.biz.logger.domain.JobLogPo;
import com.github.ltsopensource.biz.logger.domain.JobLoggerRequest;
import com.github.ltsopensource.cmd.DefaultHttpCmd;
import com.github.ltsopensource.cmd.HttpCmd;
import com.github.ltsopensource.cmd.HttpCmdClient;
import com.github.ltsopensource.cmd.HttpCmdResponse;
import com.github.ltsopensource.core.cluster.Node;
import com.github.ltsopensource.core.cluster.NodeType;
import com.github.ltsopensource.core.cmd.HttpCmdNames;
import com.github.ltsopensource.core.commons.utils.Assert;
import com.github.ltsopensource.core.commons.utils.CollectionUtils;
import com.github.ltsopensource.core.commons.utils.StringUtils;
import com.github.ltsopensource.core.domain.Job;
import com.github.ltsopensource.core.domain.Pair;
import com.github.ltsopensource.core.json.JSON;
import com.github.ltsopensource.core.support.CronExpression;
import com.github.ltsopensource.queue.domain.JobPo;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import java.text.ParseException;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.Map;
/**
* @author Robert HG (254963746@qq.com) on 6/6/15.
*/
@RestController
public class JobQueueApi extends AbstractMVC {
@Autowired
private BackendAppContext appContext;
@RequestMapping("/job-queue/executable-job-get")
public RestfulResponse executableJobGet(JobQueueReq request) {
PaginationRsp<JobPo> paginationRsp = appContext.getExecutableJobQueue().pageSelect(request);
boolean needClear = Boolean.valueOf(AppConfigurer.getProperty("lts.admin.remove.running.job.on.executable.search", "false"));
if (needClear) {
paginationRsp = clearRunningJob(paginationRsp);
}
RestfulResponse response = new RestfulResponse();
response.setSuccess(true);
response.setResults(paginationRsp.getResults());
response.setRows(paginationRsp.getRows());
return response;
}
/**
* 比较恶心的逻辑,当等待执行队列的任务同时也在执行中队列, 则不展示
*/
private PaginationRsp<JobPo> clearRunningJob(PaginationRsp<JobPo> paginationRsp) {
if (paginationRsp == null || paginationRsp.getResults() == 0) {
return paginationRsp;
}
PaginationRsp<JobPo> rsp = new PaginationRsp<JobPo>();
List<JobPo> rows = new ArrayList<JobPo>();
for (JobPo jobPo : paginationRsp.getRows()) {
if (appContext.getExecutingJobQueue().getJob(jobPo.getTaskTrackerNodeGroup(), jobPo.getTaskId()) == null) {
// 没有正在执行, 则显示在等待执行列表中
rows.add(jobPo);
}
}
rsp.setRows(rows);
rsp.setResults(paginationRsp.getResults() - paginationRsp.getRows().size() - rows.size());
return rsp;
}
@RequestMapping("/job-queue/executing-job-trigger")
public RestfulResponse triggerJobManually(JobQueueReq request) {
try {
Assert.hasLength(request.getJobId(), "jobId不能为空!");
Assert.hasLength(request.getTaskTrackerNodeGroup(), "taskTrackerNodeGroup不能为空!");
} catch (IllegalArgumentException e) {
return Builder.build(false, e.getMessage());
}
HttpCmd httpCmd = new DefaultHttpCmd();
httpCmd.setCommand(HttpCmdNames.HTTP_CMD_TRIGGER_JOB_MANUALLY);
httpCmd.addParam("jobId", request.getJobId());
httpCmd.addParam("nodeGroup", request.getTaskTrackerNodeGroup());
List<Node> jobTrackerNodeList = appContext.getNodeMemCacheAccess().getNodeByNodeType(NodeType.JOB_TRACKER);
if (CollectionUtils.isEmpty(jobTrackerNodeList)) {
return Builder.build(false, I18nManager.getMessage("job.tracker.not.found"));
}
HttpCmdResponse response = null;
for (Node node : jobTrackerNodeList) {
httpCmd.setNodeIdentity(node.getIdentity());
response = HttpCmdClient.doGet(node.getIp(), node.getHttpCmdPort(), httpCmd);
if (response.isSuccess()) {
return Builder.build(true);
}
}
if (response != null) {
return Builder.build(false, response.getMsg());
} else {
return Builder.build(false, "TriggerFailed failed");
}
}
@RequestMapping("/job-queue/executing-job-get")
public RestfulResponse executingJobGet(JobQueueReq request) {
PaginationRsp<JobPo> paginationRsp = appContext.getExecutingJobQueue().pageSelect(request);
RestfulResponse response = new RestfulResponse();
response.setSuccess(true);
response.setResults(paginationRsp.getResults());
response.setRows(paginationRsp.getRows());
return response;
}
@RequestMapping("/job-queue/executable-job-update")
public RestfulResponse executableJobUpdate(JobQueueReq request) {
// 检查参数
// 1. 检测 cronExpression是否是正确的
if (StringUtils.isNotEmpty(request.getCronExpression())) {
try {
CronExpression expression = new CronExpression(request.getCronExpression());
if (expression.getTimeAfter(new Date()) == null) {
return Builder.build(false, StringUtils.format("该CronExpression={} 已经没有执行时间点!", request.getCronExpression()));
}
} catch (ParseException e) {
return Builder.build(false, "请输入正确的 CronExpression!");
}
}
try {
Assert.hasLength(request.getJobId(), "jobId不能为空!");
Assert.hasLength(request.getTaskTrackerNodeGroup(), "taskTrackerNodeGroup不能为空!");
} catch (IllegalArgumentException e) {
return Builder.build(false, e.getMessage());
}
boolean success = appContext.getExecutableJobQueue().selectiveUpdateByJobId(request);
RestfulResponse response = new RestfulResponse();
if (success) {
response.setSuccess(true);
} else {
response.setSuccess(false);
response.setCode("DELETE_OR_RUNNING");
}
return response;
}
@RequestMapping("/job-queue/executable-job-delete")
public RestfulResponse executableJobDelete(JobQueueReq request) {
try {
Assert.hasLength(request.getJobId(), "jobId不能为空!");
Assert.hasLength(request.getTaskTrackerNodeGroup(), "taskTrackerNodeGroup不能为空!");
} catch (IllegalArgumentException e) {
return Builder.build(false, e.getMessage());
}
boolean success = appContext.getExecutableJobQueue().remove(request.getTaskTrackerNodeGroup(), request.getJobId());
if (success) {
if (StringUtils.isNotEmpty(request.getCronExpression())) {
// 是Cron任务, Cron任务队列的也要被删除
try {
appContext.getCronJobQueue().remove(request.getJobId());
} catch (Exception e) {
return Builder.build(false, "在Cron任务队列中删除该任务失败,请手动更新! error:" + e.getMessage());
}
}
return Builder.build(true);
} else {
return Builder.build(false, "更新失败,该条任务可能已经删除.");
}
}
@RequestMapping("/job-logger/job-logger-get")
public RestfulResponse jobLoggerGet(JobLoggerRequest request) {
RestfulResponse response = new RestfulResponse();
PaginationRsp<JobLogPo> paginationRsp = appContext.getJobLogger().search(request);
response.setResults(paginationRsp.getResults());
response.setRows(paginationRsp.getRows());
response.setSuccess(true);
return response;
}
/**
* 给JobTracker发消息 加载任务到内存
*/
@RequestMapping("/job-queue/load-add")
public RestfulResponse loadJob(JobQueueReq request) {
RestfulResponse response = new RestfulResponse();
String nodeGroup = request.getTaskTrackerNodeGroup();
HttpCmd httpCmd = new DefaultHttpCmd();
httpCmd.setCommand(HttpCmdNames.HTTP_CMD_LOAD_JOB);
httpCmd.addParam("nodeGroup", nodeGroup);
List<Node> jobTrackerNodeList = appContext.getNodeMemCacheAccess().getNodeByNodeType(NodeType.JOB_TRACKER);
if (CollectionUtils.isEmpty(jobTrackerNodeList)) {
response.setMsg(I18nManager.getMessage("job.tracker.not.found"));
response.setSuccess(false);
return response;
}
boolean success = false;
HttpCmdResponse cmdResponse = null;
for (Node node : jobTrackerNodeList) {
// 所有的JobTracker都load一遍
httpCmd.setNodeIdentity(node.getIdentity());
cmdResponse = HttpCmdClient.doGet(node.getIp(), node.getHttpCmdPort(), httpCmd);
if (cmdResponse.isSuccess()) {
success = true;
}
}
if (success) {
response.setMsg("Load success");
} else {
response.setMsg("Load failed");
}
response.setSuccess(success);
return response;
}
@RequestMapping("/job-queue/job-add")
public RestfulResponse jobAdd(String jobType, JobQueueReq request) {
// 表单check
try {
Assert.hasLength(request.getTaskId(), I18nManager.getMessage("taskId.not.null"));
Assert.hasLength(request.getTaskTrackerNodeGroup(), "taskTrackerNodeGroup不能为空!");
if (request.getNeedFeedback()) {
Assert.hasLength(request.getSubmitNodeGroup(), "submitNodeGroup不能为空!");
}
if (StringUtils.isNotEmpty(request.getCronExpression())) {
try {
CronExpression expression = new CronExpression(request.getCronExpression());
Date nextTime = expression.getTimeAfter(new Date());
if (nextTime == null) {
return Builder.build(false, StringUtils.format("该CronExpression={} 已经没有执行时间点!", request.getCronExpression()));
} else {
request.setTriggerTime(nextTime);
}
} catch (ParseException e) {
return Builder.build(false, "请输入正确的 CronExpression!");
}
}
} catch (IllegalArgumentException e) {
return Builder.build(false, e.getMessage());
}
Pair<Boolean, String> pair = addJob(jobType, request);
return Builder.build(pair.getKey(), pair.getValue());
}
private Pair<Boolean, String> addJob(String jobType, JobQueueReq request) {
Job job = new Job();
job.setTaskId(request.getTaskId());
if (CollectionUtils.isNotEmpty(request.getExtParams())) {
for (Map.Entry<String, String> entry : request.getExtParams().entrySet()) {
job.setParam(entry.getKey(), entry.getValue());
}
}
// 执行节点的group名称
job.setTaskTrackerNodeGroup(request.getTaskTrackerNodeGroup());
job.setSubmitNodeGroup(request.getSubmitNodeGroup());
job.setNeedFeedback(request.getNeedFeedback());
job.setReplaceOnExist(true);
// 这个是 cron expression 和 quartz 一样,可选
job.setCronExpression(request.getCronExpression());
if (request.getTriggerTime() != null) {
job.setTriggerTime(request.getTriggerTime().getTime());
}
job.setRepeatCount(request.getRepeatCount() == null ? 0 : request.getRepeatCount());
job.setRepeatInterval(request.getRepeatInterval());
job.setPriority(request.getPriority());
job.setMaxRetryTimes(request.getMaxRetryTimes() == null ? 0 : request.getMaxRetryTimes());
job.setRelyOnPrevCycle(request.getRelyOnPrevCycle() == null ? true : request.getRelyOnPrevCycle());
if ("REAL_TIME_JOB".equals(jobType)) {
job.setCronExpression(null);
job.setTriggerTime(null);
job.setRepeatInterval(null);
job.setRepeatCount(0);
job.setRelyOnPrevCycle(true);
} else if ("TRIGGER_TIME_JOB".equals(jobType)) {
job.setCronExpression(null);
job.setRepeatInterval(null);
job.setRepeatCount(0);
job.setRelyOnPrevCycle(true);
} else if ("CRON_JOB".equals(jobType)) {
job.setRepeatInterval(null);
job.setRepeatCount(0);
} else if ("REPEAT_JOB".equals(jobType)) {
job.setCronExpression(null);
}
return addJob(job);
}
private Pair<Boolean, String> addJob(Job job) {
HttpCmd httpCmd = new DefaultHttpCmd();
httpCmd.setCommand(HttpCmdNames.HTTP_CMD_ADD_JOB);
httpCmd.addParam("job", JSON.toJSONString(job));
List<Node> jobTrackerNodeList = appContext.getNodeMemCacheAccess().getNodeByNodeType(NodeType.JOB_TRACKER);
if (CollectionUtils.isEmpty(jobTrackerNodeList)) {
return new Pair<Boolean, String>(false, I18nManager.getMessage("job.tracker.not.found"));
}
HttpCmdResponse response = null;
for (Node node : jobTrackerNodeList) {
httpCmd.setNodeIdentity(node.getIdentity());
response = HttpCmdClient.doGet(node.getIp(), node.getHttpCmdPort(), httpCmd);
if (response.isSuccess()) {
return new Pair<Boolean, String>(true, "Add success");
}
}
if (response != null) {
return new Pair<Boolean, String>(false, response.getMsg());
} else {
return new Pair<Boolean, String>(false, "Add failed");
}
}
@RequestMapping("/job-queue/executing-job-terminate")
public RestfulResponse jobTerminate(String jobId) {
JobPo jobPo = appContext.getExecutingJobQueue().getJob(jobId);
if (jobPo == null) {
return Builder.build(false, "该任务已经执行完成或者被删除");
}
String taskTrackerIdentity = jobPo.getTaskTrackerIdentity();
Node node = appContext.getNodeMemCacheAccess().getNodeByIdentity(taskTrackerIdentity);
if (node == null) {
return Builder.build(false, "执行该任务的TaskTracker已经离线");
}
HttpCmd cmd = new DefaultHttpCmd();
cmd.setCommand(HttpCmdNames.HTTP_CMD_JOB_TERMINATE);
cmd.setNodeIdentity(taskTrackerIdentity);
cmd.addParam("jobId", jobId);
HttpCmdResponse response = HttpCmdClient.doPost(node.getIp(), node.getHttpCmdPort(), cmd);
if (response.isSuccess()) {
return Builder.build(true);
} else {
return Builder.build(false, response.getMsg());
}
}
}