private void cancelLongJob(String uuid, Completion completion) { LongJobVO vo = updateByUuid(uuid, it -> it.setState(LongJobState.Canceling)); LongJob job = longJobFactory.getLongJob(vo.getJobName()); logger.info(String.format("longjob [uuid:%s, name:%s] has been marked canceling", vo.getUuid(), vo.getName())); job.cancel(vo, new Completion(completion) { @Override public void success() { updateByUuid(uuid, it -> it.setState(LongJobState.Canceled)); logger.info(String.format("longjob [uuid:%s, name:%s] has been canceled", vo.getUuid(), vo.getName())); completion.success(); } @Override public void fail(ErrorCode errorCode) { updateByUuid(uuid, it -> it.setState(LongJobState.Failed)); logger.error(String.format("failed to cancel longjob [uuid:%s, name:%s]", vo.getUuid(), vo.getName())); completion.fail(errorCode); } }); }
private void doLoadLongJob(LongJobVO vo, LongJobOperation operation) { if (operation == null) { operation = getLoadOperation(vo); } if (operation == LongJobOperation.Start) { // launch the waiting jobs ThreadContext.put(Constants.THREAD_CONTEXT_API, vo.getApiId()); LongJob job = longJobFactory.getLongJob(vo.getJobName()); ThreadContext.put(Constants.THREAD_CONTEXT_TASK_NAME, job.getClass().toString()); doStartJob(job, vo, null); SQL.New(LongJobVO.class).eq(LongJobVO_.uuid, vo.getUuid()).set(LongJobVO_.state, LongJobState.Running).update(); } else if (operation == LongJobOperation.Resume) { ThreadContext.put(Constants.THREAD_CONTEXT_API, vo.getApiId()); LongJob job = longJobFactory.getLongJob(vo.getJobName()); ThreadContext.put(Constants.THREAD_CONTEXT_TASK_NAME, job.getClass().toString()); logger.info(String.format("start to resume longjob [uuid:%s, name:%s]", vo.getUuid(), vo.getName())); job.resume(vo); dbf.update(vo); } }
@Override public void run(SyncTaskChain chain) { SubmitLongJobReply reply = new SubmitLongJobReply(); LongJobVO vo = updateByUuid(msg.getJobUuid(), it -> it.setState(LongJobState.Running)); // launch the long job right now ThreadContext.put(Constants.THREAD_CONTEXT_API, vo.getApiId()); LongJob job = longJobFactory.getLongJob(vo.getJobName()); ThreadContext.put(Constants.THREAD_CONTEXT_TASK_NAME, job.getClass().toString()); doStartJob(job, vo, msg); reply.setInventory(LongJobInventory.valueOf(vo)); if (job.getAuditType() != null) { reply.setNeedAudit(true); } logger.info(String.format("longjob [uuid:%s, name:%s] has been started", vo.getUuid(), vo.getName())); bus.reply(msg, reply); chain.next(); }
private void validate(APIRerunLongJobMsg msg) { LongJobVO vo = Q.New(LongJobVO.class) .eq(LongJobVO_.uuid, msg.getUuid()) .find(); LongJobState state = vo.getState(); if (state != LongJobState.Succeeded && state != LongJobState.Canceled && state != LongJobState.Failed) { throw new ApiMessageInterceptionException(argerr("rerun longjob only when it's succeeded, canceled, or failed")); } Class<APIMessage> apiClass = apiMsgOfLongJob.get(vo.getJobName()); APIMessage jobMsg = JSONObjectUtil.toObject(vo.getJobData(), apiClass); jobMsg.setSession(msg.getSession()); try { apiMediator.getProcesser().process(jobMsg); // may throw ApiMessageInterceptionException } catch (StopRoutingException e) { APISubmitLongJobEvent evt = new APISubmitLongJobEvent(msg.getId()); evt.setInventory(LongJobInventory.valueOf(vo)); evt.setNeedAudit(false); bus.publish(evt); throw e; } dbf.updateAndRefresh(vo); }
public LongJobInventory(LongJobVO vo) { this.setUuid(vo.getUuid()); this.setName(vo.getName()); this.setState(vo.getState()); this.setApiId(vo.getApiId()); this.setJobData(vo.getJobData()); this.setJobResult(vo.getJobResult()); this.setJobName(vo.getJobName()); this.setCreateDate(vo.getCreateDate()); this.setLastOpDate(vo.getLastOpDate()); this.setExecuteTime(vo.getExecuteTime()); this.setDescription(vo.getDescription()); this.setTargetResourceUuid(vo.getTargetResourceUuid()); this.setManagementNodeUuid(vo.getManagementNodeUuid()); }
private void handle(APIRerunLongJobMsg msg) { APIRerunLongJobEvent evt = new APIRerunLongJobEvent(msg.getId()); SubmitLongJobMsg smsg = new SubmitLongJobMsg(); LongJobVO job = dbf.findByUuid(msg.getUuid(), LongJobVO.class); smsg.setJobUuid(job.getUuid()); smsg.setDescription(job.getDescription()); smsg.setJobData(job.getJobData()); smsg.setJobName(job.getJobName()); smsg.setName(job.getName()); smsg.setTargetResourceUuid(job.getTargetResourceUuid()); smsg.setResourceUuid(job.getUuid()); smsg.setSystemTags(msg.getSystemTags()); smsg.setUserTags(msg.getUserTags()); smsg.setAccountUuid(msg.getSession().getAccountUuid()); bus.makeLocalServiceId(smsg, LongJobConstants.SERVICE_ID); bus.send(smsg, new CloudBusCallBack(msg) { @Override public void run(MessageReply rly) { SubmitLongJobReply reply = rly.castReply(); evt.setInventory(reply.getInventory()); bus.publish(evt); } }); }