package cn.iocoder.yudao.module.mes.service.pro.bagcode;
|
|
import cn.iocoder.yudao.module.mes.dal.mysql.pro.bagcode.MesProBagCodeGenerateTaskMapper;
|
import cn.iocoder.yudao.module.mes.enums.pro.MesProBagCodeGenerateTaskStatusEnum;
|
import cn.iocoder.yudao.module.mes.framework.bagcode.MesProBagCodeGenerateTaskRegistry;
|
import jakarta.annotation.Resource;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.stereotype.Component;
|
|
import java.time.LocalDateTime;
|
import java.util.concurrent.atomic.AtomicBoolean;
|
|
/**
|
* 袋码生成任务执行器
|
* <p>
|
* 在后台线程内完成分段生成，并负责：回写进度、响应取消、失败/取消后清理半成品。
|
*/
|
@Slf4j
|
@Component
|
public class MesProBagCodeGenerateRunner {
|
|
@Resource
|
private MesProBagCodeGenerateTaskMapper taskMapper;
|
@Resource
|
private MesProBagCodeBatchGenerator generator;
|
@Resource
|
private MesProBagCodeGenerateCleaner cleaner;
|
@Resource
|
private MesProBagCodeGenerateTaskRegistry registry;
|
|
/**
|
* 执行生成任务（由线程池调用，勿在请求线程内直接调用）
|
*
|
* @param taskId 任务编号
|
* @param spec 生成参数
|
*/
|
public void run(Long taskId, MesProBagCodeGenerateSpec spec) {
|
// 取消标记在提交时已注册，这里只取用；取不到说明注册环节被绕过，补一个以保证取消可用
|
AtomicBoolean cancelFlag = registry.get(taskId);
|
if (cancelFlag == null) {
|
cancelFlag = registry.register(taskId);
|
}
|
try {
|
taskMapper.updateStatus(taskId, MesProBagCodeGenerateTaskStatusEnum.RUNNING.getStatus(),
|
null, null);
|
int finished = generator.generate(spec, taskId, cancelFlag).finishedCount();
|
// 最后一段写入后、任务收尾前才收到取消：仍按取消处理，保证"取消后无半成品"
|
if (cancelFlag.get()) {
|
throw new MesProBagCodeBatchGenerator.CancelledException();
|
}
|
taskMapper.updateProgress(taskId, finished);
|
taskMapper.updateStatus(taskId, MesProBagCodeGenerateTaskStatusEnum.FINISHED.getStatus(),
|
null, LocalDateTime.now());
|
log.info("[run][生成任务 {} 完成，共 {} 条袋码]", taskId, finished);
|
} catch (MesProBagCodeBatchGenerator.CancelledException e) {
|
cleanup(taskId, spec.uuid());
|
taskMapper.updateStatus(taskId, MesProBagCodeGenerateTaskStatusEnum.CANCELED.getStatus(),
|
null, LocalDateTime.now());
|
log.info("[run][生成任务 {} 已取消]", taskId);
|
} catch (Exception e) {
|
cleanup(taskId, spec.uuid());
|
taskMapper.updateStatus(taskId, MesProBagCodeGenerateTaskStatusEnum.FAILED.getStatus(),
|
truncateError(e), LocalDateTime.now());
|
log.error("[run][生成任务 {} 生成失败]", taskId, e);
|
} finally {
|
registry.unregister(taskId);
|
registry.release();
|
}
|
}
|
|
/**
|
* 清理半成品：清理失败只记录日志，不能影响任务终态回写，否则任务会一直挂在"生成中"
|
*/
|
private void cleanup(Long taskId, String uuid) {
|
try {
|
cleaner.cleanup(uuid);
|
} catch (Exception e) {
|
log.error("[cleanup][生成任务 {} 半成品清理失败，uuid={}]", taskId, uuid, e);
|
}
|
}
|
|
private String truncateError(Exception e) {
|
String message = e.getMessage();
|
if (message == null || message.isBlank()) {
|
message = e.getClass().getSimpleName();
|
}
|
return message.length() > 500 ? message.substring(0, 500) : message;
|
}
|
|
}
|
