TaskService.java 29 KB

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