TaskService.java 29 KB

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