package org.jeecg.modules.demo.copywriting.controller;
|
|
import cn.hutool.core.date.DateUtil;
|
import org.jeecg.common.api.vo.Result;
|
import org.jeecg.modules.demo.contract.controller.ContractParam;
|
import org.jeecg.modules.demo.contract.entity.SemanticWord;
|
import org.jeecg.modules.demo.copywriting.entity.CopyGenerateTask;
|
import org.jeecg.modules.demo.copywriting.entity.Copywriting;
|
import org.jeecg.modules.demo.copywriting.service.ICopywritingService;
|
import org.jeecg.modules.demo.copywriting.service.impl.CopyGenerateTaskServiceImpl;
|
import org.jeecg.modules.demo.copywritingScheme.entity.CopywritingScheme;
|
import org.jeecg.modules.demo.copywritingScheme.service.ICopywritingSchemeService;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.scheduling.annotation.Async;
|
import org.springframework.stereotype.Service;
|
|
import java.text.SimpleDateFormat;
|
import java.util.ArrayList;
|
import java.util.Date;
|
import java.util.List;
|
import java.util.UUID;
|
|
@Service
|
public class CopywritingAsyncService {
|
|
@Autowired
|
private CopyGenerateTaskServiceImpl taskService; // 新增的任务状态服务
|
// 其他依赖...
|
|
@Autowired
|
private CopywritingController copywritingController;
|
|
@Autowired
|
private ICopywritingSchemeService copywritingSchemeService;
|
@Autowired
|
public ICopywritingService copywritingService;
|
@Async("copywritingAsyncExecutor")
|
public void asyncBatchGenerateCopy(List<SemanticWord> semanticWords, ContractParam cp) {
|
// 1. 生成唯一任务ID，创建待处理任务
|
String taskId = UUID.randomUUID().toString().replace("-", "");
|
CopyGenerateTask task = new CopyGenerateTask();
|
task.setId(taskId);
|
task.setContractId(cp.getContractId());
|
task.setStatus("PENDING");
|
taskService.save(task);
|
|
try {
|
// 2. 更新为处理中
|
task.setStatus("PROCESSING");
|
taskService.updateById(task);
|
|
// 3. 执行实际的文案生成逻辑（你的核心业务）
|
System.out.println("开始处理合同[" + cp.getContractId() + "]的文案生成");
|
for (SemanticWord semanticWord : semanticWords) {
|
Copywriting copywriting = new Copywriting();
|
Result title = copywritingController.getResult(semanticWord.getOutWord(),
|
semanticWord.getWord(),
|
DateUtil.format(new Date(), new SimpleDateFormat("yyyy-MM-dd")),
|
DateUtil.format(new Date(), new SimpleDateFormat("yyyy-MM-dd")),
|
"e9ca23d68d884d4ebb19d07889727dae");
|
if (title.isSuccess()) {
|
copywriting.setTitle(title.getResult().toString());
|
}
|
copywriting.setStatus("1");
|
copywriting.setWordId(semanticWord.getId());
|
CopywritingScheme copywritingScheme = new CopywritingScheme();
|
copywritingScheme.setId(cp.getCsId());
|
Result text = copywritingController.getResult(
|
cp.getFileName(),
|
cp.getBenchmarkUrl(),
|
cp.getBenchmarkUrl(),
|
semanticWord.getOutWord(),
|
cp.getYoushang(),
|
semanticWord.getWord(),
|
"e9ca23d68d884d4ebb19d07889727dae",copywritingScheme);
|
if (text.isSuccess()) {
|
copywriting.setText(text.getResult().toString());
|
} else {
|
throw new IllegalArgumentException("生成失败！"+text.getMessage());
|
}
|
copywriting.setAuditer(cp.getAuditor());
|
copywriting.setOutStatus("1");
|
copywriting.setCreateBy(cp.getAuditorName());
|
copywriting.setFileName(cp.getFileName());
|
copywriting.setUrlName(cp.getBenchmarkUrl());
|
copywriting.setSchemeId(cp.getCsId());
|
copywritingService.save(copywriting);
|
Thread.sleep(10000);
|
}
|
// 假设生成成功，这里可以记录生成结果
|
task.setStatus("SUCCESS");
|
task.setMessage("文案生成完成，共生成" + semanticWords.size() + "条");
|
task.setFinishTime(new Date());
|
System.out.println("完成处理合同[" + cp.getContractId() + "]的文案生成");
|
|
} catch (Exception e) {
|
// 4. 异常时更新为失败状态，记录原因
|
task.setStatus("FAIL");
|
task.setMessage("生成失败：" + e.getMessage());
|
task.setFinishTime(new Date());
|
e.printStackTrace();
|
} finally {
|
// 5. 最终更新任务状态
|
taskService.updateById(task);
|
}
|
}
|
}
|
