CustomFieldService.java 37 KB

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