CustomFieldService.java 37 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869
  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. }
  225. }
  226. }
  227. public List<AllCustomFieldVO> getAllProjectCustomField(Integer projectId) {
  228. List<AllCustomFieldVO> allCustomFieldVOS = cacheUtils.getPatentCustomField(projectId);
  229. if (CollectionUtils.isEmpty(allCustomFieldVOS)) {
  230. allCustomFieldVOS = new ArrayList<>();
  231. }
  232. Integer order = allCustomFieldVOS.size();
  233. //根据 project获得栏位值
  234. List<CustomField> customFields = new ArrayList<>();
  235. LambdaQueryWrapper<CustomField> queryWrapper = new LambdaQueryWrapper<>();
  236. queryWrapper.eq(CustomField::getProjectId, projectId);
  237. customFields = this.list(queryWrapper);
  238. //
  239. LambdaQueryWrapper<AssoProjectTreeNode> assoQueryWrapper = new LambdaQueryWrapper<>();
  240. assoQueryWrapper.eq(AssoProjectTreeNode::getProjectId, projectId);
  241. List<AssoProjectTreeNode> projectTreeNodes = assoProjectTreeNodeService.list(assoQueryWrapper);
  242. for (CustomField item : customFields) {
  243. AllCustomFieldVO allCustomFieldVO = allCustomFieldVOS.stream().filter(i -> i.getId().equals(item.getId()) && i.getType().equals(item.getType())).findFirst().orElse(null);
  244. if (allCustomFieldVO == null) {
  245. allCustomFieldVO = new AllCustomFieldVO();
  246. allCustomFieldVO.setId(item.getId());
  247. allCustomFieldVO.setName(item.getName());
  248. allCustomFieldVO.setType(item.getType());
  249. allCustomFieldVO.setSysOrder(order);
  250. allCustomFieldVOS.add(allCustomFieldVO);
  251. order += 1;
  252. }
  253. }
  254. for (AssoProjectTreeNode item : projectTreeNodes) {
  255. //根据类型和id
  256. AllCustomFieldVO allCustomFieldVO = allCustomFieldVOS.stream().filter(i -> i.getId().equals(item.getId()) && i.getType().equals(item.getTreeType())).findFirst().orElse(null);
  257. if (allCustomFieldVO == null) {
  258. allCustomFieldVO = new AllCustomFieldVO();
  259. allCustomFieldVO.setId(item.getId());
  260. allCustomFieldVO.setType(item.getTreeType());
  261. allCustomFieldVO.setSysOrder(order);
  262. switch (item.getTreeType()) {
  263. case 2:
  264. Product product = productService.getById(item.getFieldId());
  265. allCustomFieldVO.setName(product.getName());
  266. break;
  267. case 3:
  268. ProductCategory productCategory = productCategoryService.getById(item.getFieldId());
  269. allCustomFieldVO.setName(productCategory.getName());
  270. break;
  271. case 1:
  272. //TODO 技术类别
  273. break;
  274. }
  275. allCustomFieldVOS.add(allCustomFieldVO);
  276. order += 1;
  277. }
  278. }
  279. return allCustomFieldVOS;
  280. }
  281. /**
  282. * @param taskId
  283. * @return
  284. */
  285. public List<AllCustomFieldVO> getAllTaskCustomField(Integer taskId) {
  286. Integer order = 0;
  287. List<AllCustomFieldVO> allCustomFieldVOS = cacheUtils.getTaskCustomField(taskId);
  288. if (allCustomFieldVOS == null || allCustomFieldVOS.size() == 0) {
  289. allCustomFieldVOS = new ArrayList<>();
  290. //根据任务id查询
  291. List<AssoTaskField> assoTaskFields = new ArrayList<>();
  292. LambdaQueryWrapper<AssoTaskField> queryWrapper = new LambdaQueryWrapper<>();
  293. queryWrapper.eq(AssoTaskField::getTaskId, taskId);
  294. assoTaskFields = assoTaskFieldService.list(queryWrapper);
  295. //
  296. if (assoTaskFields != null) {
  297. //从自定义栏位表中获得自定义栏位id
  298. List<Integer> fieldIds = assoTaskFields.stream()
  299. .filter(item -> item.getFieldType().equals(0)
  300. || item.getFieldType().equals(1)
  301. || item.getFieldType().equals(2)
  302. || item.getFieldType().equals(3)
  303. || item.getFieldType().equals(4)
  304. || item.getFieldType().equals(5)
  305. || item.getFieldType().equals(6)).map(AssoTaskField::getFieldId).collect(Collectors.toList());
  306. if (fieldIds.size() != 0) {
  307. LambdaQueryWrapper<CustomField> queryCWrapper = new LambdaQueryWrapper<>();
  308. queryCWrapper.in(CustomField::getId, fieldIds);
  309. List<CustomField> customFields = this.list(queryCWrapper);
  310. for (CustomField item : customFields) {
  311. AllCustomFieldVO allCustomFieldVO = new AllCustomFieldVO();
  312. allCustomFieldVO.setId(item.getId());
  313. allCustomFieldVO.setName(item.getName());
  314. allCustomFieldVO.setType(item.getType());
  315. allCustomFieldVO.setSysOrder(order);
  316. allCustomFieldVOS.add(allCustomFieldVO);
  317. order += 1;
  318. }
  319. }
  320. //从关联表中获得产品id
  321. List<Integer> productIds = assoTaskFields.stream()
  322. .filter(item -> item.getFieldId().equals(7))
  323. .map(AssoTaskField::getFieldId).collect(Collectors.toList());
  324. if (productIds != null && productIds.size() != 0) {
  325. LambdaQueryWrapper<Product> productLambdaQueryWrapper = new LambdaQueryWrapper<>();
  326. List<Product> products = productService.list(productLambdaQueryWrapper);
  327. for (Product item : products) {
  328. AllCustomFieldVO allCustomFieldVO = new AllCustomFieldVO();
  329. allCustomFieldVO.setId(item.getId());
  330. allCustomFieldVO.setName(item.getName());
  331. allCustomFieldVO.setType(7);
  332. allCustomFieldVO.setSysOrder(order);
  333. allCustomFieldVOS.add(allCustomFieldVO);
  334. order += 1;
  335. }
  336. }
  337. List<Integer> productCategoryIds = assoTaskFields.stream()
  338. .filter(item -> item.getFieldId().equals(8))
  339. .map(AssoTaskField::getFieldId).collect(Collectors.toList());
  340. if (productCategoryIds != null && productCategoryIds.size() != 0) {
  341. LambdaQueryWrapper<ProductCategory> LambdaQueryWrapper = new LambdaQueryWrapper<>();
  342. List<ProductCategory> productCategories = productCategoryService.list(LambdaQueryWrapper);
  343. for (ProductCategory item : productCategories) {
  344. AllCustomFieldVO allCustomFieldVO = new AllCustomFieldVO();
  345. allCustomFieldVO.setId(item.getId());
  346. allCustomFieldVO.setName(item.getName());
  347. allCustomFieldVO.setType(8);
  348. allCustomFieldVO.setSysOrder(order);
  349. allCustomFieldVOS.add(allCustomFieldVO);
  350. order += 1;
  351. }
  352. }
  353. //TODO 技术类别
  354. // List<Integer> technicalIds = assoTaskFields.stream()
  355. // .filter(item -> item.getFieldId().equals(9))
  356. // .map(AssoTaskField::getFieldId).collect(Collectors.toList());
  357. }
  358. cacheUtils.addTaskCustomField(taskId, allCustomFieldVOS);
  359. }
  360. allCustomFieldVOS = allCustomFieldVOS.stream().sorted(Comparator.comparing(AllCustomFieldVO::getSysOrder)).collect(Collectors.toList());
  361. return allCustomFieldVOS;
  362. }
  363. /**
  364. * 查询栏位选项
  365. *
  366. * @param queryEsCustomFieldDTO
  367. * @return
  368. * @throws Exception
  369. */
  370. public Records getCustomFieldValues(QueryEsCustomFieldDTO queryEsCustomFieldDTO) throws Exception {
  371. List<FieldValueVO> fieldValueVOS = new ArrayList<>();
  372. Integer type = queryEsCustomFieldDTO.getFieldType();
  373. EsCustomFieldDTO esCustomFieldDTO = new EsCustomFieldDTO();
  374. esCustomFieldDTO.setFieldId(queryEsCustomFieldDTO.getFieldId());
  375. esCustomFieldDTO.setProjectId((queryEsCustomFieldDTO.getProjectId()));
  376. esCustomFieldDTO.setTaskId(queryEsCustomFieldDTO.getTaskId());
  377. esCustomFieldDTO.setPatentNo(queryEsCustomFieldDTO.getPatentNo());
  378. esCustomFieldDTO.setFieldType(queryEsCustomFieldDTO.getFieldType());
  379. PatentWithIdVO patentVO = esCustomFieldService.getEsCustomField(esCustomFieldDTO);
  380. if (patentVO != null) {
  381. List<String> value = patentVO.getPatent().getESCustomField().getFieldValue();
  382. switch (type) {
  383. case 0:
  384. case 1:
  385. case 2:
  386. value.forEach(item -> {
  387. FieldValueVO fieldValueVO = new FieldValueVO();
  388. fieldValueVO.setValue(item);
  389. fieldValueVOS.add(fieldValueVO);
  390. });
  391. break;
  392. case 3:
  393. case 4:
  394. case 5:
  395. if (value != null && value.size() > 0) {
  396. List<Integer> ids = FormatUtil.StringTOIntegerList(value);
  397. LambdaQueryWrapper<CustomOption> queryWrapper = new LambdaQueryWrapper<>();
  398. queryWrapper.in(CustomOption::getId, ids);
  399. List<CustomOption> customOptions = customOptionService.list(queryWrapper);
  400. customOptions.forEach(item -> {
  401. FieldValueVO fieldValueVO = new FieldValueVO();
  402. fieldValueVO.setValueId(item.getId());
  403. fieldValueVO.setValue(item.getName());
  404. fieldValueVOS.add(fieldValueVO);
  405. });
  406. }
  407. break;
  408. case 6:
  409. case 7:
  410. case 8:
  411. case 9:
  412. if (value != null && value.size() > 0) {
  413. List<Integer> ids = FormatUtil.StringTOIntegerList(value);
  414. LambdaQueryWrapper<TreeNode> queryWrapper = new LambdaQueryWrapper<>();
  415. queryWrapper.in(TreeNode::getId, ids);
  416. List<TreeNode> treeNodes = treeNodeService.list(queryWrapper);
  417. treeNodes.forEach(item -> {
  418. FieldValueVO fieldValueVO = new FieldValueVO();
  419. fieldValueVO.setValueId(item.getId());
  420. fieldValueVO.setPath(item.getPath());
  421. List<String> idStrs = new ArrayList<>();
  422. List<String> pathNames = new ArrayList<>();
  423. if (item.getPath() != null && !item.getPath().equals("")) {
  424. idStrs = Arrays.asList(item.getPath().split("/"));
  425. pathNames = treeNodeService.getBaseMapper().getTreeFieldPath(idStrs);
  426. }
  427. pathNames.add(item.getName());
  428. String pathValue = StringUtils.join(pathNames, "/");
  429. fieldValueVO.setPathValue(pathValue);
  430. fieldValueVO.setValue(pathValue);
  431. fieldValueVOS.add(fieldValueVO);
  432. });
  433. }
  434. break;
  435. }
  436. }
  437. Records records = new Records();
  438. records.setData(fieldValueVOS);
  439. return records;
  440. }
  441. public Records addAllPatentCustomFieldOrder(AllFieldOrderDTO allFieldOrderDTO) {
  442. Integer projectId = allFieldOrderDTO.getProjectId();
  443. List<AllCustomFieldVO> allCustomFieldVOS = allFieldOrderDTO.getAllCustomFieldVOs();
  444. if (allFieldOrderDTO.getTaskId() != null) {
  445. cacheUtils.addTaskCustomField(allFieldOrderDTO.getTaskId(), allCustomFieldVOS);
  446. } else {
  447. cacheUtils.addPatentCustomField(projectId, allCustomFieldVOS);
  448. }
  449. Records records = new Records();
  450. records.setData(allCustomFieldVOS);
  451. return records;
  452. }
  453. /**
  454. * 根据名称获得标引栏位以及标引值
  455. *
  456. * @param name
  457. * @param projectId
  458. */
  459. public List<EsCustomFieldValueDTO> getCustomFieldByValueName(String name, Integer projectId) {
  460. List<EsCustomFieldValueDTO> esCustomFieldDTOS = new ArrayList<>();
  461. //根据名称获得标引栏位以及标引值
  462. LambdaQueryWrapper<CustomField> queryWrapper = new LambdaQueryWrapper<>();
  463. queryWrapper.in(CustomField::getType, Arrays.asList(4, 5))
  464. .eq(CustomField::getProjectId, projectId);
  465. List<CustomField> customFields = this.list(queryWrapper);
  466. List<Integer> ids = customFields.stream().map(CustomField::getId).collect(Collectors.toList());
  467. if (ids.size() == 0) {
  468. return esCustomFieldDTOS;
  469. }
  470. //从option 查询名称相同的选项
  471. LambdaQueryWrapper<CustomOption> optionWrapper = new LambdaQueryWrapper<>();
  472. optionWrapper.eq(CustomOption::getName, name)
  473. .in(CustomOption::getCustomFieldId, ids);
  474. List<CustomOption> options = customOptionService.list(optionWrapper);
  475. if (options != null && options.size() != 0) {
  476. options.forEach(item -> {
  477. EsCustomFieldValueDTO esCustomFieldDTO = esCustomFieldDTOS.stream()
  478. .filter(i -> i.getFieldId()
  479. .equals(item.getCustomFieldId()))
  480. .findFirst().orElse(null);
  481. if (esCustomFieldDTO == null) {
  482. esCustomFieldDTO = new EsCustomFieldValueDTO();
  483. esCustomFieldDTO.setFieldId(item.getCustomFieldId().toString());
  484. List<String> values = new ArrayList<>();
  485. values.add(item.getId().toString());
  486. esCustomFieldDTO.setFieldValue(values);
  487. esCustomFieldDTOS.add(esCustomFieldDTO);
  488. } else {
  489. esCustomFieldDTO.getFieldValue().add(item.getId().toString());
  490. }
  491. });
  492. }
  493. return esCustomFieldDTOS;
  494. }
  495. public List<EsCustomFieldDTO> getCustomFieldDTOs(List<UploadParamsVO.Field> customerFieldList, Integer projectId, Integer taskId) {
  496. List<EsCustomFieldDTO> esCustomFieldDTOS = new ArrayList<>();
  497. for (UploadParamsVO.Field field : customerFieldList) {
  498. String key = field.getKey();
  499. List<String> fields = StringUtils.changeStringToString(key, ":");
  500. if (fields.size() == 2) {
  501. String name = fields.get(0);
  502. Integer fieldId = null;
  503. List<String> value = field.getFieldList();
  504. Integer type = Integer.parseInt(fields.get(1));
  505. List<String> select = field.getFieldList();
  506. if (StringUtils.isNull(select)) {
  507. return esCustomFieldDTOS;
  508. }
  509. switch (type) {
  510. case 0:
  511. ;
  512. case 1:
  513. ;
  514. case 2:
  515. ;
  516. case 3:
  517. fieldId = this.GetIdByName(name, projectId, type);
  518. break;
  519. case 4:
  520. case 5:
  521. fieldId = this.GetIdByName(name, projectId, type);
  522. value = customOptionService.getIdsByNames(value, fieldId);
  523. break;
  524. case 6:
  525. //TODO 根据名称和类型查询
  526. fieldId = this.GetIdByName(name, projectId, type);
  527. value = treeNodeService.getIdByNames(value, fieldId, type);
  528. ;
  529. }
  530. EsCustomFieldDTO esCustomFieldDTO = new EsCustomFieldDTO();
  531. esCustomFieldDTO.setFieldType(type);
  532. esCustomFieldDTO.setFieldId(fieldId.toString());
  533. esCustomFieldDTO.setFieldValue(value);
  534. esCustomFieldDTO.setProjectId(projectId);
  535. esCustomFieldDTO.setTaskId(taskId);
  536. esCustomFieldDTOS.add(esCustomFieldDTO);
  537. }
  538. }
  539. return esCustomFieldDTOS;
  540. }
  541. /**
  542. * 根据名称获得自定义字段id
  543. *
  544. * @param name
  545. * @param projectId
  546. * @param type
  547. * @return
  548. */
  549. public Integer GetIdByName(String name, Integer projectId, Integer type) {
  550. LambdaQueryWrapper<CustomField> queryWrapper = new LambdaQueryWrapper<>();
  551. queryWrapper.eq(CustomField::getName, name)
  552. .eq(CustomField::getProjectId, projectId)
  553. .eq(CustomField::getType, type);
  554. CustomField customField = this.getOne(queryWrapper);
  555. if (customField != null) {
  556. return customField.getId();
  557. }
  558. CustomField customField1 = new CustomField();
  559. customField1.setName(name);
  560. customField1.setProjectId(projectId);
  561. customField1.setType(type);
  562. customField1.insert();
  563. return customField1.getId();
  564. }
  565. public void copyCustomField(Integer oldProject, Integer newProject) {
  566. LambdaQueryWrapper<CustomField> queryWrapper = new LambdaQueryWrapper<>();
  567. queryWrapper.eq(CustomField::getProjectId, oldProject);
  568. List<CustomField> fields = this.list(queryWrapper);
  569. if (fields.size() != 0) {
  570. for (CustomField customField : fields) {
  571. CustomField newCustomField = new CustomField();
  572. BeanUtils.copyProperties(fields, newCustomField);
  573. newCustomField.setId(null);
  574. newCustomField.setProjectId(newProject);
  575. newCustomField.insert();
  576. this.copyCustomFieldValues(customField, newCustomField);
  577. }
  578. this.saveBatch(fields);
  579. }
  580. }
  581. public void copyCustomFieldValues(CustomField customField, CustomField newCustomField) {
  582. Integer type = customField.getType();
  583. switch (type) {
  584. case 0:
  585. ;
  586. case 1:
  587. ;
  588. case 2:
  589. ;
  590. case 3:
  591. this.copyTextValues(customField, newCustomField);
  592. break;
  593. case 4:
  594. case 5:
  595. this.copyOptionValues(customField, newCustomField);
  596. break;
  597. case 6:
  598. this.copyTreeValues(customField, newCustomField);
  599. break;
  600. }
  601. }
  602. public void copyTextValues(CustomField customField, CustomField newCustomField) {
  603. List<PatentWithIdVO> patents = esCustomFieldService.getEsCustomFields(customField.getId().toString(), customField.getType(), customField.getProjectId());
  604. for (PatentWithIdVO patentWithIdVO : patents) {
  605. Patent patent = patentWithIdVO.getPatent();
  606. patent.getESCustomField().setField(newCustomField.getId().toString());
  607. patent.getESCustomField().setCreateTime(new Date());
  608. patent.getESCustomField().setTaskId(null);
  609. patent.getESCustomField().setProjectId(newCustomField.getProjectId());
  610. String parentId = patent.getPatentJoin().getParent();
  611. try {
  612. esService.addChildPatent(patent, parentId);
  613. } catch (Exception e) {
  614. throw new XiaoShiException("保存出错");
  615. }
  616. }
  617. }
  618. public void copyOptionValues(CustomField customField, CustomField newCustomField) {
  619. Integer newId = newCustomField.getId();
  620. //查询自定义栏位的选项值
  621. Map<String, String> map = new HashMap<>();
  622. LambdaQueryWrapper<CustomOption> queryWrapper = new LambdaQueryWrapper<>();
  623. queryWrapper.eq(CustomOption::getCustomFieldId, customField.getId());
  624. List<CustomOption> customOptions = customOptionService.list(queryWrapper);
  625. for (CustomOption customOption : customOptions) {
  626. CustomOption customOption1 = new CustomOption();
  627. BeanUtils.copyProperties(customOption, customOption1);
  628. customOption1.setCustomFieldId(newId);
  629. customOption1.insert();
  630. map.put(customOption.getId().toString(), customOption1.getId().toString());
  631. }
  632. List<PatentWithIdVO> patents = esCustomFieldService.getEsCustomFields(customField.getId().toString(), customField.getType(), customField.getProjectId());
  633. for (PatentWithIdVO patentWithIdVO : patents) {
  634. Patent patent = patentWithIdVO.getPatent();
  635. patent.getESCustomField().setField(customField.getId().toString());
  636. patent.getESCustomField().setCreateTime(new Date());
  637. patent.getESCustomField().setTaskId(null);
  638. List<String> fieldValues = patent.getESCustomField().getFieldValue();
  639. fieldValues.forEach(item -> item = map.get(item));
  640. patent.getESCustomField().setFieldValue(fieldValues);
  641. patent.getESCustomField().setStatsValue(fieldValues);
  642. String parentId = patent.getPatentJoin().getParent();
  643. try {
  644. esService.addChildPatent(patent, parentId);
  645. } catch (Exception e) {
  646. throw new XiaoShiException("保存出错");
  647. }
  648. }
  649. }
  650. public void copyTreeValues(CustomField customField, CustomField newCustomField) {
  651. Integer newId = newCustomField.getId();
  652. Map<String, String> map = new HashMap<>();
  653. //查询自定义栏位的选项值
  654. LambdaQueryWrapper<TreeNode> queryWrapper = new LambdaQueryWrapper<>();
  655. queryWrapper.eq(TreeNode::getTypeId, customField.getId())
  656. .eq(TreeNode::getType, customField.getType())
  657. .orderByAsc(TreeNode::getLevel);
  658. List<TreeNode> treeNodes = treeNodeService.list(queryWrapper);
  659. for (TreeNode treeNode : treeNodes) {
  660. TreeNode treeNode1 = new TreeNode();
  661. BeanUtils.copyProperties(treeNode, treeNode1);
  662. treeNode1.setTypeId(newCustomField.getId());
  663. treeNode1.setId(null);
  664. String path = treeNode.getPath();
  665. if (path != null && !path.trim().equals("")) {
  666. List<String> paths = Arrays.asList(path.split("/"));
  667. paths.forEach(item -> item = map.get(item));
  668. path = StringUtils.join(paths, "/");
  669. treeNode1.setPath(path);
  670. }
  671. treeNode1.insert();
  672. map.put(treeNode.getId().toString(), treeNode1.getId().toString());
  673. }
  674. List<PatentWithIdVO> patents = esCustomFieldService.getEsCustomFields(customField.getId().toString(), customField.getType(), customField.getProjectId());
  675. for (PatentWithIdVO patentWithIdVO : patents) {
  676. Patent patent = patentWithIdVO.getPatent();
  677. patent.getESCustomField().setField(customField.getId().toString());
  678. patent.getESCustomField().setCreateTime(new Date());
  679. patent.getESCustomField().setTaskId(null);
  680. List<String> fieldValues = patent.getESCustomField().getFieldValue();
  681. List<String> statusValues = patent.getESCustomField().getStatsValue();
  682. fieldValues.forEach(item -> item = map.get(item));
  683. statusValues.forEach(item -> item = map.get(item));
  684. patent.getESCustomField().setFieldValue(fieldValues);
  685. patent.getESCustomField().setStatsValue(statusValues);
  686. String parentId = patent.getPatentJoin().getParent();
  687. //TODO添加path
  688. try {
  689. esService.addChildPatent(patent, parentId);
  690. } catch (Exception e) {
  691. throw new XiaoShiException("保存出错");
  692. }
  693. }
  694. }
  695. // public void copyProductValues(CustomField customField,CustomField newCustomField) {
  696. // Integer newId =newCustomField.getId();
  697. // //查询自定义栏位的选项值
  698. // List<Map<String,String>> maps =new ArrayList<>();
  699. // LambdaQueryWrapper<TreeNode> queryWrapper =new LambdaQueryWrapper<>();
  700. // queryWrapper.eq(TreeNode::getTypeId,customField.getId())
  701. // .eq(TreeNode::getType,customField.getType())
  702. // .orderByAsc(TreeNode::getLevel);
  703. // List<TreeNode> treeNodes = treeNodeService.list(queryWrapper);
  704. // for (TreeNode treeNode:treeNodes){
  705. // TreeNode treeNode1 =new TreeNode();
  706. // BeanUtils.copyProperties(treeNode,treeNode1);
  707. // treeNode1.setTypeId(newCustomField.getId());
  708. // treeNode1.setId(null);
  709. // //TODO添加path
  710. // Map<String,String> map =new HashMap<>();
  711. // map.put(treeNode.getId().toString(),treeNode1.getId().toString());
  712. //
  713. // }
  714. // List<PatentWithIdVO> patents = esCustomFieldService.getEsCustomFields(customField.getId().toString(), customField.getType(), customField.getProjectId());
  715. // for (PatentWithIdVO patentWithIdVO : patents) {
  716. // Patent patent =patentWithIdVO.getPatent();
  717. // patent.getESCustomField().setField(customField.getId().toString());
  718. // patent.getESCustomField().setCreateTime(new Date());
  719. // patent.getESCustomField().setTaskId(null);
  720. // String parentId =patent.getPatentJoin().getParent();
  721. // //TODO添加path
  722. // try {
  723. // esService.addChildPatent(patent,parentId);
  724. // }
  725. // catch (Exception e){
  726. // throw new XiaoShiException("保存出错");
  727. // }
  728. //
  729. // }
  730. //
  731. // }
  732. /**
  733. * @param customerFieldList
  734. */
  735. public List<EsCustomFieldDTO> getFieldDTOByStrings(List<UploadParamsVO.Field> customerFieldList, Integer projectId) {
  736. List<EsCustomFieldDTO> esCustomFieldDTOs =new ArrayList<>();
  737. if (customerFieldList != null && customerFieldList.size() != 0) {
  738. for (UploadParamsVO.Field field : customerFieldList) {
  739. EsCustomFieldDTO esCustomFieldDTO =new EsCustomFieldDTO();
  740. String key = field.getKey();
  741. List<String> fieldList = field.getFieldList();
  742. String[] keys = key.split(":");
  743. if (keys.length < 2) {
  744. continue;
  745. }
  746. String type = keys[1].trim();
  747. String fieldName = keys[0].trim();
  748. Integer typeInt = Integer.parseInt(type);
  749. Integer fieldId = this.GetIdByName(fieldName, projectId, typeInt);
  750. esCustomFieldDTO.setFieldId(fieldId.toString());
  751. List<String> values =new ArrayList<>();
  752. switch (type) {
  753. case "0":
  754. case "1":
  755. case "2":
  756. esCustomFieldDTO.setFieldType(Integer.parseInt(type));
  757. values =fieldList;
  758. break;
  759. case "4":
  760. case "5":
  761. esCustomFieldDTO.setFieldType(Integer.parseInt(type));
  762. values = customOptionService.getIdsByNames(fieldList,fieldId);
  763. break;
  764. case "6":
  765. esCustomFieldDTO.setFieldType(Integer.parseInt(type));
  766. values =treeNodeService.getIdByNames(fieldList,fieldId);
  767. break;
  768. }
  769. esCustomFieldDTO.setFieldValue(values);
  770. esCustomFieldDTO.setProjectId(projectId);
  771. esCustomFieldDTOs.add(esCustomFieldDTO);
  772. }
  773. }
  774. return esCustomFieldDTOs;
  775. }
  776. }