CustomFieldService.java 37 KB

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