SavePatentToEsThread.java 9.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195
  1. package cn.cslg.pas.service.importPatent;
  2. import cn.cslg.pas.common.dto.UploadPatentWebDTO;
  3. import cn.cslg.pas.common.dto.es.EsCustomFieldDTO;
  4. import cn.cslg.pas.common.utils.FormatUtil;
  5. import cn.cslg.pas.common.vo.ImportTaskAMVO;
  6. import cn.cslg.pas.common.vo.PatentWithIdVO;
  7. import cn.cslg.pas.domain.business.ImportTask;
  8. import cn.cslg.pas.domain.es.ESImportTask;
  9. import cn.cslg.pas.domain.es.Patent;
  10. import cn.cslg.pas.domain.es.PatentJoin;
  11. import cn.cslg.pas.service.business.CompareLiteratureService;
  12. import cn.cslg.pas.service.business.ImportTaskService;
  13. import cn.cslg.pas.service.business.es.EsCustomFieldService;
  14. import cn.cslg.pas.service.business.es.EsMergePersonService;
  15. import cn.cslg.pas.service.business.es.EsProductPatentService;
  16. import cn.cslg.pas.service.business.es.EsService;
  17. import cn.cslg.pas.service.common.FileManagerService;
  18. import cn.cslg.pas.service.common.MessageService;
  19. import cn.cslg.pas.service.common.PatentStarApiService;
  20. import org.springframework.beans.BeanUtils;
  21. import org.springframework.beans.factory.annotation.Autowired;
  22. import org.springframework.context.ApplicationContext;
  23. import java.io.IOException;
  24. import java.util.ArrayList;
  25. import java.util.Arrays;
  26. import java.util.List;
  27. import java.util.concurrent.locks.Condition;
  28. import java.util.concurrent.locks.Lock;
  29. import java.util.concurrent.locks.ReentrantLock;
  30. public class SavePatentToEsThread extends Thread {
  31. private ApplicationContext applicationContext;
  32. private List<UploadPatentWebDTO> uploadPatentWebDTOS = new ArrayList<>();
  33. private Lock taskLock = new ReentrantLock();
  34. private Condition taskCondition = taskLock.newCondition();
  35. private TaskThread taskThread;
  36. private ImportTaskAMVO importTaskAMVO;
  37. private Boolean ifProductAll = false;
  38. @Override
  39. public void run() {
  40. while ((!ifProductAll || uploadPatentWebDTOS.size() > 0) && importTaskAMVO.getState().equals(1)) {
  41. try {
  42. //判断任务队列是否有任务,若没有则线程等待唤醒
  43. if (uploadPatentWebDTOS.size() == 0) {
  44. taskLock.lock();
  45. taskCondition.await();
  46. }
  47. } catch (Exception e) {
  48. }
  49. if(uploadPatentWebDTOS.size()==0){
  50. break;
  51. }
  52. UploadPatentWebDTO uploadPatentWebDTO = uploadPatentWebDTOS.remove(0);
  53. Patent patent = uploadPatentWebDTO.getPatent();
  54. try {
  55. //根据专利号查询专利
  56. EsService esService = applicationContext.getBean(EsService.class);
  57. PatentWithIdVO patentWithIdVO = esService.getIdByPatentNo(patent.getPatentNo());
  58. String patentId = null;
  59. // 若查出专利则更新
  60. if (patentWithIdVO != null) {
  61. patentId = patentWithIdVO.getId();
  62. Patent orgPatent = patentWithIdVO.getPatent();
  63. BeanUtils.copyProperties(patent, orgPatent, FormatUtil.getNullPropertyNames(patent));
  64. esService.updatePatent(orgPatent, patentWithIdVO.getId());
  65. } else {
  66. PatentJoin patentJoin = new PatentJoin();
  67. patentJoin.setName("patent");
  68. patent.setPatentJoin(patentJoin);
  69. patentId = esService.addPatent(patent);
  70. }
  71. //判断是否和专题库或报告关联
  72. ImportTaskAMVO importTaskAMVO = taskThread.getImportTaskAMVO();
  73. //和专题库或报告进行关联
  74. if (importTaskAMVO.getProjectId() != null) {
  75. //根据projectId和专利id查询
  76. if (patentId != null) {
  77. Boolean ifInproject = esService.searchPatent(patentId, importTaskAMVO.getProjectId());
  78. if (!ifInproject) {
  79. Patent patentChild = new Patent();
  80. PatentJoin patentJoin = new PatentJoin();
  81. patentJoin.setParent(patentId);
  82. patentJoin.setName("project");
  83. patentChild.setPatentJoin(patentJoin);
  84. patentChild.setProjectId(importTaskAMVO.getProjectId());
  85. esService.addChildPatent(patentChild, patentId);
  86. //当导入专题库时添加默认合并申请人
  87. if (importTaskAMVO.getProjectType() != null && importTaskAMVO.getProjectType().equals(0)) {
  88. EsMergePersonService esMergePersonService = applicationContext.getBean(EsMergePersonService.class);
  89. esMergePersonService.addDefaultMergePerson(patent, patentId, importTaskAMVO.getProjectId());
  90. }
  91. }
  92. //添加报告对比文件
  93. if (importTaskAMVO.getProjectType() != null && importTaskAMVO.getProjectType().equals(1)) {
  94. if (importTaskAMVO.getReportType().equals(7) || importTaskAMVO.getReportType().equals(1) || importTaskAMVO.getReportType().equals(2)) {
  95. CompareLiteratureService compareLiteratureService = applicationContext.getBean(CompareLiteratureService.class);
  96. compareLiteratureService.addPatentCompareLiterature(patent, importTaskAMVO.getProjectId(), importTaskAMVO.getCreateId());
  97. }
  98. }
  99. }
  100. //和任务关联
  101. if (importTaskAMVO.getId() != null) {
  102. if (patentId != null) {
  103. Boolean ifInTask = esService.ifInTask(patentId, importTaskAMVO.getProjectId(), importTaskAMVO.getId());
  104. if (!ifInTask) {
  105. Patent patentChild = new Patent();
  106. PatentJoin patentJoin = new PatentJoin();
  107. patentJoin.setParent(patentId);
  108. patentJoin.setName("import_task");
  109. patentChild.setPatentJoin(patentJoin);
  110. ESImportTask esImportTask = new ESImportTask();
  111. esImportTask.setProjectId(importTaskAMVO.getProjectId());
  112. esImportTask.setTaskId(importTaskAMVO.getId());
  113. patentChild.setImportTask(esImportTask);
  114. esService.addChildPatent(patentChild, patentId);
  115. }
  116. }
  117. }
  118. } else if (importTaskAMVO.getProductId() != null && patentId != null) {
  119. EsProductPatentService esProductPatentService = applicationContext.getBean(EsProductPatentService.class);
  120. esProductPatentService.addProductPatent(patentId, importTaskAMVO.getProductId());
  121. }
  122. //保存和自定义字段关联
  123. if (importTaskAMVO.getFieldDTOS() != null && importTaskAMVO.getFieldDTOS().size() > 0) {
  124. for (EsCustomFieldDTO esCustomFieldDTO : importTaskAMVO.getFieldDTOS()
  125. ) {
  126. esCustomFieldDTO.setPatentId(patentId);
  127. esCustomFieldDTO.setOptionType(2);
  128. esCustomFieldDTO.setPatentNo(patent.getPatentNo());
  129. EsCustomFieldService esCustomFieldService = applicationContext.getBean(EsCustomFieldService.class);
  130. esCustomFieldService.addCustomField(esCustomFieldDTO);
  131. }
  132. }
  133. if (uploadPatentWebDTO.getEsCustomFieldDTOList() != null && uploadPatentWebDTO.getEsCustomFieldDTOList().size() != 0) {
  134. for (EsCustomFieldDTO esCustomFieldDTO : uploadPatentWebDTO.getEsCustomFieldDTOList()
  135. ) {
  136. esCustomFieldDTO.setPatentId(patentId);
  137. esCustomFieldDTO.setOptionType(2);
  138. esCustomFieldDTO.setPatentNo(patent.getPatentNo());
  139. EsCustomFieldService esCustomFieldService = applicationContext.getBean(EsCustomFieldService.class);
  140. esCustomFieldService.addCustomField(esCustomFieldDTO);
  141. }
  142. }
  143. //导入完成,通知前台
  144. taskThread.updateProcess(false, 1, "");
  145. } catch (Exception e) {
  146. taskThread.updateProcess(true, 1, "");
  147. }
  148. }
  149. taskThread.awakeTaskThread();
  150. System.out.println("专利信息装载完毕");
  151. }
  152. public SavePatentToEsThread(TaskThread taskThread, ApplicationContext applicationContext) {
  153. this.taskThread = taskThread;
  154. this.applicationContext = applicationContext;
  155. this.importTaskAMVO = taskThread.getImportTaskAMVO();
  156. }
  157. public void awakeTask(UploadPatentWebDTO uploadPatentWebDTO) {
  158. UploadPatentWebDTO uploadPatentWebDTO1 = new UploadPatentWebDTO();
  159. BeanUtils.copyProperties(uploadPatentWebDTO, uploadPatentWebDTO1);
  160. uploadPatentWebDTOS.add(uploadPatentWebDTO1);
  161. if (taskLock.tryLock()) {
  162. taskCondition.signalAll();
  163. taskLock.unlock();
  164. }
  165. }
  166. public void setIfProductAll(Boolean ifProductAll) {
  167. this.ifProductAll = ifProductAll;
  168. System.out.println("专利导入全部结束" + this.ifProductAll + importTaskAMVO.getId());
  169. if (taskLock.tryLock()) {
  170. System.out.println("专利导入到es解锁" + importTaskAMVO.getId());
  171. taskCondition.signalAll();
  172. taskLock.unlock();
  173. }
  174. }
  175. }