CustomFieldService.java 36 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838
  1. package cn.cslg.pas.service.business;
  2. import cn.cslg.pas.common.dto.AllFieldOrderDTO;
  3. import cn.cslg.pas.common.dto.business.CustomFieldDTO;
  4. import cn.cslg.pas.common.dto.business.UpdateCustomFieldDTO;
  5. import cn.cslg.pas.common.dto.es.EsCustomFieldDTO;
  6. import cn.cslg.pas.common.dto.es.EsCustomFieldValueDTO;
  7. import cn.cslg.pas.common.dto.es.QueryEsCustomFieldDTO;
  8. import cn.cslg.pas.common.model.cronModel.Personnel;
  9. import cn.cslg.pas.common.model.cronModel.PersonnelVO;
  10. import cn.cslg.pas.common.model.cronModel.Records;
  11. import cn.cslg.pas.common.model.request.GroupRequest;
  12. import cn.cslg.pas.common.model.request.QueryRequest;
  13. import cn.cslg.pas.common.utils.*;
  14. import cn.cslg.pas.common.vo.FieldValueVO;
  15. import cn.cslg.pas.common.vo.PatentWithIdVO;
  16. import cn.cslg.pas.common.vo.UploadParamsVO;
  17. import cn.cslg.pas.common.vo.business.AllCustomFieldVO;
  18. import cn.cslg.pas.common.vo.business.CustomFieldVO;
  19. import cn.cslg.pas.common.vo.business.ProductVO;
  20. import cn.cslg.pas.common.vo.es.QueryESCustomFieldVO;
  21. import cn.cslg.pas.domain.business.*;
  22. import cn.cslg.pas.domain.es.Patent;
  23. import cn.cslg.pas.exception.UnLoginException;
  24. import cn.cslg.pas.exception.XiaoShiException;
  25. import cn.cslg.pas.factorys.businessFactory.Business;
  26. import cn.cslg.pas.mapper.CustomFieldMapper;
  27. import cn.cslg.pas.service.AssoTaskFieldService;
  28. import cn.cslg.pas.service.business.es.EsCustomFieldService;
  29. import cn.cslg.pas.service.business.es.EsService;
  30. import cn.cslg.pas.service.permissions.PermissionService;
  31. import cn.cslg.pas.service.query.FormatQueryService;
  32. import com.alibaba.fastjson.JSONArray;
  33. import com.alibaba.fastjson.JSONObject;
  34. import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
  35. import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
  36. import org.springframework.beans.BeanUtils;
  37. import org.springframework.beans.factory.annotation.Autowired;
  38. import org.springframework.context.annotation.Lazy;
  39. import org.springframework.stereotype.Service;
  40. import org.springframework.util.CollectionUtils;
  41. import org.springframework.web.multipart.MultipartFile;
  42. import java.io.IOException;
  43. import java.util.*;
  44. import java.util.stream.Collectors;
  45. /**
  46. * 自定义栏位Service层
  47. *
  48. * @Author xiexiang
  49. * @Date 2023/11/8
  50. */
  51. @Service
  52. public class CustomFieldService extends ServiceImpl<CustomFieldMapper, CustomField> implements Business {
  53. @Autowired
  54. private CacheUtils cacheUtils;
  55. @Autowired
  56. private LoginUtils loginUtils;
  57. @Autowired
  58. private FormatQueryService formatQueryService;
  59. @Autowired
  60. private CustomFieldMapper customFieldMapper;
  61. @Autowired
  62. private PermissionService permissionService;
  63. @Autowired
  64. private AssoProjectTreeNodeService assoProjectTreeNodeService;
  65. @Autowired
  66. private ProductCategoryService productCategoryService;
  67. @Autowired
  68. private ProductService productService;
  69. @Autowired
  70. private EsCustomFieldService esCustomFieldService;
  71. @Lazy
  72. @Autowired
  73. private CustomOptionService customOptionService;
  74. @Autowired
  75. private TreeNodeService treeNodeService;
  76. @Autowired
  77. private AssoTaskFieldService assoTaskFieldService;
  78. @Autowired
  79. private EsService esService;
  80. @Override
  81. public Object queryMessage(QueryRequest queryRequest) throws Exception {
  82. //根据专题库/报告id查询自定义栏位
  83. List<String> sqls = formatQueryService.reSqls(queryRequest, "customField");
  84. //根据sql查询自定义栏位信息
  85. List<CustomFieldVO> customFieldVOS = customFieldMapper.getCustomField(sqls.get(0), sqls.get(1), sqls.get(2));
  86. //查询总数
  87. Long total = customFieldMapper.getCustomFieldCount(sqls.get(0));
  88. //装载自定义栏位信息
  89. this.loadCustomField(customFieldVOS);
  90. //装载返回信息
  91. Records records = new Records();
  92. records.setCurrent(queryRequest.getCurrent());
  93. records.setSize(queryRequest.getSize());
  94. records.setData(customFieldVOS);
  95. records.setTotal(total);
  96. return records;
  97. }
  98. @Override
  99. public Object addMessage(Object object, List<MultipartFile> files) {
  100. return null;
  101. }
  102. @Override
  103. public Object deleteMessage(List<Integer> ids) throws IOException {
  104. if (ids == null || ids.size() == 0) {
  105. throw new XiaoShiException("ids不能为空");
  106. }
  107. this.removeBatchByIds(ids);
  108. //TODO 删除缓存中保存的栏位信息
  109. //TODO 删除与专题库的关联关系
  110. return ids;
  111. }
  112. @Override
  113. public Object updateMessage(Object object, List<MultipartFile> files) {
  114. return null;
  115. }
  116. @Override
  117. public Object getGroup(GroupRequest groupRequest, String tableName) throws Exception {
  118. return null;
  119. }
  120. @Override
  121. public Object addMessage(Object object) {
  122. if (object == null) {
  123. throw new XiaoShiException("参数不能为空");
  124. }
  125. //object to customFieldDTO
  126. CustomFieldDTO customFieldDTO = (CustomFieldDTO) object;
  127. //非空校验
  128. if (customFieldDTO.getProjectId().equals(null)) {
  129. throw new XiaoShiException("专题库/报告id不能为空!");
  130. }
  131. if (customFieldDTO.getName().equals(null) || customFieldDTO.getName().equals("")) {
  132. throw new XiaoShiException("自定义栏位名称不能为空!");
  133. }
  134. if (customFieldDTO.getType().equals(null)) {
  135. throw new XiaoShiException("自定义栏位类型不能为空!");
  136. }
  137. String name = customFieldDTO.getName();
  138. Integer projectId = customFieldDTO.getProjectId();
  139. //检查名称是否规范
  140. customFieldDTO.setName(name.trim());
  141. //同一个专题库/报告id下名称不能重复
  142. LambdaQueryWrapper<CustomField> queryWrapper = new LambdaQueryWrapper<>();
  143. queryWrapper.eq(CustomField::getName, name);
  144. queryWrapper.eq(CustomField::getProjectId, projectId);
  145. List<CustomField> customFields = this.list(queryWrapper);
  146. if (customFields != null && customFields.size() != 0) {
  147. throw new XiaoShiException("名称不能重复");
  148. }
  149. //获取登录人信息
  150. PersonnelVO personnelVO = new PersonnelVO();
  151. try {
  152. personnelVO = cacheUtils.getLoginUser(loginUtils.getId());
  153. } catch (Exception e) {
  154. throw new UnLoginException("未登录");
  155. }
  156. CustomField customField = new CustomField();
  157. BeanUtils.copyProperties(customFieldDTO, customField);
  158. customField.setCreateId(personnelVO.getId());
  159. customField.insert();
  160. //TODO 删除缓存中保存的栏位信息
  161. return customField.getId();
  162. }
  163. @Override
  164. public Object updateMessage(Object object) {
  165. if (object == null) {
  166. throw new XiaoShiException("参数不能为空");
  167. }
  168. //object to updateCustomFieldDTO
  169. UpdateCustomFieldDTO updateCustomFieldDTO = (UpdateCustomFieldDTO) object;
  170. //非空校验
  171. if (updateCustomFieldDTO.getProjectId().equals(null)) {
  172. throw new XiaoShiException("专题库/报告id不能为空!");
  173. }
  174. if (updateCustomFieldDTO.getName().equals(null) || updateCustomFieldDTO.getName().equals("")) {
  175. throw new XiaoShiException("自定义栏位名称不能为空!");
  176. }
  177. if (updateCustomFieldDTO.getType().equals(null)) {
  178. throw new XiaoShiException("自定义栏位类型不能为空!");
  179. }
  180. String name = updateCustomFieldDTO.getName();
  181. Integer projectId = updateCustomFieldDTO.getProjectId();
  182. CustomField customField = this.getById(updateCustomFieldDTO.getId());
  183. //检查名称是否规范
  184. updateCustomFieldDTO.setName(name.trim());
  185. //同一个专题库/报告id下名称不能重复
  186. LambdaQueryWrapper<CustomField> queryWrapper = new LambdaQueryWrapper<>();
  187. queryWrapper.eq(CustomField::getName, name);
  188. queryWrapper.eq(CustomField::getProjectId, projectId);
  189. List<CustomField> customFields = this.list(queryWrapper);
  190. if (!name.equals(customField.getName()) && customFields.size() != 0) {
  191. throw new XiaoShiException("名称重复");
  192. }
  193. //TODO 删除缓存中保存的栏位信息
  194. BeanUtils.copyProperties(updateCustomFieldDTO, customField);
  195. customField.updateById();
  196. return customField.getId();
  197. }
  198. /**
  199. * 装载自定义栏位
  200. *
  201. * @param customFieldVOS
  202. */
  203. private void loadCustomField(List<CustomFieldVO> customFieldVOS) throws IOException {
  204. List<String> createIds = new ArrayList<>();
  205. customFieldVOS.forEach(
  206. item -> {
  207. if (item.getCreateId() != null) {
  208. createIds.add(item.getCreateId());
  209. }
  210. }
  211. );
  212. List<Personnel> personnels = new ArrayList<>();
  213. //查询创建人名称
  214. if (createIds.size() != 0) {
  215. String res = permissionService.getPersonnelByIdsFromPCS(createIds);
  216. JSONObject jsonObject = JSONObject.parseObject(res);
  217. personnels = JSONObject.parseArray(jsonObject.getString("data"), Personnel.class);
  218. }
  219. for (CustomFieldVO customFieldVO : customFieldVOS) {
  220. //装载人员信息
  221. Personnel personnel = personnels.stream().filter(item -> item.getId().equals(customFieldVO.getCreateId())).findFirst().orElse(null);
  222. if (personnel != null) {
  223. customFieldVO.setCreateName(personnel.getPersonnelName());
  224. } else {
  225. throw new XiaoShiException("未获取到当前登陆人信息");
  226. }
  227. }
  228. }
  229. public List<AllCustomFieldVO> getAllProjectCustomField(Integer projectId) {
  230. Integer order = 0;
  231. List<AllCustomFieldVO> allCustomFieldVOS = cacheUtils.getPatentCustomField(projectId);
  232. if (CollectionUtils.isEmpty(allCustomFieldVOS)) {
  233. allCustomFieldVOS = new ArrayList<>();
  234. //根据 project获得栏位值
  235. List<CustomField> customFields = new ArrayList<>();
  236. LambdaQueryWrapper<CustomField> queryWrapper = new LambdaQueryWrapper<>();
  237. queryWrapper.eq(CustomField::getProjectId, projectId);
  238. customFields = this.list(queryWrapper);
  239. //
  240. LambdaQueryWrapper<AssoProjectTreeNode> assoQueryWrapper = new LambdaQueryWrapper<>();
  241. assoQueryWrapper.eq(AssoProjectTreeNode::getProjectId, projectId);
  242. List<AssoProjectTreeNode> projectTreeNodes = assoProjectTreeNodeService.list(assoQueryWrapper);
  243. for (CustomField item : customFields) {
  244. AllCustomFieldVO allCustomFieldVO = new AllCustomFieldVO();
  245. allCustomFieldVO.setId(item.getId());
  246. allCustomFieldVO.setName(item.getName());
  247. allCustomFieldVO.setType(item.getType());
  248. allCustomFieldVO.setSysOrder(order);
  249. allCustomFieldVOS.add(allCustomFieldVO);
  250. order += 1;
  251. }
  252. for (AssoProjectTreeNode item : projectTreeNodes) {
  253. //根据类型和id
  254. AllCustomFieldVO allCustomFieldVO = new AllCustomFieldVO();
  255. allCustomFieldVO.setId(item.getId());
  256. allCustomFieldVO.setType(item.getTreeType());
  257. allCustomFieldVO.setSysOrder(order);
  258. switch (item.getTreeType()) {
  259. case 7:
  260. Product product = productService.getById(item.getId());
  261. allCustomFieldVO.setName(product.getName());
  262. break;
  263. case 8:
  264. ProductCategory productCategory = productCategoryService.getById(item.getId());
  265. allCustomFieldVO.setName(productCategory.getName());
  266. break;
  267. case 9:
  268. //TODO 技术类别
  269. break;
  270. }
  271. allCustomFieldVOS.add(allCustomFieldVO);
  272. order += 1;
  273. }
  274. cacheUtils.addPatentCustomField(projectId, allCustomFieldVOS);
  275. }
  276. return allCustomFieldVOS;
  277. }
  278. /**
  279. * @param taskId
  280. * @return
  281. */
  282. public List<AllCustomFieldVO> getAllTaskCustomField(Integer taskId) {
  283. Integer order = 0;
  284. List<AllCustomFieldVO> allCustomFieldVOS = cacheUtils.getTaskCustomField(taskId);
  285. if (allCustomFieldVOS == null || allCustomFieldVOS.size() == 0) {
  286. allCustomFieldVOS = new ArrayList<>();
  287. //根据任务id查询
  288. List<AssoTaskField> assoTaskFields = new ArrayList<>();
  289. LambdaQueryWrapper<AssoTaskField> queryWrapper = new LambdaQueryWrapper<>();
  290. queryWrapper.eq(AssoTaskField::getTaskId, taskId);
  291. assoTaskFields = assoTaskFieldService.list(queryWrapper);
  292. //
  293. if (assoTaskFields != null) {
  294. //从自定义栏位表中获得自定义栏位id
  295. List<Integer> fieldIds = assoTaskFields.stream()
  296. .filter(item -> item.getFieldType().equals(0)
  297. || item.getFieldType().equals(1)
  298. || item.getFieldType().equals(2)
  299. || item.getFieldType().equals(3)
  300. || item.getFieldType().equals(4)
  301. || item.getFieldType().equals(5)
  302. || item.getFieldType().equals(6)).map(AssoTaskField::getFieldId).collect(Collectors.toList());
  303. if (fieldIds.size() != 0) {
  304. LambdaQueryWrapper<CustomField> queryCWrapper = new LambdaQueryWrapper<>();
  305. queryCWrapper.in(CustomField::getId, fieldIds);
  306. List<CustomField> customFields = this.list(queryCWrapper);
  307. for (CustomField item : customFields) {
  308. AllCustomFieldVO allCustomFieldVO = new AllCustomFieldVO();
  309. allCustomFieldVO.setId(item.getId());
  310. allCustomFieldVO.setName(item.getName());
  311. allCustomFieldVO.setType(item.getType());
  312. allCustomFieldVO.setSysOrder(order);
  313. allCustomFieldVOS.add(allCustomFieldVO);
  314. order += 1;
  315. }
  316. }
  317. //从关联表中获得产品id
  318. List<Integer> productIds = assoTaskFields.stream()
  319. .filter(item -> item.getFieldId().equals(7))
  320. .map(AssoTaskField::getFieldId).collect(Collectors.toList());
  321. if (productIds != null && productIds.size() != 0) {
  322. LambdaQueryWrapper<Product> productLambdaQueryWrapper = new LambdaQueryWrapper<>();
  323. List<Product> products = productService.list(productLambdaQueryWrapper);
  324. for (Product item : products) {
  325. AllCustomFieldVO allCustomFieldVO = new AllCustomFieldVO();
  326. allCustomFieldVO.setId(item.getId());
  327. allCustomFieldVO.setName(item.getName());
  328. allCustomFieldVO.setType(7);
  329. allCustomFieldVO.setSysOrder(order);
  330. allCustomFieldVOS.add(allCustomFieldVO);
  331. order += 1;
  332. }
  333. }
  334. List<Integer> productCategoryIds = assoTaskFields.stream()
  335. .filter(item -> item.getFieldId().equals(8))
  336. .map(AssoTaskField::getFieldId).collect(Collectors.toList());
  337. if (productCategoryIds != null && productCategoryIds.size() != 0) {
  338. LambdaQueryWrapper<ProductCategory> LambdaQueryWrapper = new LambdaQueryWrapper<>();
  339. List<ProductCategory> productCategories = productCategoryService.list(LambdaQueryWrapper);
  340. for (ProductCategory item : productCategories) {
  341. AllCustomFieldVO allCustomFieldVO = new AllCustomFieldVO();
  342. allCustomFieldVO.setId(item.getId());
  343. allCustomFieldVO.setName(item.getName());
  344. allCustomFieldVO.setType(8);
  345. allCustomFieldVO.setSysOrder(order);
  346. allCustomFieldVOS.add(allCustomFieldVO);
  347. order += 1;
  348. }
  349. }
  350. //TODO 技术类别
  351. // List<Integer> technicalIds = assoTaskFields.stream()
  352. // .filter(item -> item.getFieldId().equals(9))
  353. // .map(AssoTaskField::getFieldId).collect(Collectors.toList());
  354. }
  355. cacheUtils.addTaskCustomField(taskId, allCustomFieldVOS);
  356. }
  357. allCustomFieldVOS = allCustomFieldVOS.stream().sorted(Comparator.comparing(AllCustomFieldVO::getSysOrder)).collect(Collectors.toList());
  358. return allCustomFieldVOS;
  359. }
  360. /**
  361. * 查询栏位选项
  362. *
  363. * @param queryEsCustomFieldDTO
  364. * @return
  365. * @throws Exception
  366. */
  367. public Records getCustomFieldValues(QueryEsCustomFieldDTO queryEsCustomFieldDTO) throws Exception {
  368. List<FieldValueVO> fieldValueVOS = new ArrayList<>();
  369. Integer type = queryEsCustomFieldDTO.getFieldType();
  370. EsCustomFieldDTO esCustomFieldDTO = new EsCustomFieldDTO();
  371. esCustomFieldDTO.setFieldId(queryEsCustomFieldDTO.getFieldId());
  372. esCustomFieldDTO.setProjectId((queryEsCustomFieldDTO.getProjectId()));
  373. esCustomFieldDTO.setTaskId(queryEsCustomFieldDTO.getTaskId());
  374. esCustomFieldDTO.setPatentNo(queryEsCustomFieldDTO.getPatentNo());
  375. esCustomFieldDTO.setFieldType(queryEsCustomFieldDTO.getFieldType());
  376. PatentWithIdVO patentVO = esCustomFieldService.getEsCustomField(esCustomFieldDTO);
  377. if (patentVO != null) {
  378. List<String> value = patentVO.getPatent().getESCustomField().getFieldValue();
  379. switch (type) {
  380. case 0:
  381. case 1:
  382. case 2:
  383. value.forEach(item -> {
  384. FieldValueVO fieldValueVO = new FieldValueVO();
  385. fieldValueVO.setValue(item);
  386. fieldValueVOS.add(fieldValueVO);
  387. });
  388. break;
  389. case 3:
  390. case 4:
  391. case 5:
  392. if (value != null && value.size() > 0) {
  393. List<Integer> ids = FormatUtil.StringTOIntegerList(value);
  394. LambdaQueryWrapper<CustomOption> queryWrapper = new LambdaQueryWrapper<>();
  395. queryWrapper.in(CustomOption::getId, ids);
  396. List<CustomOption> customOptions = customOptionService.list(queryWrapper);
  397. customOptions.forEach(item -> {
  398. FieldValueVO fieldValueVO = new FieldValueVO();
  399. fieldValueVO.setValueId(item.getId());
  400. fieldValueVO.setValue(item.getName());
  401. fieldValueVOS.add(fieldValueVO);
  402. });
  403. }
  404. break;
  405. case 6:
  406. case 7:
  407. case 8:
  408. case 9:
  409. if (value != null && value.size() > 0) {
  410. List<Integer> ids = FormatUtil.StringTOIntegerList(value);
  411. LambdaQueryWrapper<TreeNode> queryWrapper = new LambdaQueryWrapper<>();
  412. queryWrapper.in(TreeNode::getId, ids);
  413. List<TreeNode> treeNodes = treeNodeService.list(queryWrapper);
  414. treeNodes.forEach(item -> {
  415. FieldValueVO fieldValueVO = new FieldValueVO();
  416. fieldValueVO.setValueId(item.getId());
  417. fieldValueVO.setPath(item.getPath());
  418. List<String> idStrs = new ArrayList<>();
  419. List<String> pathNames =new ArrayList<>();
  420. if (item.getPath() != null && !item.getPath().equals("")) {
  421. idStrs = Arrays.asList(item.getPath().split("/"));
  422. pathNames= treeNodeService.getBaseMapper().getTreeFieldPath(idStrs);
  423. }
  424. pathNames.add(item.getName());
  425. String pathValue= StringUtils.join(pathNames,"/");
  426. fieldValueVO.setPathValue(pathValue);
  427. fieldValueVO.setValue(pathValue);
  428. fieldValueVOS.add(fieldValueVO);
  429. });
  430. }
  431. break;
  432. }
  433. }
  434. Records records = new Records();
  435. records.setData(fieldValueVOS);
  436. return records;
  437. }
  438. public Records addAllPatentCustomFieldOrder(AllFieldOrderDTO allFieldOrderDTO) {
  439. Integer projectId = allFieldOrderDTO.getProjectId();
  440. List<AllCustomFieldVO> allCustomFieldVOS = allFieldOrderDTO.getAllCustomFieldVOs();
  441. if (allFieldOrderDTO.getTaskId() != null) {
  442. cacheUtils.addTaskCustomField(allFieldOrderDTO.getTaskId(), allCustomFieldVOS);
  443. } else {
  444. cacheUtils.addPatentCustomField(projectId, allCustomFieldVOS);
  445. }
  446. Records records = new Records();
  447. records.setData(allCustomFieldVOS);
  448. return records;
  449. }
  450. /**
  451. * 根据名称获得标引栏位以及标引值
  452. *
  453. * @param name
  454. * @param projectId
  455. */
  456. public List<EsCustomFieldValueDTO> getCustomFieldByValueName(String name, Integer projectId) {
  457. List<EsCustomFieldValueDTO> esCustomFieldDTOS = new ArrayList<>();
  458. //根据名称获得标引栏位以及标引值
  459. LambdaQueryWrapper<CustomField> queryWrapper = new LambdaQueryWrapper<>();
  460. queryWrapper.in(CustomField::getType, Arrays.asList(4, 5, 6))
  461. .eq(CustomField::getProjectId, projectId);
  462. List<CustomField> customFields = this.list(queryWrapper);
  463. List<Integer> ids = customFields.stream().map(CustomField::getId).collect(Collectors.toList());
  464. if (ids.size() == 0) {
  465. return esCustomFieldDTOS;
  466. }
  467. //从option 查询名称相同的选项
  468. LambdaQueryWrapper<CustomOption> optionWrapper = new LambdaQueryWrapper<>();
  469. optionWrapper.eq(CustomOption::getName, name)
  470. .in(CustomOption::getCustomFieldId, ids);
  471. List<CustomOption> options = customOptionService.list(optionWrapper);
  472. // 从 treeNode 查询名称相同的节点
  473. LambdaQueryWrapper<TreeNode> treeNodeWrapper = new LambdaQueryWrapper<>();
  474. treeNodeWrapper.eq(TreeNode::getName, name)
  475. .eq(TreeNode::getType, 4)
  476. .in(TreeNode::getTypeId, ids);
  477. List<TreeNode> treeNodes = treeNodeService.list(treeNodeWrapper);
  478. if (options != null && options.size() != 0) {
  479. options.forEach(item -> {
  480. EsCustomFieldValueDTO esCustomFieldDTO = esCustomFieldDTOS.stream()
  481. .filter(i -> i.getFieldId()
  482. .equals(item.getCustomFieldId()))
  483. .findFirst().orElse(null);
  484. if (esCustomFieldDTO == null) {
  485. esCustomFieldDTO.setFieldId(item.getCustomFieldId().toString());
  486. List<String> values = new ArrayList<>();
  487. values.add(item.getId().toString());
  488. esCustomFieldDTO.setFieldValue(values);
  489. } else {
  490. esCustomFieldDTO.getFieldValue().add(item.getId().toString());
  491. }
  492. });
  493. }
  494. if (treeNodes != null && treeNodes.size() != 0) {
  495. treeNodes.forEach(item -> {
  496. EsCustomFieldValueDTO esCustomFieldDTO = esCustomFieldDTOS.stream()
  497. .filter(i -> i.getFieldId()
  498. .equals(item.getTypeId()))
  499. .findFirst().orElse(null);
  500. if (esCustomFieldDTO == null) {
  501. esCustomFieldDTO.setFieldId(item.getTypeId().toString());
  502. List<String> values = new ArrayList<>();
  503. values.add(item.getId().toString());
  504. esCustomFieldDTO.setFieldValue(values);
  505. } else {
  506. esCustomFieldDTO.getFieldValue().add(item.getId().toString());
  507. }
  508. });
  509. }
  510. return esCustomFieldDTOS;
  511. }
  512. public List<EsCustomFieldDTO> getCustomFieldDTOs(List<UploadParamsVO.Field> customerFieldList, Integer projectId, Integer taskId) {
  513. List<EsCustomFieldDTO> esCustomFieldDTOS = new ArrayList<>();
  514. for (UploadParamsVO.Field field : customerFieldList) {
  515. String key = field.getKey();
  516. List<String> fields = StringUtils.changeStringToString(key, ":");
  517. if (fields.size() == 2) {
  518. String name = fields.get(0);
  519. Integer fieldId = null;
  520. List<String> value = field.getFieldList();
  521. Integer type = Integer.parseInt(fields.get(1));
  522. List<String> select = field.getFieldList();
  523. if (StringUtils.isNull(select)) {
  524. return esCustomFieldDTOS;
  525. }
  526. switch (type) {
  527. case 0:
  528. ;
  529. case 1:
  530. ;
  531. case 2:
  532. ;
  533. case 3:
  534. fieldId = this.GetIdByName(name, projectId, type);
  535. break;
  536. case 4:
  537. case 5:
  538. fieldId = this.GetIdByName(name, projectId, type);
  539. value = customOptionService.getIdsByNames(value, fieldId);
  540. break;
  541. case 6:
  542. //TODO 根据名称和类型查询
  543. fieldId = this.GetIdByName(name, projectId, type);
  544. value = treeNodeService.getIdByNames(value, fieldId, type);
  545. ;
  546. }
  547. EsCustomFieldDTO esCustomFieldDTO = new EsCustomFieldDTO();
  548. esCustomFieldDTO.setFieldType(type);
  549. esCustomFieldDTO.setFieldId(fieldId.toString());
  550. esCustomFieldDTO.setFieldValue(value);
  551. esCustomFieldDTO.setProjectId(projectId);
  552. esCustomFieldDTO.setTaskId(taskId);
  553. esCustomFieldDTOS.add(esCustomFieldDTO);
  554. }
  555. }
  556. return esCustomFieldDTOS;
  557. }
  558. /**
  559. * 根据名称获得自定义字段id
  560. *
  561. * @param name
  562. * @param projectId
  563. * @param type
  564. * @return
  565. */
  566. public Integer GetIdByName(String name, Integer projectId, Integer type) {
  567. LambdaQueryWrapper<CustomField> queryWrapper = new LambdaQueryWrapper<>();
  568. queryWrapper.eq(CustomField::getName, name)
  569. .eq(CustomField::getProjectId, projectId)
  570. .eq(CustomField::getType, type);
  571. CustomField customField = this.getOne(queryWrapper);
  572. if (customField != null) {
  573. return customField.getId();
  574. }
  575. PersonnelVO personnelVO = cacheUtils.getLoginUser(loginUtils.getId());
  576. CustomField customField1 = new CustomField();
  577. customField1.setName(name);
  578. customField1.setProjectId(projectId);
  579. customField1.setType(type);
  580. customField1.setCreateId(personnelVO.getId());
  581. customField1.insert();
  582. return customField1.getId();
  583. }
  584. public void copyCustomField(Integer oldProject, Integer newProject) {
  585. LambdaQueryWrapper<CustomField> queryWrapper = new LambdaQueryWrapper<>();
  586. queryWrapper.eq(CustomField::getProjectId, oldProject);
  587. List<CustomField> fields = this.list(queryWrapper);
  588. if (fields.size() != 0) {
  589. for (CustomField customField : fields) {
  590. CustomField newCustomField = new CustomField();
  591. BeanUtils.copyProperties(fields, newCustomField);
  592. newCustomField.setId(null);
  593. newCustomField.setProjectId(newProject);
  594. newCustomField.insert();
  595. this.copyCustomFieldValues(customField, newCustomField);
  596. }
  597. this.saveBatch(fields);
  598. }
  599. }
  600. public void copyCustomFieldValues(CustomField customField, CustomField newCustomField) {
  601. Integer type = customField.getType();
  602. switch (type) {
  603. case 0:
  604. ;
  605. case 1:
  606. ;
  607. case 2:
  608. ;
  609. case 3:
  610. this.copyTextValues(customField, newCustomField);
  611. break;
  612. case 4:
  613. case 5:
  614. this.copyOptionValues(customField, newCustomField);
  615. break;
  616. case 6:
  617. this.copyTreeValues(customField, newCustomField);
  618. break;
  619. }
  620. }
  621. public void copyTextValues(CustomField customField, CustomField newCustomField) {
  622. List<PatentWithIdVO> patents = esCustomFieldService.getEsCustomFields(customField.getId().toString(), customField.getType(), customField.getProjectId());
  623. for (PatentWithIdVO patentWithIdVO : patents) {
  624. Patent patent = patentWithIdVO.getPatent();
  625. patent.getESCustomField().setField(newCustomField.getId().toString());
  626. patent.getESCustomField().setCreateTime(new Date());
  627. patent.getESCustomField().setTaskId(null);
  628. patent.getESCustomField().setProjectId(newCustomField.getProjectId());
  629. String parentId = patent.getPatentJoin().getParent();
  630. try {
  631. esService.addChildPatent(patent, parentId);
  632. } catch (Exception e) {
  633. throw new XiaoShiException("保存出错");
  634. }
  635. }
  636. }
  637. public void copyOptionValues(CustomField customField, CustomField newCustomField) {
  638. Integer newId = newCustomField.getId();
  639. //查询自定义栏位的选项值
  640. Map<String, String> map = new HashMap<>();
  641. LambdaQueryWrapper<CustomOption> queryWrapper = new LambdaQueryWrapper<>();
  642. queryWrapper.eq(CustomOption::getCustomFieldId, customField.getId());
  643. List<CustomOption> customOptions = customOptionService.list(queryWrapper);
  644. for (CustomOption customOption : customOptions) {
  645. CustomOption customOption1 = new CustomOption();
  646. BeanUtils.copyProperties(customOption, customOption1);
  647. customOption1.setCustomFieldId(newId);
  648. customOption1.insert();
  649. map.put(customOption.getId().toString(), customOption1.getId().toString());
  650. }
  651. List<PatentWithIdVO> patents = esCustomFieldService.getEsCustomFields(customField.getId().toString(), customField.getType(), customField.getProjectId());
  652. for (PatentWithIdVO patentWithIdVO : patents) {
  653. Patent patent = patentWithIdVO.getPatent();
  654. patent.getESCustomField().setField(customField.getId().toString());
  655. patent.getESCustomField().setCreateTime(new Date());
  656. patent.getESCustomField().setTaskId(null);
  657. List<String> fieldValues = patent.getESCustomField().getFieldValue();
  658. fieldValues.forEach(item -> item = map.get(item));
  659. patent.getESCustomField().setFieldValue(fieldValues);
  660. patent.getESCustomField().setStatsValue(fieldValues);
  661. String parentId = patent.getPatentJoin().getParent();
  662. try {
  663. esService.addChildPatent(patent, parentId);
  664. } catch (Exception e) {
  665. throw new XiaoShiException("保存出错");
  666. }
  667. }
  668. }
  669. public void copyTreeValues(CustomField customField, CustomField newCustomField) {
  670. Integer newId = newCustomField.getId();
  671. Map<String, String> map = new HashMap<>();
  672. //查询自定义栏位的选项值
  673. LambdaQueryWrapper<TreeNode> queryWrapper = new LambdaQueryWrapper<>();
  674. queryWrapper.eq(TreeNode::getTypeId, customField.getId())
  675. .eq(TreeNode::getType, customField.getType())
  676. .orderByAsc(TreeNode::getLevel);
  677. List<TreeNode> treeNodes = treeNodeService.list(queryWrapper);
  678. for (TreeNode treeNode : treeNodes) {
  679. TreeNode treeNode1 = new TreeNode();
  680. BeanUtils.copyProperties(treeNode, treeNode1);
  681. treeNode1.setTypeId(newCustomField.getId());
  682. treeNode1.setId(null);
  683. String path = treeNode.getPath();
  684. if (path != null && !path.trim().equals("")) {
  685. List<String> paths = Arrays.asList(path.split("/"));
  686. paths.forEach(item -> item = map.get(item));
  687. path = StringUtils.join(paths, "/");
  688. treeNode1.setPath(path);
  689. }
  690. treeNode1.insert();
  691. map.put(treeNode.getId().toString(), treeNode1.getId().toString());
  692. }
  693. List<PatentWithIdVO> patents = esCustomFieldService.getEsCustomFields(customField.getId().toString(), customField.getType(), customField.getProjectId());
  694. for (PatentWithIdVO patentWithIdVO : patents) {
  695. Patent patent = patentWithIdVO.getPatent();
  696. patent.getESCustomField().setField(customField.getId().toString());
  697. patent.getESCustomField().setCreateTime(new Date());
  698. patent.getESCustomField().setTaskId(null);
  699. List<String> fieldValues = patent.getESCustomField().getFieldValue();
  700. List<String> statusValues = patent.getESCustomField().getStatsValue();
  701. fieldValues.forEach(item -> item = map.get(item));
  702. statusValues.forEach(item -> item = map.get(item));
  703. patent.getESCustomField().setFieldValue(fieldValues);
  704. patent.getESCustomField().setStatsValue(statusValues);
  705. String parentId = patent.getPatentJoin().getParent();
  706. //TODO添加path
  707. try {
  708. esService.addChildPatent(patent, parentId);
  709. } catch (Exception e) {
  710. throw new XiaoShiException("保存出错");
  711. }
  712. }
  713. }
  714. // public void copyProductValues(CustomField customField,CustomField newCustomField) {
  715. // Integer newId =newCustomField.getId();
  716. // //查询自定义栏位的选项值
  717. // List<Map<String,String>> maps =new ArrayList<>();
  718. // LambdaQueryWrapper<TreeNode> queryWrapper =new LambdaQueryWrapper<>();
  719. // queryWrapper.eq(TreeNode::getTypeId,customField.getId())
  720. // .eq(TreeNode::getType,customField.getType())
  721. // .orderByAsc(TreeNode::getLevel);
  722. // List<TreeNode> treeNodes = treeNodeService.list(queryWrapper);
  723. // for (TreeNode treeNode:treeNodes){
  724. // TreeNode treeNode1 =new TreeNode();
  725. // BeanUtils.copyProperties(treeNode,treeNode1);
  726. // treeNode1.setTypeId(newCustomField.getId());
  727. // treeNode1.setId(null);
  728. // //TODO添加path
  729. // Map<String,String> map =new HashMap<>();
  730. // map.put(treeNode.getId().toString(),treeNode1.getId().toString());
  731. //
  732. // }
  733. // List<PatentWithIdVO> patents = esCustomFieldService.getEsCustomFields(customField.getId().toString(), customField.getType(), customField.getProjectId());
  734. // for (PatentWithIdVO patentWithIdVO : patents) {
  735. // Patent patent =patentWithIdVO.getPatent();
  736. // patent.getESCustomField().setField(customField.getId().toString());
  737. // patent.getESCustomField().setCreateTime(new Date());
  738. // patent.getESCustomField().setTaskId(null);
  739. // String parentId =patent.getPatentJoin().getParent();
  740. // //TODO添加path
  741. // try {
  742. // esService.addChildPatent(patent,parentId);
  743. // }
  744. // catch (Exception e){
  745. // throw new XiaoShiException("保存出错");
  746. // }
  747. //
  748. // }
  749. //
  750. // }
  751. }