TaskService.java 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642
  1. package cn.cslg.pas.service;
  2. import cn.cslg.pas.common.JsonPage;
  3. import cn.cslg.pas.common.model.dto.QueryQrtzTaskLogsDTO;
  4. import cn.cslg.pas.common.model.dto.TaskAddNewDTO;
  5. import cn.cslg.pas.common.model.dto.QueryTaskDTO;
  6. import cn.cslg.pas.common.model.vo.*;
  7. import cn.cslg.pas.common.utils.*;
  8. import cn.cslg.pas.common.utils.SecurityUtils.LoginUtils;
  9. import cn.cslg.pas.domain.Personnel;
  10. import cn.cslg.pas.domain.Project;
  11. import cn.cslg.pas.domain.Report;
  12. import cn.cslg.pas.domain.Task;
  13. import cn.cslg.pas.domain.asso.TaskCondition;
  14. import cn.cslg.pas.mapper.TaskMapper;
  15. import cn.cslg.pas.common.model.dto.UploadFileDTO;
  16. import cn.cslg.pas.service.asso.TaskConditionService;
  17. import cn.cslg.pas.service.upLoadPatent.*;
  18. import cn.hutool.core.io.FileUtil;
  19. import com.alibaba.fastjson.JSON;
  20. import com.alibaba.fastjson.JSONArray;
  21. import com.alibaba.fastjson2.JSONObject;
  22. import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
  23. import com.baomidou.mybatisplus.core.metadata.IPage;
  24. import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
  25. import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
  26. import com.github.pagehelper.PageHelper;
  27. import com.github.pagehelper.PageInfo;
  28. import lombok.RequiredArgsConstructor;
  29. import lombok.extern.slf4j.Slf4j;
  30. import org.quartz.SchedulerException;
  31. import org.springframework.beans.BeanUtils;
  32. import org.springframework.context.annotation.Lazy;
  33. import org.springframework.stereotype.Service;
  34. import java.io.IOException;
  35. import java.text.SimpleDateFormat;
  36. import java.util.*;
  37. import java.util.stream.Collectors;
  38. /**
  39. * <p>
  40. * 任务表 服务类
  41. * </p>
  42. *
  43. * @author 王岩
  44. * @since 2022-02-27
  45. */
  46. @Slf4j
  47. @Service
  48. @RequiredArgsConstructor(onConstructor_ = {@Lazy})
  49. public class TaskService extends ServiceImpl<TaskMapper, Task> implements ITaskService {
  50. private final FileUtils fileUtils;
  51. private final LoginUtils loginUtils;
  52. private final RequestService requestService;
  53. private final OutInterfaceService outInterfaceService;
  54. private final PantentQueueService pantentQueueService;
  55. private final ExcutePatentDataExcel excutePatentDataExcel;
  56. private final ExcutePatentDataEpo excutePatentDataEpo;
  57. private final ExcutePatentDataStar excutePatentDataStar;
  58. private final TaskConditionService taskConditionService;
  59. private final ProjectService projectService;
  60. private final ReportService reportService;
  61. private final JobService jobService;
  62. private final TaskMapper taskMapper;
  63. public String getFileUrl(Integer id) {
  64. Task task = this.getById(id);
  65. return fileUtils.getSystemPath() + task.getUrl();
  66. }
  67. public Integer add(UploadFileDTO fileDTO, Integer projectId, Integer productId, Integer total, Integer type, Integer fieldNum, String oldName) {
  68. //创建任务表实体类,准备装载数据 ↓
  69. Task task = new Task();
  70. //任务开始时间
  71. task.setStartTime(DateUtils.getDateTime());
  72. //任务状态(0.队列中 1.进行中 2.成功 3.失败)
  73. task.setStatus(0);
  74. if (projectId != null) {
  75. //专题库id
  76. task.setProjectId(projectId);
  77. }
  78. if (productId != null) {
  79. //产品id
  80. task.setProductId(productId);
  81. }
  82. //文件名称
  83. task.setFileName(fileDTO.getFileName());
  84. //文件路径
  85. task.setUrl(fileDTO.getPath());
  86. //文件的专利总数量
  87. task.setTotal(total);
  88. //文件大小
  89. task.setFileSize(fileDTO.getFileSize());
  90. //任务类型 (1.上传 2导出)
  91. task.setType(type);
  92. //导入导出字段数量
  93. task.setFieldNum(fieldNum);
  94. //创建人id
  95. task.setCreateBy(loginUtils.getId());
  96. //文件原始名称
  97. task.setOldName(oldName);
  98. //数据入任务表
  99. task.insert();
  100. return task.getId();
  101. }
  102. /**
  103. * 新增导入专利任务
  104. *
  105. * @param fileDTO Excel文件
  106. * @param json 前台参数
  107. * @return 返回任务id
  108. */
  109. public Integer addTask2(UploadFileDTO fileDTO, Integer total, String json) {
  110. //将前台参数json格式转换为实体类(从中取出专题库id存入任务表)
  111. ProjectImportPatentVO projectImportPatentVO = JsonUtils.jsonToPojo(json, ProjectImportPatentVO.class);
  112. //创建任务表实体类,准备装载数据 ↓
  113. Task task = new Task();
  114. //任务开始时间
  115. //task.setStartTime(DateUtils.getDateTime());
  116. //任务状态(0.队列中 1.进行中 2.成功 3.失败)
  117. task.setStatus(0);
  118. if (projectImportPatentVO.getProjectId() != null && projectImportPatentVO.getReportId() == null) {
  119. //若是和专题库关联
  120. task.setProjectId(projectImportPatentVO.getProjectId());
  121. } else if (projectImportPatentVO.getProjectId() == null && projectImportPatentVO.getReportId() != null) {
  122. //若是和报告关联
  123. task.setReportId(projectImportPatentVO.getReportId());
  124. }
  125. //文件名称
  126. task.setFileName(fileDTO.getFileName());
  127. //文件路径
  128. task.setUrl(fileDTO.getPath());
  129. //文件的专利总数量
  130. task.setTotal(total);
  131. //文件大小
  132. task.setFileSize(fileDTO.getFileSize());
  133. //任务类型 (1.上传 2导出)
  134. task.setType(1);
  135. //导入导出字段数量
  136. task.setFieldNum(0);
  137. //创建人id
  138. task.setCreateBy(loginUtils.getId());
  139. //文件原始名称
  140. task.setOldName(fileDTO.getName());
  141. //成功条数
  142. task.setSuccessNum(0);
  143. //失败条数
  144. task.setDefaultNum(0);
  145. //前台参数json格式
  146. task.setPramJson(json);
  147. //任务类型1(0普通任务 1定时任务)
  148. task.setTaskType(0);
  149. task.setAsCompare(projectImportPatentVO.getAsCompare());
  150. //数据入任务表
  151. task.insert();
  152. return task.getId();
  153. }
  154. /**
  155. * 新增网站导入专利任务
  156. *
  157. * @param taskAddNewDTO 前台参数
  158. * @return 返回任务id
  159. */
  160. public Integer addTask3(TaskAddNewDTO taskAddNewDTO, Integer total, TaskCondition taskCondition) {
  161. //创建任务表实体类,准备装载数据 ↓
  162. Task task = new Task();
  163. //任务开始时间
  164. //task.setStartTime(DateUtils.getDateTime());
  165. //任务状态(0.队列中 1.进行中 2.成功 3.失败)
  166. task.setStatus(0);
  167. //判断当前导入专利任务是与专题库关联还是与报告关联
  168. if (taskAddNewDTO.getProjectId() != null && taskAddNewDTO.getReportId() == null) {
  169. //专题库id
  170. task.setProjectId(taskAddNewDTO.getProjectId());
  171. } else if (taskAddNewDTO.getProjectId() == null && taskAddNewDTO.getReportId() != null) {
  172. //报告id
  173. task.setReportId(taskAddNewDTO.getReportId());
  174. }
  175. //专利总数量
  176. task.setTotal(total);
  177. //任务类型 (1Excel导入 2导出 3欧专局导入 4专利之星导入 5专利号导入)
  178. task.setType(taskAddNewDTO.getTaskType2());
  179. //导入导出字段数量
  180. task.setFieldNum(0);
  181. //创建人id
  182. task.setCreateBy(loginUtils.getId());
  183. //成功条数
  184. task.setSuccessNum(0);
  185. //失败条数
  186. task.setDefaultNum(0);
  187. //所属任务条件id
  188. task.setTaskConditionId(taskCondition.getId());
  189. //与专题库关联信息(自定义字段、文件夹)
  190. ProjectImportPatentVO projectImportPatentVO = new ProjectImportPatentVO();
  191. BeanUtils.copyProperties(taskAddNewDTO, projectImportPatentVO);
  192. String pramJson = JSONObject.toJSONString(projectImportPatentVO);
  193. task.setPramJson(pramJson);
  194. //任务类型1(0普通任务 1定时任务)
  195. task.setTaskType(taskAddNewDTO.getTaskType());
  196. //是否直接作为对比数据
  197. task.setAsCompare(taskAddNewDTO.getAsCompare());
  198. //数据入任务表
  199. log.info("任务表新增数据");
  200. task.insert();
  201. return task.getId();
  202. }
  203. public Task edit(Integer id, Integer status) {
  204. Task task = this.getById(id);
  205. task.setStatus(status);
  206. task.setEndTime(DateUtils.getDateTime());
  207. task.updateById();
  208. return task;
  209. }
  210. public String delete(Integer id) {
  211. Task temp = this.getById(id);
  212. this.removeById(id);
  213. if (StringUtils.isNotEmpty(temp.getUrl())) {
  214. FileUtil.del(fileUtils.getSystemPath(temp.getUrl()));
  215. }
  216. return Response.success(true);
  217. }
  218. public void deleteTask(Integer taskId) {
  219. log.info("开始处理【删除任务】的业务,参数为:id={}", taskId);
  220. //检查任务是否存在,若不存在则提示用户"删除失败,当前任务已不存在,请尝试刷新页面"
  221. Task task = this.getById(taskId);
  222. if (task == null) {
  223. ThrowException.throwXiaoShiException("删除失败,当前任务已不存在,请尝试刷新页面");
  224. }
  225. //检查任务状态
  226. //若是已完成或已暂停,则直接删除
  227. if (task.getStatus().equals(2) || task.getStatus().equals(4)) {
  228. this.removeById(task);
  229. //若是网站导入任务,则还要删除任务条件表对应数据
  230. if (task.getType().equals(3) || task.getType().equals(4)) {
  231. taskConditionService.removeById(task.getTaskConditionId());
  232. }
  233. }
  234. //若是等待中,则先要将生产者队列中的该任务剔除,再删除
  235. //若是进行中,则不允许删除,并提示用户"当前任务正在进行中,无法删除"
  236. }
  237. public IPage<Task> getPageList(TaskVO params) throws IOException {
  238. String createName = params.getCreateName();
  239. //如果此次查询是要根据创建人名称查询则👇
  240. if (createName != null && !createName.equals("")) {
  241. //远程调用权限系统的根据创建人名称模糊查询创建人ids的接口
  242. String res = outInterfaceService.getPersonIdByNamePCS(createName);
  243. List<Integer> createPersonIds = JSONArray.parseArray(res, Integer.class);
  244. //当未查询到时给创建人ids集合赋值唯一元素id=-1,即查空页返回
  245. if (createPersonIds == null || createPersonIds.size() == 0) {
  246. createPersonIds = new ArrayList<>(Collections.singletonList(-1));
  247. }
  248. params.setCreatePersonIds(createPersonIds);
  249. }
  250. IPage<Task> pageList = baseMapper.getPageList(new Page<>(params.getCurrent(), params.getSize()), params);
  251. this.setDataList(pageList.getRecords());
  252. return pageList;
  253. }
  254. public PageVO queryTasks(QueryTaskDTO queryTaskDTO) throws IOException {
  255. log.info("开始处理【查询任务】的业务,参数为:{}", queryTaskDTO);
  256. //装载创建人id
  257. queryTaskDTO.setCreateId(loginUtils.getId());
  258. //根据创建人名称模糊查询
  259. if (queryTaskDTO.getCreateName() != null && !queryTaskDTO.getCreateName().equals("")) {
  260. String res = outInterfaceService.getPersonIdByNamePCS(queryTaskDTO.getCreateName());
  261. List<Integer> personIds = JSON.parseArray(res, Integer.class);
  262. if (personIds.size() == 0) { //若根据该名称查询不到任何人员,则将人员ids集合添加一个元素-1(防止sql报错)
  263. personIds.add(-1);
  264. }
  265. queryTaskDTO.setPersonIds(personIds);
  266. }
  267. //若有排序,则根据创建时间排
  268. if (queryTaskDTO.getOrderBy() != null) {
  269. queryTaskDTO.setOrderBy("create_time");
  270. }
  271. //若要分页
  272. if (queryTaskDTO.getCurrent() != null) {
  273. PageHelper.startPage(queryTaskDTO.getCurrent(), queryTaskDTO.getSize());
  274. }
  275. List<QueryTaskVO> tasks = taskMapper.queryTasks(queryTaskDTO);
  276. PageInfo<QueryTaskVO> pageInfo = new PageInfo<>(tasks);
  277. //装载创建人名称、装载任务名称
  278. if (tasks.size() > 0) {
  279. //装载创建人名称
  280. List<Integer> createPersonIds = tasks.stream().map(QueryTaskVO::getCreateBy).collect(Collectors.toList());
  281. String res = outInterfaceService.getPersonnelByIdsFromPCS(createPersonIds);
  282. com.alibaba.fastjson.JSONObject jsonObject = com.alibaba.fastjson.JSONObject.parseObject(res);
  283. List<Personnel> personnels = com.alibaba.fastjson.JSONObject.parseArray(jsonObject.getString("data"), Personnel.class);
  284. for (QueryTaskVO task : tasks) {
  285. for (Personnel personnel : personnels) {
  286. if (personnel.getId().equals(task.getCreateBy())) {
  287. task.setCreateName(personnel.getPersonnelName());
  288. break;
  289. }
  290. }
  291. }
  292. //装载任务名称(专利号/检索式 导入到 某某 专题库/报告中)
  293. if (queryTaskDTO.getProjectId() != null) { //若当前查询是查询所有导入专题库的任务,则装载专题库名称
  294. List<Integer> projectIds = tasks.stream().map(QueryTaskVO::getProjectId).collect(Collectors.toList());
  295. List<Project> projects = projectService.list(new LambdaQueryWrapper<Project>().in(Project::getId, projectIds));
  296. tasks.forEach(task -> {
  297. projects.forEach(project -> {
  298. if (task.getProjectId().equals(project.getId())) {
  299. task.setProjectName(project.getName());
  300. }
  301. });
  302. });
  303. } else if (queryTaskDTO.getReportId() != null) { //若当前查询是查询所有导入报告的任务,则装载报告名称
  304. List<Integer> reportIds = tasks.stream().map(QueryTaskVO::getReportId).collect(Collectors.toList());
  305. String res2 = outInterfaceService.getReportsFromRMS(reportIds);
  306. JSONObject jsonObject2 = JSONObject.parseObject(res2);
  307. List<Report> reports = JSON.parseArray(jsonObject2.getString("data"), Report.class);
  308. tasks.forEach(task -> {
  309. reports.forEach(report -> {
  310. if (task.getReportId().equals(report.getId())) {
  311. task.setReportName(report.getName());
  312. }
  313. });
  314. });
  315. }
  316. for (QueryTaskVO task : tasks) {
  317. String taskName = "";
  318. if (task.getType().equals(5)) { //若是专利号导入任务
  319. if (task.getProjectName() != null) {
  320. taskName = "专利号导入到专题库【" + task.getProjectName() + "】中 - " + task.getCreateName();
  321. } else if (task.getReportName() != null) {
  322. taskName = "专利号导入到报告【" + task.getReportName() + "】中 - " + task.getCreateName();
  323. }
  324. task.setTaskName(taskName);
  325. } else if (task.getType().equals(3) || task.getType().equals(4)) { //若是检索式导入(欧专局或专利之星)
  326. if (task.getProjectName() != null) {
  327. taskName = "检索式导入到专题库【" + task.getProjectName() + "】中 - " + task.getCreateName();
  328. } else if (task.getReportName() != null) {
  329. taskName = "检索式导入到报告【" + task.getReportName() + "】中 - " + task.getCreateName();
  330. }
  331. task.setTaskName(taskName);
  332. } else if (task.getType().equals(2)) { //若是导出
  333. if (task.getProjectName() != null) {
  334. taskName = "从专题库【" + task.getProjectName() + "】导出 - " + task.getCreateName();
  335. } else if (task.getReportName() != null) {
  336. taskName = "从报告【" + task.getReportName() + "】导出 - " + task.getCreateName();
  337. }
  338. task.setTaskName(taskName);
  339. }
  340. }
  341. }
  342. //手动创建分页信息返回对象
  343. PageVO pageVO = new PageVO()
  344. .setCurrent(pageInfo.getPageNum())
  345. .setSize(pageInfo.getPageSize())
  346. .setTotal((int) pageInfo.getTotal())
  347. .setRecords(tasks);
  348. return pageVO;
  349. }
  350. public Page<Task> queryQrtzTaskLogs(QueryQrtzTaskLogsDTO queryQrtzTaskLogsDTO) {
  351. log.info("开始处理【查询定时任务执行日志】的业务,参数为:taskId={}", queryQrtzTaskLogsDTO.getTaskId());
  352. //检查任务是否存在,若不存在则提示用户"该定时任务已不存在,请尝试刷新页面"
  353. Task task = this.getById(queryQrtzTaskLogsDTO.getTaskId());
  354. if (task == null) {
  355. ThrowException.throwXiaoShiException("该定时任务已不存在,请尝试刷新页面");
  356. }
  357. //检查任务是否为定时任务,若不是则提示用户"该任务不为定时任务,暂无更新日志"
  358. TaskCondition taskCondition = taskConditionService.getById(task.getTaskConditionId());
  359. if (!taskCondition.getTaskType().equals(1)) {
  360. ThrowException.throwXiaoShiException("该任务不为定时任务,暂无更新日志");
  361. }
  362. Page<Task> pageList = this.page(new Page<>(queryQrtzTaskLogsDTO.getCurrent(), queryQrtzTaskLogsDTO.getSize()), new LambdaQueryWrapper<Task>().eq(Task::getTaskConditionId, task.getTaskConditionId()).orderByDesc(Task::getCreateTime));
  363. return pageList;
  364. }
  365. public void cancelTask(Integer taskId) throws SchedulerException, InterruptedException {
  366. log.info("开始处理【取消任务】的业务,参数为:taskId={}", taskId);
  367. //检查任务是否存在,若不存在则提示用户"取消任务失败,当前任务已不存在,请尝试刷新页面"
  368. Task task = this.getById(taskId);
  369. if (task == null) {
  370. ThrowException.throwXiaoShiException("取消任务失败,当前任务已不存在,请尝试刷新页面");
  371. return;
  372. }
  373. //检查任务状态是否为已完成(普通任务),若是则提示用户"取消任务失败,已完成的任务无法取消"
  374. if (task.getStatus().equals(2) && task.getTaskType().equals(0)) {
  375. ThrowException.throwXiaoShiException("取消任务失败,已完成的任务无法取消");
  376. }
  377. //若是定时任务(即本次请求目的是取消该定时任务),删除job中的该定时任务条件
  378. if (task.getTaskType().equals(1)) {
  379. //删除job中的该定时任务条件
  380. jobService.deleteJob(task.getTaskConditionId());
  381. }
  382. //若该任务处于进行中
  383. if (task.getStatus().equals(1)) {
  384. //将生产者生产专利中的公共任务状态改为暂停
  385. if (task.getType() == 1) {
  386. excutePatentDataExcel.setTaskStatus(taskId, 4);
  387. } else if (task.getType() == 3) {
  388. excutePatentDataEpo.setTaskStatus(taskId, 4);
  389. } else if (task.getType() == 4) {
  390. excutePatentDataStar.setTaskStatus(taskId, 4);
  391. }
  392. if (task.getType().equals(3) || task.getType().equals(4)) {
  393. Thread.sleep(500);
  394. //将专利各部分装载线程队列中该任务的专利清空
  395. pantentQueueService.setQueueRemoveTasks(taskId);
  396. }
  397. Thread.sleep(1000);
  398. //将消费者的专利队列中所有该任务的专利剔除
  399. pantentQueueService.consumerQueueRemoveTasks(taskId);
  400. }
  401. //若该任务处于等待中
  402. if (task.getStatus().equals(0)) {
  403. //将生产者任务队列中该任务剔除
  404. pantentQueueService.taskQueueListRemoveTask(Arrays.asList(task.getId()));
  405. }
  406. //最后修改任务状态为5已取消
  407. Task updateTask = new Task();
  408. updateTask.setId(task.getId());
  409. updateTask.setStatus(5);
  410. updateTask.setEndTime(DateUtils.getDateTime());
  411. log.info("os_task表修改任务状态为取消");
  412. this.updateById(updateTask);
  413. }
  414. public List<Task> getQueueList(Integer type, Integer projectId, Integer productId) {
  415. LambdaQueryWrapper<Task> queryWrapper = new LambdaQueryWrapper<>();
  416. queryWrapper.eq(Task::getCreateBy, loginUtils.getId());
  417. queryWrapper.eq(Task::getType, type);
  418. queryWrapper.in(Task::getStatus, new ArrayList<Integer>() {{
  419. add(0);
  420. add(1);
  421. }});
  422. if (projectId != null) {
  423. queryWrapper.eq(Task::getProjectId, projectId);
  424. }
  425. if (productId != null) {
  426. queryWrapper.eq(Task::getProductId, productId);
  427. }
  428. queryWrapper.orderByAsc(Task::getStartTime);
  429. return this.list(queryWrapper);
  430. }
  431. public Integer add2(Integer projectId, Long fieldNum, Integer total) {
  432. Task task = new Task();
  433. task.setStatus(0);
  434. task.setProjectId(projectId);
  435. task.setType(2);
  436. task.setTotal(total);
  437. task.setFieldNum(fieldNum.intValue());
  438. task.setCreateBy(loginUtils.getId());
  439. task.setStartTime(DateUtils.getDateTime());
  440. task.insert();
  441. return task.getId();
  442. }
  443. public void updateStatus(Integer id, Integer status, Integer endTime) {
  444. Task task = this.getById(id);
  445. task.setStatus(status);
  446. if (endTime != null) {
  447. task.setEndTime(endTime);
  448. }
  449. task.updateById();
  450. }
  451. public void updateTime(Integer id, Integer status, Integer endTime, String url, String fileName) {
  452. Task task = this.getById(id);
  453. task.setStatus(status);
  454. task.setEndTime(endTime);
  455. task.setFileName(fileName);
  456. task.setUrl(url);
  457. task.updateById();
  458. }
  459. private void setDataList(List<Task> taskList) throws IOException {
  460. //获得创建人的id集合
  461. List<Integer> createIds = taskList.stream().map(Task::getCreateBy).collect(Collectors.toList());
  462. //获取专题库负责人对应信息
  463. String jsonObject1 = requestService.getPersonnelFromPCS(createIds);
  464. JSONArray jsonArray = JSON.parseArray(jsonObject1);
  465. List<ProjectVO.Personnel> personnelList = jsonArray.toJavaList(ProjectVO.Personnel.class);
  466. for (Task task : taskList) {
  467. for (ProjectVO.Personnel personnel : personnelList) {
  468. //装载创建人名
  469. if (task.getCreateBy() != null) {
  470. if (task.getCreateBy().equals(personnel.getId())) {
  471. task.setCreateName(personnel.getPersonnelName());
  472. }
  473. }
  474. }
  475. }
  476. }
  477. /**
  478. * 暂停任务
  479. *
  480. * @param taskId 任务id
  481. * @return 返回任务状态4,已暂停
  482. */
  483. @Override
  484. public Integer pauseTask(Integer taskId) throws InterruptedException {
  485. log.info("开始处理【暂停任务】的业务,参数为:taskId={}", taskId);
  486. //检查任务是否存在,若不存在则提示用户 "暂停失败,当前任务已不存在,请尝试刷新页面"
  487. log.info("检查尝试暂停的任务是否存在");
  488. Task task = this.getById(taskId);
  489. if (task == null) {
  490. ThrowException.throwXiaoShiException("暂停失败,当前任务已不存在,请尝试刷新页面");
  491. return null;
  492. }
  493. //检查任务是否处于进行中状态,若不是则提示用户 "暂停失败,当前任务不处于进行中状态,请尝试刷新页面"
  494. if (!task.getStatus().equals(1)) {
  495. ThrowException.throwXiaoShiException("暂停失败,当前任务不在进行中,请尝试刷新页面");
  496. }
  497. //任务表将任务状态修改为4暂停
  498. Task updateTask = new Task();
  499. updateTask.setId(taskId);
  500. updateTask.setStatus(4);
  501. log.info("任务表修改数据,暂停任务,任务id={}", taskId);
  502. boolean updateResult = this.updateById(updateTask);
  503. if (!updateResult) {
  504. ThrowException.throwXiaoShiException("暂停失败,服务器忙请稍后再试");
  505. }
  506. //将检索生产专利时的公共任务状态改为暂停(若还在检索生产,则会结束)
  507. if (task.getType() == 1) { //Excel导入
  508. excutePatentDataExcel.setTaskStatus(taskId, 4);
  509. } else if (task.getType() == 3) { //欧专局
  510. excutePatentDataEpo.setTaskStatus(taskId, 4);
  511. } else if (task.getType() == 4) { //专利之星
  512. excutePatentDataStar.setTaskStatus(taskId, 4);
  513. }
  514. //若是网站导入任务,则还要将专利各部分装载线程队列中该任务元素清空
  515. if (task.getType().equals(3) || task.getType().equals(4)) {
  516. //等上半秒(上述代码虽然使得生产暂停,但是生产暂停前正在存入生产队列的过程中,等2秒使得这些放入消费者队列后再清空消费者)
  517. pantentQueueService.setQueueRemoveTasks(taskId);
  518. }
  519. //等上一秒(上述代码虽然使得生产暂停,但是生产暂停前已生产的专利正在存入消费者队列的过程中,等2秒使得这些放入消费者队列后再清空消费者)
  520. Thread.sleep(1500);
  521. //将消费者的专利队列中所有该任务的专利剔除
  522. pantentQueueService.consumerQueueRemoveTasks(taskId);
  523. log.info("暂停任务成功");
  524. return updateTask.getStatus();
  525. }
  526. /**
  527. * 继续任务
  528. *
  529. * @param taskId 任务id
  530. */
  531. @Override
  532. public Integer continueTask(Integer taskId) {
  533. log.info("开始处理【继续任务】的业务,参数为:taskId={}", taskId);
  534. //检查任务是否存在,若不存在则提示用户 "继续任务失败,当前任务已不存在,请尝试刷新页面"
  535. log.info("检查尝试继续的任务是否存在");
  536. Task task = this.getById(taskId);
  537. if (task == null) {
  538. ThrowException.throwXiaoShiException("继续任务失败,当前任务已不存在,请尝试刷新页面");
  539. return null;
  540. }
  541. //检查任务是否处于暂停状态,若不是则提示用户 "继续任务失败,当前任务未暂停,请尝试刷新页面"
  542. if (!task.getStatus().equals(4)) {
  543. ThrowException.throwXiaoShiException("继续任务失败,当前任务未暂停,请尝试刷新页面");
  544. }
  545. //继续任务(将任务状态改为0即等待中)
  546. Task updateTask = new Task();
  547. updateTask.setId(taskId);
  548. updateTask.setStatus(0);
  549. log.info("任务表修改数据,继续任务,任务id={}", taskId);
  550. boolean updateResult = this.updateById(updateTask);
  551. if (!updateResult) {
  552. ThrowException.throwXiaoShiException("继续任务失败,服务器忙请稍后再试");
  553. }
  554. //将当前任务添加进生产者任务队列进行排队,并唤醒生产者线程
  555. pantentQueueService.taskQueueListAddTask(new ArrayList<>(Arrays.asList(taskId)));
  556. //将生产者生产专利类中的公共任务状态从暂停改为0
  557. if (task.getType().equals(1) && excutePatentDataExcel.getpTaskId().equals(task.getId())) {
  558. excutePatentDataExcel.setTaskStatus(taskId, 0);
  559. } else if (task.getType().equals(3) && excutePatentDataEpo.getpTaskId().equals(task.getId())) {
  560. excutePatentDataEpo.setTaskStatus(taskId, 0);
  561. } else if (task.getType().equals(4) && excutePatentDataStar.getpTaskId().equals(task.getId())) {
  562. excutePatentDataStar.setTaskStatus(taskId, 0);
  563. }
  564. pantentQueueService.awakeTasktch();
  565. log.info("继续任务成功");
  566. return updateTask.getStatus();
  567. }
  568. }