123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176 |
- package cn.cslg.pas.service.business;
- import cn.cslg.pas.common.dto.GetAllCountryDTO;
- import cn.cslg.pas.common.dto.GetAllPersonDTO;
- import cn.cslg.pas.common.dto.MergePersonQueryDTO;
- import cn.cslg.pas.common.dto.PatentColumnDTO;
- import cn.cslg.pas.common.dto.es.EsMergePersonDTO;
- import cn.cslg.pas.common.dto.patentCount.GetEsAllPersonDTO;
- import cn.cslg.pas.common.model.cronModel.PersonnelVO;
- import cn.cslg.pas.common.model.cronModel.Records;
- import cn.cslg.pas.common.utils.CacheUtils;
- import cn.cslg.pas.common.utils.LoginUtils;
- import cn.cslg.pas.common.utils.parseQueryToTree.expressManager;
- import cn.cslg.pas.common.utils.parseQueryToTree.operateNode;
- import cn.cslg.pas.common.utils.parseQueryToTree.treeNode;
- import cn.cslg.pas.common.vo.business.*;
- import cn.cslg.pas.domain.BaseEntity;
- import cn.cslg.pas.domain.business.FollowUp;
- import cn.cslg.pas.domain.business.MergePerson;
- import cn.cslg.pas.domain.business.SystemDict;
- import cn.cslg.pas.domain.es.*;
- import cn.cslg.pas.exception.UnLoginException;
- import cn.cslg.pas.exception.XiaoShiException;
- import cn.cslg.pas.mapper.MergePersonMapper;
- import cn.cslg.pas.mapper.SystemDictMapper;
- import cn.cslg.pas.service.business.es.EsPatentService;
- import cn.cslg.pas.service.business.es.EsService;
- import cn.cslg.pas.service.query.FormatQueryService;
- import cn.hutool.core.util.ObjectUtil;
- import cn.hutool.core.util.PageUtil;
- import co.elastic.clients.elasticsearch.ElasticsearchClient;
- import co.elastic.clients.elasticsearch._types.FieldValue;
- import co.elastic.clients.elasticsearch._types.InlineScript;
- import co.elastic.clients.elasticsearch._types.Refresh;
- import co.elastic.clients.elasticsearch._types.Script;
- import co.elastic.clients.elasticsearch._types.aggregations.*;
- import co.elastic.clients.elasticsearch._types.query_dsl.IdsQuery;
- import co.elastic.clients.elasticsearch._types.query_dsl.Query;
- import co.elastic.clients.elasticsearch._types.query_dsl.QueryBuilders;
- import co.elastic.clients.elasticsearch.core.*;
- import co.elastic.clients.elasticsearch.core.search.Hit;
- import co.elastic.clients.elasticsearch.core.search.TrackHits;
- import co.elastic.clients.json.JsonData;
- import com.alibaba.fastjson.JSON;
- import com.alibaba.fastjson.JSONArray;
- import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
- import com.baomidou.mybatisplus.core.metadata.IPage;
- import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
- import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
- import com.github.pagehelper.PageHelper;
- import com.google.gson.JsonArray;
- import lombok.extern.slf4j.Slf4j;
- import org.apache.commons.lang3.ObjectUtils;
- import org.apache.commons.lang3.StringUtils;
- import org.springframework.beans.BeanUtils;
- import org.springframework.beans.factory.annotation.Autowired;
- import org.springframework.stereotype.Service;
- import org.springframework.transaction.annotation.Propagation;
- import org.springframework.transaction.annotation.Transactional;
- import org.springframework.util.CollectionUtils;
- import java.io.IOException;
- import java.util.*;
- import java.util.function.BiConsumer;
- import java.util.regex.Matcher;
- import java.util.regex.Pattern;
- import java.util.stream.Collectors;
- /**
- * @Author xiexiang
- * @Date 2024/01/02
- */
- @Slf4j
- @Service
- public class MergePersonService extends ServiceImpl<MergePersonMapper, MergePerson> {
- @Autowired
- private ElasticsearchClient client;
- @Autowired
- private CacheUtils cacheUtils;
- @Autowired
- private LoginUtils loginUtils;
- @Autowired
- private MergePersonMapper mergePersonMapper;
- @Autowired
- private FormatQueryService formatQueryService;
- @Autowired
- private SystemDictMapper systemDictMapper;
- /**
- * 专利列表上新增或编辑发明人/权利人/申请人合并共享接口
- * @param commonVO
- * @return
- * @throws Exception
- */
- @Transactional(propagation = Propagation.REQUIRED, rollbackFor = Throwable.class)
- public Integer commonMerge(MergePersonCommonVO commonVO) throws Exception {
- Integer mergeId = null;
- if (commonVO.getId() != null) {
- MergePerson person = mergePersonMapper.selectById(commonVO.getId());
- List<String> names = JSON.parseArray(person.getMergedName(), String.class);
- String delDiffName = "";
- if (!commonVO.getName().equals(person.getName())) {
- names.add(person.getName());
- delDiffName = person.getName();
- }
- loadMerging(commonVO, names, person,delDiffName);
- } else {
- loadMerging(commonVO, new ArrayList<>(), null, null);
- }
- return mergeId;
- }
- /**
- * 封装合并
- * @param commonVO
- * @param names
- * @param person
- * @param delDiffName
- * @return
- * @throws Exception
- */
- public Integer loadMerging(MergePersonCommonVO commonVO, List<String> names, MergePerson person,String delDiffName) throws Exception {
- Integer mergeId = null;
- Integer projectId = commonVO.getProjectId();
- Integer type = commonVO.getType();
- List<GetAllPersonDTO> mergePersonDTOS = commonVO.getMergePersonDTOS();
- List<GetAllPersonDTO> delMergePersonDTOS = commonVO.getDelMergePersonDTOS();
- if (!CollectionUtils.isEmpty(mergePersonDTOS)) {
- List<Integer> mergeIds = mergePersonDTOS.stream().map(GetAllPersonDTO::getMergeId).filter(Objects::nonNull).collect(Collectors.toList());
- if (!CollectionUtils.isEmpty(mergeIds)) {
- List<MergePerson> personList = mergePersonMapper.selectBatchIds(mergeIds);
- for (MergePerson mergePerson : personList) {
- List<String> getNames = JSON.parseArray(mergePerson.getMergedName(), String.class);
- names.addAll(getNames);
- String personName = mergePerson.getName();
- loadNewChild(type, projectId, personName,false);
- }
- mergePersonMapper.deleteBatchIds(mergeIds);
- }
- List<String> mergeNameList = mergePersonDTOS.stream().filter(i -> i.getMergeId() == null).map(GetAllPersonDTO::getName).collect(Collectors.toList());
- if (!CollectionUtils.isEmpty(mergeNameList)) {
- names.addAll(mergeNameList);
- }
- }
- if (!CollectionUtils.isEmpty(delMergePersonDTOS)) {
- List<String> delNames = delMergePersonDTOS.stream().map(GetAllPersonDTO::getName).collect(Collectors.toList());
- names.removeAll(delNames);
- loadNewChild(type, projectId, person.getName(),true);
- }
- if (commonVO.getId() != null) {
- mergePersonMapper.deleteById(commonVO.getId());
- }
- if (!CollectionUtils.isEmpty(names)) {
- List<String> collect = names.stream().distinct().collect(Collectors.toList());
- if (!CollectionUtils.isEmpty(collect)) {
- mergeId = merge(commonVO, collect,delDiffName);
- }
- }
- return mergeId;
- }
- /**
- * 封装子文档
- * @param type
- * @param projectId
- * @param personName
- * @param flag
- * @throws Exception
- */
- private void loadNewChild(Integer type, Integer projectId, String personName,boolean flag) throws Exception {
- List<Hit<Patent>> hits = searchChild(type, projectId, personName);
- List<String> patentIds = new ArrayList<>();
- List<PatentMergePerson> list = new ArrayList<>();
- for (Hit<Patent> hit : hits) {
- String routing = hit.routing();
- patentIds.add(routing);
- if (flag) {
- Patent patent = hit.source();
- if (type == 0) {
- if (!CollectionUtils.isEmpty(patent.getMergeApplicant())) {
- List<PatentMergePerson> applicant = patent.getMergeApplicant();
- List<PatentMergePerson> collect = applicant.stream().filter(i -> i.getIfMerged().equals(Boolean.TRUE)).collect(Collectors.toList());
- list.addAll(collect);
- }
- if (!CollectionUtils.isEmpty(patent.getMergeRightHolder())) {
- List<PatentMergePerson> rights = patent.getMergeRightHolder();
- List<PatentMergePerson> collect = rights.stream().filter(i -> i.getIfMerged().equals(Boolean.TRUE)).collect(Collectors.toList());
- list.addAll(collect);
- }
- } else {
- if (!CollectionUtils.isEmpty(patent.getMergeInventor())) {
- List<PatentMergePerson> inventors = patent.getMergeInventor();
- List<PatentMergePerson> collect = inventors.stream().filter(i -> i.getIfMerged().equals(Boolean.TRUE)).collect(Collectors.toList());
- list.addAll(collect);
- }
- }
- }
- }
- //删除原子文档
- deleteChild(patentIds, type, projectId);
- //查询父文档拿出对应type的数据后添加新的子文档
- searchPatent(patentIds, type, projectId);
- if (flag) {
- List<String> collect = list.stream().map(PatentMergePerson::getName).distinct().filter(i -> !i.equals(personName)).collect(Collectors.toList());
- if (!CollectionUtils.isEmpty(collect)) {
- for (String key : collect) {
- LambdaQueryWrapper<MergePerson> wrapper = new LambdaQueryWrapper<MergePerson>()
- .eq(MergePerson::getProjectId, projectId)
- .eq(MergePerson::getName, key)
- .eq(MergePerson::getType,type);
- MergePerson mergePerson = mergePersonMapper.selectOne(wrapper);
- String mergedName = mergePerson.getMergedName();
- List<String> mergedNames = JSONArray.parseArray(mergedName, String.class);
- againMerge(mergedNames,type,projectId,key);
- }
- }
- }
- }
- /**
- * 已存在的合并记录移除名称,不影响另一个合并记录
- * @param list
- * @param type
- * @param projectId
- * @param name
- * @throws Exception
- */
- private void againMerge(List<String> list, Integer type, Integer projectId, String name) throws Exception {
- List<Hit<Patent>> hits = searchChildDoc(list, type, projectId);
- //更新合并后的数据
- final int num = CollectionUtils.isEmpty(hits) ? 1 : hits.size();
- for (Hit<Patent> hit : hits) {
- String id = hit.id();
- Patent patent = hit.source();
- createPatent(patent, type, list, name, num);
- Integer count = updateChild(patent, id, hit.routing());
- if (count < 0) {
- throw new XiaoShiException("合并失败");
- }
- }
- }
- /**
- * 进行合并
- * @param commonVO
- * @param list
- * @param delDiffName
- * @return
- * @throws Exception
- */
- private Integer merge(MergePersonCommonVO commonVO, List<String> list, String delDiffName) throws Exception {
- Integer projectId = commonVO.getProjectId();
- Integer type = commonVO.getType();
- String name = commonVO.getName();
- //保存合并记录
- Integer mergeId = saveMerge(commonVO, list,delDiffName);
- //查询合并后的数据
- if (StringUtils.isNotEmpty(delDiffName)) {
- list.add(delDiffName);
- }
- List<Hit<Patent>> hits = searchChildDoc(list, type, projectId);
- //更新合并后的数据
- final int num = CollectionUtils.isEmpty(hits) ? 1 : hits.size();
- for (Hit<Patent> hit : hits) {
- String id = hit.id();
- Patent patent = hit.source();
- createPatent(patent, type, list, name, num);
- Integer count = updateChild(patent, id, hit.routing());
- if (count < 0) {
- throw new XiaoShiException("合并失败");
- }
- }
- return mergeId;
- }
- /**
- * 新增合并记录
- * @param vo
- * @param mergeNames
- * @param delDiffName
- * @return
- */
- private Integer saveMerge(MergePersonCommonVO vo, List<String> mergeNames, String delDiffName) {
- List<GetAllPersonDTO> mergePersonDTOS = vo.getMergePersonDTOS();
- List<String> names = new ArrayList<>();
- if (!CollectionUtils.isEmpty(mergeNames)) {
- names = mergeNames;
- } else {
- names = mergePersonDTOS.stream().map(GetAllPersonDTO::getName).collect(Collectors.toList());
- }
- if (StringUtils.isNotEmpty(delDiffName)) {
- names.remove(delDiffName);
- }
- //获取登陆人信息 用于设置创建人
- PersonnelVO personnelVO = new PersonnelVO();
- personnelVO = cacheUtils.getLoginUser(loginUtils.getId());
- //判断是否名称重复
- LambdaQueryWrapper<MergePerson> wrapper = new LambdaQueryWrapper<MergePerson>()
- .eq(MergePerson::getProjectId, vo.getProjectId())
- .eq(MergePerson::getType, vo.getType())
- .eq(MergePerson::getName, vo.getName());
- List<MergePerson> list = mergePersonMapper.selectList(wrapper);
- if (!list.isEmpty()) {
- throw new XiaoShiException("合并名称不可重复");
- }
- MergePerson person = new MergePerson();
- BeanUtils.copyProperties(vo, person);
- person.setMergedName(JSONArray.toJSONString(names));
- person.setCreateId(personnelVO.getId());
- person.setCreateTime(new Date());
- person.insert();
- return person.getId();
- }
- /**
- * 封装合并人的名称,被合并名称进行替换成合并名称
- * @param patent
- * @param type
- * @param mergedName
- * @param name
- * @param num
- */
- private void createPatent(Patent patent, Integer type, List<String> mergedName, String name, int num) {
- if (type == 0) {
- if (!CollectionUtils.isEmpty(patent.getMergeApplicant())) {
- List<PatentMergePerson> mergeApplicant = patent.getMergeApplicant();
- for (PatentMergePerson applicant : mergeApplicant) {
- if (mergedName.contains(applicant.getName().toUpperCase(Locale.ROOT))) {
- applicant.setName(name);
- applicant.setIfMerged(true);
- applicant.setMergedNum(num);
- }
- }
- List<PatentMergePerson> applicantList = new ArrayList<>();
- Map<String, List<PatentMergePerson>> map = mergeApplicant.stream().filter(i -> StringUtils.isNotEmpty(i.getName())).collect(Collectors.groupingBy(PatentMergePerson::getName));
- for (String key : map.keySet()) {
- applicantList.add(map.get(key).get(0));
- }
- patent.setMergeApplicant(applicantList);
- }
- if (!CollectionUtils.isEmpty(patent.getMergeRightHolder())) {
- List<PatentMergePerson> mergeRightHolder = patent.getMergeRightHolder();
- for (PatentMergePerson right : mergeRightHolder) {
- if (mergedName.contains(right.getName().toUpperCase(Locale.ROOT))) {
- right.setName(name);
- right.setIfMerged(true);
- right.setMergedNum(num);
- }
- }
- List<PatentMergePerson> rightList = new ArrayList<>();
- Map<String, List<PatentMergePerson>> map = mergeRightHolder.stream().filter(i -> StringUtils.isNotEmpty(i.getName())).collect(Collectors.groupingBy(PatentMergePerson::getName));
- for (String key : map.keySet()) {
- rightList.add(map.get(key).get(0));
- }
- patent.setMergeRightHolder(rightList);
- }
- } else {
- if (!CollectionUtils.isEmpty(patent.getMergeInventor())) {
- List<PatentMergePerson> mergeInventor = patent.getMergeInventor();
- for (PatentMergePerson inventor : mergeInventor) {
- if (mergedName.contains(inventor.getName().toUpperCase(Locale.ROOT))) {
- inventor.setName(name);
- inventor.setIfMerged(true);
- inventor.setMergedNum(num);
- }
- }
- List<PatentMergePerson> inventorList = new ArrayList<>();
- Map<String, List<PatentMergePerson>> map = mergeInventor.stream().filter(i -> StringUtils.isNotEmpty(i.getName())).collect(Collectors.groupingBy(PatentMergePerson::getName));
- for (String key : map.keySet()) {
- inventorList.add(map.get(key).get(0));
- }
- patent.setMergeInventor(inventorList);
- }
- }
- }
- /**
- * 根据多个已合并名称查询合并后的数据
- * @param mergedNames
- * @param type
- * @param projectId
- * @return
- * @throws IOException
- */
- private List<Hit<Patent>> searchChildDoc(List<String> mergedNames,Integer type,Integer projectId) throws IOException {
- SearchRequest.Builder builder = new SearchRequest.Builder();
- //设置查询索引
- builder.index("patent");
- List<Query> queries = new ArrayList<>();
- if (type == 0) {
- for (String mergedName : mergedNames) {
- //合并申请人
- Query idQ1 = QueryBuilders.term(i -> i.field("merge_applicant.project_id").value(projectId));
- Query q1 = QueryBuilders.term(i -> i.field("merge_applicant.name.raw").value(mergedName));
- Query bool = QueryBuilders.bool(i -> i.must(idQ1, q1));
- Query query1 = QueryBuilders.nested(i -> i.path("merge_applicant").query(bool));
- //合并权利人
- Query idQ2 = QueryBuilders.term(i -> i.field("merge_right_holder.project_id").value(projectId));
- Query q2 = QueryBuilders.term(i -> i.field("merge_right_holder.name.raw").value(mergedName));
- Query bool1 = QueryBuilders.bool(i -> i.must(idQ2, q2));
- Query query2 = QueryBuilders.nested(i -> i.path("merge_right_holder").query(bool1));
- queries.add(query1);
- queries.add(query2);
- }
- } else {
- for (String mergedName : mergedNames) {
- Query idQ = QueryBuilders.term(i -> i.field("merge_inventor.project_id").value(projectId));
- Query q = QueryBuilders.term(i -> i.field("merge_inventor.name.raw").value(mergedName));
- Query bool = QueryBuilders.bool(i -> i.must(idQ, q));
- Query query = QueryBuilders.nested(i -> i.path("merge_inventor").query(bool));
- queries.add(query);
- }
- }
- Query query = QueryBuilders.bool(i -> i.should(queries));
- builder.query(query);
- builder.size(1000);
- builder.trackTotalHits(TrackHits.of(i -> i.enabled(true)));
- SearchResponse<Patent> response = client.search(builder.build(), Patent.class);
- return response.hits().hits();
- }
- /**
- * 根据已合并名称查询合并后的数据
- * @param type
- * @param projectId
- * @param mergedName
- * @return
- * @throws IOException
- */
- private List<Hit<Patent>> searchChild(Integer type, Integer projectId, String mergedName) throws IOException {
- SearchRequest.Builder builder = new SearchRequest.Builder();
- //设置查询索引
- builder.index("patent");
- List<Query> queries = new ArrayList<>();
- if (type == 0) {
- //合并申请人
- Query idQ1 = QueryBuilders.term(i -> i.field("merge_applicant.project_id").value(projectId));
- Query q1 = QueryBuilders.term(i -> i.field("merge_applicant.name.raw").value(mergedName));
- Query bool = QueryBuilders.bool(i -> i.must(idQ1, q1));
- Query query1 = QueryBuilders.nested(i -> i.path("merge_applicant").query(bool));
- //合并权利人
- Query idQ2 = QueryBuilders.term(i -> i.field("merge_right_holder.project_id").value(projectId));
- Query q2 = QueryBuilders.term(i -> i.field("merge_right_holder.name.raw").value(mergedName));
- Query bool1 = QueryBuilders.bool(i -> i.must(idQ2, q2));
- Query query2 = QueryBuilders.nested(i -> i.path("merge_right_holder").query(bool1));
- queries.add(query1);
- queries.add(query2);
- } else {
- Query idQ = QueryBuilders.term(i -> i.field("merge_inventor.project_id").value(projectId));
- Query q = QueryBuilders.term(i -> i.field("merge_inventor.name.raw").value(mergedName));
- Query bool = QueryBuilders.bool(i -> i.must(idQ, q));
- Query query = QueryBuilders.nested(i -> i.path("merge_inventor").query(bool));
- queries.add(query);
- }
- Query query = QueryBuilders.bool(i -> i.should(queries));
- builder.query(query);
- builder.size(1000);
- builder.trackTotalHits(TrackHits.of(i -> i.enabled(true)));
- SearchResponse<Patent> response = client.search(builder.build(), Patent.class);
- return response.hits().hits();
- }
- /**
- * 合并人员详情
- *
- * @param vo
- * @return
- */
- public MergePersonQueryDTO selectMergePersonDetail(MergePersonIdVO vo) {
- MergePersonQueryDTO dto = new MergePersonQueryDTO();
- MergePerson person = mergePersonMapper.selectById(vo.getId());
- if (ObjectUtil.isNotEmpty(person)) {
- BeanUtils.copyProperties(person, dto);
- List<String> names = JSONArray.parseArray(person.getMergedName(), String.class);
- dto.setMergedName(names);
- }
- return dto;
- }
- /**
- * 获取所有发明人/权利人/申请人
- *
- * @param vo
- * @return
- * @throws Exception
- */
- public Records getAllMergePerson(GetAllPersonVO vo) throws Exception {
- Integer projectId = vo.getProjectId();
- Long pageNum = vo.getPageNum();
- Long pageSize = vo.getPageSize();
- Integer type = vo.getType();
- String condition = vo.getSearchQuery();
- //es获取的名称
- List<GetAllPersonDTO> personList = new ArrayList<>();
- SearchRequest.Builder builder = new SearchRequest.Builder();
- //设置查询索引
- builder.index("patent");
- String operate = "";
- if (StringUtils.isNotEmpty(condition)) {
- //1. 解析检索条件
- treeNode tree = expressManager.getInstance().Parse(condition, false);
- operateNode operateNode = (operateNode) tree;
- operate = operateNode.getoperate().getShowName();
- //3. 从es中检索数据
- Query q = formatQueryService.EsQueryToQuery((operateNode) tree, "patent", projectId);
- builder.query(q);
- if (type == 0) {
- this.loadAppAndRightAgg(builder,pageNum.intValue(),pageSize.intValue());
- } else {
- this.loadInventorAgg(builder,pageNum.intValue(),pageSize.intValue());
- }
- } else {
- if (type == 0) {
- //申请人
- Query query = this.loadQueryByType(type, projectId);
- builder.query(query);
- this.loadAppAndRightAgg(builder,pageNum.intValue(),pageSize.intValue());
- } else {
- Query query = this.loadQueryByType(type, projectId);
- builder.query(query);
- this.loadInventorAgg(builder,pageNum.intValue(),pageSize.intValue());
- }
- }
- builder.trackTotalHits(i -> i.enabled(true));
- SearchResponse<Patent> response = client.search(builder.build(), Patent.class);
- Aggregate totalAgg = response.aggregations().get("totalAgg");
- Aggregate totalAggregate = totalAgg.nested().aggregations().get("statsBucket");
- long total = totalAggregate.statsBucket().count();
- Aggregate totalAgg1 = null;
- if (type == 0) {
- totalAgg1 = response.aggregations().get("totalAgg1");
- Aggregate totalAggregate1 = totalAgg1.nested().aggregations().get("statsBucket1");
- long count = totalAggregate1.statsBucket().count();
- total = Math.max(total, count);
- }
- List<String> mergeList = new ArrayList<>();
- if (StringUtils.isNotEmpty(condition)) {
- Aggregate terms = totalAgg.nested().aggregations().get("terms");
- List<StringTermsBucket> termsBucketList = terms.sterms().buckets().array();
- Aggregate terms1 = null;
- if (type == 0) {
- terms1 = totalAgg1.nested().aggregations().get("terms1");
- List<StringTermsBucket> termsBucketList1 = terms1.sterms().buckets().array();
- termsBucketList.addAll(termsBucketList1);
- }
- if (condition.contains("AND") && operate.equals("AND")) {
- String[] parts = condition.split(" AND ");
- for (String part : parts) {
- List<String> expresses = getConditionExpress(part);
- List<String> merges = termsBucketList.stream().map(StringTermsBucket::key).map(FieldValue::stringValue).distinct()
- .filter(value -> expresses.stream().anyMatch(express -> value.contains(express.toUpperCase(Locale.ROOT))))
- .collect(Collectors.toList());
- total = total > merges.size() ? merges.size() : total;
- merges = merges.stream().skip((pageNum - 1) * pageSize).limit(pageSize).collect(Collectors.toList());
- if (CollectionUtils.isEmpty(mergeList)) {
- mergeList.addAll(merges);
- } else {
- merges.retainAll(mergeList);
- mergeList = new ArrayList<>(merges);
- total = total > mergeList.size() ? mergeList.size() : total;
- }
- }
- } else {
- List<String> expresses = getConditionExpress(condition);
- List<String> merges = termsBucketList.stream().map(StringTermsBucket::key).map(FieldValue::stringValue).distinct()
- .filter(value -> expresses.stream().anyMatch(express -> value.contains(express.toUpperCase(Locale.ROOT))))
- .collect(Collectors.toList());
- total = total > merges.size() ? merges.size() : total;
- merges = merges.stream().skip((pageNum - 1) * pageSize).limit(pageSize).collect(Collectors.toList());
- mergeList.addAll(merges);
- }
- } else {
- loadMergeList(type, response, mergeList);
- }
- List<String> nameList = mergeList.stream().distinct().collect(Collectors.toList());
- if (!CollectionUtils.isEmpty(nameList)) {
- personList = this.loadMergedDetail(nameList, projectId, type);
- }
- List<GetAllPersonDTO> collect = personList.stream().filter(i -> !StringUtils.isEmpty(i.getName())).collect(Collectors.toList());
- while (collect.size() > pageSize) {
- collect.remove(collect.size() - 1); // 从列表末尾开始删除
- }
- List<GetAllPersonDTO> list = collect.stream().sorted(Comparator.comparing(GetAllPersonDTO::getName)).collect(Collectors.toList());
- Records records = new Records();
- records.setCurrent(pageNum);
- records.setSize(pageSize);
- records.setData(list);
- records.setTotal(total);
- return records;
- }
- /**
- * 发明人/权利人/申请人合并记录删除
- *
- * @param vo
- * @return
- * @throws Exception
- */
- @Transactional(propagation = Propagation.REQUIRED, rollbackFor = Throwable.class)
- public Integer delMergePerson(MergePersonIdVO vo) throws Exception {
- Integer type = vo.getType();
- //1.查询合并记录
- MergePerson mergePerson = mergePersonMapper.selectById(vo.getId());
- //合并后的名称
- String name = mergePerson.getName();
- //专题库id
- Integer projectId = mergePerson.getProjectId();
- //2.查询合并后的数据
- List<Hit<Patent>> hits = searchChild(type, projectId, name);
- List<String> patentIds = new ArrayList<>();
- for (Hit<Patent> hit : hits) {
- String routing = hit.routing();
- patentIds.add(routing);
- }
- //3.删除原子文档
- deleteChild(patentIds, type, projectId);
- //4.查询父文档拿出对应type的数据后添加新的子文档
- searchPatent(patentIds, type, projectId);
- //5.删除合并记录
- mergePersonMapper.deleteById(vo.getId());
- return vo.getId();
- }
- public List<GetAllPersonDTO> loadMergedDetail(List<String> nameList, Integer projectId, Integer type) {
- List<GetAllPersonDTO> personList = new ArrayList<>();
- for (String name : nameList) {
- MergePerson person = mergePersonMapper.selectOne(new LambdaQueryWrapper<MergePerson>()
- .eq(MergePerson::getProjectId, projectId)
- .eq(MergePerson::getType, type)
- .eq(MergePerson::getName, name));
- if (!ObjectUtils.isEmpty(person)) {
- GetAllPersonDTO dto = new GetAllPersonDTO();
- dto.setName(name);
- dto.setMergeId(person.getId());
- dto.setProjectId(person.getProjectId());
- dto.setType(person.getType());
- dto.setRemark(person.getRemark());
- dto.setAbbreviation(person.getAbbreviation());
- dto.setCountry(person.getCountry());
- dto.setProvince(person.getProvince());
- dto.setAddress(person.getAddress());
- personList.add(dto);
- } else {
- GetAllPersonDTO dto = new GetAllPersonDTO();
- dto.setName(name);
- personList.add(dto);
- }
- }
- return personList;
- }
- public void loadMergeList(Integer type,SearchResponse<Patent> response,List<String> mergeList) {
- if (type == 0) {
- Aggregate agg = response.aggregations().get("Agg");
- Aggregate nestedAgg = agg.nested().aggregations().get("nestedAgg");
- List<StringTermsBucket> buckets = nestedAgg.sterms().buckets().array();
- Aggregate rightAgg = response.aggregations().get("rightAgg");
- Aggregate rightNestedAgg = rightAgg.nested().aggregations().get("rightNestedAgg");
- List<StringTermsBucket> bucketList = rightNestedAgg.sterms().buckets().array();
- if (buckets.size() >= bucketList.size()) {
- for (StringTermsBucket bucket : buckets) {
- String value = bucket.key().stringValue();
- mergeList.add(value);
- }
- for (StringTermsBucket bucket : bucketList) {
- String value = bucket.key().stringValue();
- mergeList.add(value);
- }
- } else {
- for (StringTermsBucket bucket : bucketList) {
- String value = bucket.key().stringValue();
- mergeList.add(value);
- }
- for (StringTermsBucket bucket : buckets) {
- String value = bucket.key().stringValue();
- mergeList.add(value);
- }
- }
- } else {
- Aggregate inventorAgg = response.aggregations().get("Agg");
- Aggregate inventorNestedAgg = inventorAgg.nested().aggregations().get("nestedAgg");
- List<StringTermsBucket> bucketList = inventorNestedAgg.sterms().buckets().array();
- if (!CollectionUtils.isEmpty(bucketList)) {
- for (StringTermsBucket bucket : bucketList) {
- String value = bucket.key().stringValue();
- mergeList.add(value);
- }
- }
- }
- }
- public Query loadQueryByType(Integer type,Integer projectId) {
- Map<Integer, Query> map = new HashMap<>();
- Query q1 = QueryBuilders.term(i -> i.field("merge_applicant.project_id").value(projectId));
- Query nestedQ1 = QueryBuilders.nested(i -> i.path("merge_applicant").query(q1));
- Query q2 = QueryBuilders.term(i -> i.field("merge_right_holder.project_id").value(projectId));
- Query nestedQ2 = QueryBuilders.nested(i -> i.path("merge_right_holder").query(q2));
- Query bool = QueryBuilders.bool(i -> i.should(nestedQ1, nestedQ2));
- map.put(0, bool);
- Query q = QueryBuilders.term(i -> i.field("merge_inventor.project_id").value(projectId));
- Query nestedQ = QueryBuilders.nested(i -> i.path("merge_inventor").query(q));
- map.put(2, nestedQ);
- return map.get(type);
- }
- public void loadAppAndRightAgg(SearchRequest.Builder builder, Integer pageNum, Integer pageSize) {
- Aggregation bucketSort = AggregationBuilders.bucketSort(i -> i.from((pageNum - 1) * pageSize).size(pageSize));
- Aggregation termAgg = new Aggregation.Builder().terms(i -> i.field("merge_applicant.name.raw").size(100000))
- .aggregations(new HashMap() {{
- put("termAgg", bucketSort);
- }}).build();
- Aggregation aggregation = new Aggregation.Builder().nested(new NestedAggregation.Builder().
- path("merge_applicant").build())
- .aggregations(new HashMap() {{
- put("nestedAgg", termAgg);
- }}).build();
- builder.aggregations("Agg", aggregation);
- //对聚合结果统计出总数
- Aggregation terms = AggregationBuilders.terms(i -> i.field("merge_applicant.name.raw").size(100000));
- BucketsPath bucketsPath = BucketsPath.of(i -> i.single("terms>_count"));
- Aggregation statsBucket = AggregationBuilders.statsBucket(i -> i.bucketsPath(bucketsPath));
- Aggregation totalAgg = new Aggregation.Builder().nested(new NestedAggregation.Builder().
- path("merge_applicant").build())
- .aggregations(new HashMap() {{
- put("terms", terms);
- put("statsBucket", statsBucket);
- }}).build();
- builder.aggregations("totalAgg", totalAgg);
- //权利人
- Aggregation bucketSort1 = AggregationBuilders.bucketSort(i -> i.from((pageNum - 1) * pageSize).size(pageSize));
- Aggregation termAgg1 = new Aggregation.Builder().terms(i -> i.field("merge_right_holder.name.raw").size(100000))
- .aggregations(new HashMap() {{
- put("rightTermAgg", bucketSort1);
- }}).build();
- Aggregation aggregation1 = new Aggregation.Builder().nested(new NestedAggregation.Builder().
- path("merge_right_holder").build())
- .aggregations(new HashMap() {{
- put("rightNestedAgg", termAgg1);
- }}).build();
- builder.aggregations("rightAgg", aggregation1);
- // //对聚合结果统计出总数
- Aggregation terms1 = AggregationBuilders.terms(i -> i.field("merge_right_holder.name.raw").size(100000));
- BucketsPath bucketsPath1 = BucketsPath.of(i -> i.single("terms1>_count"));
- Aggregation statsBucket1 = AggregationBuilders.statsBucket(i -> i.bucketsPath(bucketsPath1));
- Aggregation totalAgg1 = new Aggregation.Builder().nested(new NestedAggregation.Builder().
- path("merge_right_holder").build())
- .aggregations(new HashMap() {{
- put("terms1", terms1);
- put("statsBucket1", statsBucket1);
- }}).build();
- builder.aggregations("totalAgg1", totalAgg1);
- }
- public void loadInventorAgg(SearchRequest.Builder builder, Integer pageNum, Integer pageSize) {
- Aggregation bucketSort = AggregationBuilders.bucketSort(i -> i.from((pageNum - 1) * pageSize).size(pageSize));
- Aggregation termAgg = new Aggregation.Builder().terms(i -> i.field("merge_inventor.name.raw").size(100000))
- .aggregations(new HashMap() {{
- put("termAgg", bucketSort);
- }}).build();
- Aggregation aggregation = new Aggregation.Builder().nested(new NestedAggregation.Builder().
- path("merge_inventor").build())
- .aggregations(new HashMap() {{
- put("nestedAgg", termAgg);
- }}).build();
- builder.aggregations("Agg", aggregation);
- //对聚合结果统计出总数
- Aggregation terms = AggregationBuilders.terms(i -> i.field("merge_inventor.name.raw").size(100000));
- BucketsPath bucketsPath = BucketsPath.of(i -> i.single("terms>_count"));
- Aggregation statsBucket = AggregationBuilders.statsBucket(i -> i.bucketsPath(bucketsPath));
- Aggregation totalAgg = new Aggregation.Builder().nested(new NestedAggregation.Builder().
- path("merge_inventor").build())
- .aggregations(new HashMap() {{
- put("terms", terms);
- put("statsBucket", statsBucket);
- }}).build();
- builder.aggregations("totalAgg", totalAgg);
- }
- private List<String> getConditionExpress(String input) {
- // 编写正则表达式来匹配 = 后面的双括号内的内容
- String regex = "=\\(\\(([^)]*)\\)\\)";
- // 编译正则表达式
- Pattern pattern = Pattern.compile(regex);
- // 创建matcher对象
- Matcher matcher = pattern.matcher(input);
- // 使用List来存储结果,因为数组的大小是固定的
- List<String> values = new ArrayList<>();
- // 查找所有匹配项
- while (matcher.find()) {
- // 提取双括号中的内容
- String content = matcher.group(1);
- // 将提取的内容添加到列表中
- values.add(content);
- }
- List<String> list = new ArrayList<>();
- List<String> collect = values.stream().filter(cn.cslg.pas.common.utils.StringUtils::isNotEmpty).distinct().collect(Collectors.toList());
- for (String s : collect) {
- List<String> split = splitWithQuotes(s);
- list.addAll(split);
- }
- List<String> list1 = new ArrayList<>();
- for (String s : list) {
- if (s.contains("\"")) {
- String replace = s.replace("\"", "");
- list1.add(replace);
- } else {
- list1.add(s);
- }
- }
- return list1.stream().filter(cn.cslg.pas.common.utils.StringUtils::isNotEmpty).distinct().collect(Collectors.toList());
- }
- public static List<String> splitWithQuotes(String input) {
- List<String> result = new ArrayList<>();
- StringBuilder currentPart = new StringBuilder();
- boolean inQuotes = false;
- for (int i = 0; i < input.length(); i++) {
- char c = input.charAt(i);
- if (c == '"' || c == '\'') {
- // 切换引号状态
- inQuotes = !inQuotes;
- // 如果我们不是在引号内,并且前面有内容,则添加当前部分到结果中
- if (!inQuotes && currentPart.length() > 0) {
- result.add(currentPart.toString());
- currentPart.setLength(0); // 重置当前部分
- }
- // 引号内的字符直接添加到当前部分
- currentPart.append(c);
- // 如果引号后面紧跟着的是空格,并且我们刚刚离开了引号,则跳过这个空格
- if (i + 1 < input.length() && input.charAt(i + 1) == ' ' && !inQuotes) {
- i++; // 跳过空格
- }
- } else if (c == ' ' && !inQuotes) {
- // 如果我们在空格处且不在引号内,则添加当前部分到结果中,并重置当前部分
- if (currentPart.length() > 0) {
- result.add(currentPart.toString());
- currentPart.setLength(0);
- }
- } else {
- // 其他字符直接添加到当前部分
- currentPart.append(c);
- }
- }
- // 添加最后一个部分(如果有的话)
- if (currentPart.length() > 0) {
- result.add(currentPart.toString());
- }
- // 将List转换为数组
- return result;
- }
- //添加子文档
- private void addChild(List<PatentPerson> personList, Integer projectId, String id, String joinName, Integer type) throws Exception {
- List<PatentMergePerson> mergePersonList = createMergePersonList(personList, projectId);
- // 根据type决定设置哪个字段
- Map<Integer, BiConsumer<Patent, List<PatentMergePerson>>> setterMap = new HashMap<>();
- setterMap.put(0, (patent, list) -> patent.setMergeApplicant(list));
- setterMap.put(1, (patent, list) -> patent.setMergeRightHolder(list));
- setterMap.put(2, (patent, list) -> patent.setMergeInventor(list));
- BiConsumer<Patent, List<PatentMergePerson>> setter = setterMap.getOrDefault(type, (p, l) -> {});
- // 通用添加逻辑
- String child = addChildCommon(id, joinName, mergePersonList, setter);
- if (StringUtils.isEmpty(child)) {
- throw new XiaoShiException("添加子文档失败");
- }
- // 如果type为0,还需要额外添加一个字段
- if (type == 0) {
- setter = setterMap.get(1); // 假设第二个字段是mergeRightHolder
- String child1 = addChildCommon(id, joinName, mergePersonList, setter);
- if (StringUtils.isEmpty(child1)) {
- throw new XiaoShiException("添加子文档失败");
- }
- }
- }
- private List<PatentMergePerson> createMergePersonList(List<PatentPerson> personList, Integer projectId) {
- List<PatentMergePerson> mergePersonList = new ArrayList<>();
- if (!CollectionUtils.isEmpty(personList)) {
- for (PatentPerson person : personList) {
- PatentMergePerson merge = new PatentMergePerson();
- merge.setName(person.getName());
- merge.setType(person.getType());
- merge.setOrder(person.getOrder());
- merge.setProjectId(String.valueOf(projectId));
- mergePersonList.add(merge);
- }
- }
- return mergePersonList;
- }
- private String addChildCommon(String id, String joinName, List<PatentMergePerson> mergePersonList, BiConsumer<Patent, List<PatentMergePerson>> setter) throws Exception {
- Patent newPatent = new Patent();
- PatentJoin patentJoin = new PatentJoin();
- patentJoin.setParent(id);
- patentJoin.setName(joinName);
- newPatent.setPatentJoin(patentJoin);
- setter.accept(newPatent, mergePersonList);
- return this.addChildToES(newPatent, id);
- }
- private void deleteChild(List<String> ids, Integer type, Integer projectId) throws IOException {
- SearchRequest.Builder builder = new SearchRequest.Builder();
- //设置查询索引
- builder.index("patent");
- List<Query> queries = new ArrayList<>();
- if (type == 0) {
- //合并申请人
- Query q = QueryBuilders.term(i -> i.field("merge_applicant.project_id").value(projectId));
- Query query1 = QueryBuilders.nested(i -> i.path("merge_applicant").query(q));
- //合并权利人
- Query q1 = QueryBuilders.term(i -> i.field("merge_right_holder.project_id").value(projectId));
- Query query2 = QueryBuilders.nested(i -> i.path("merge_right_holder").query(q1));
- queries.add(query1);
- queries.add(query2);
- } else {
- Query q = QueryBuilders.term(i -> i.field("merge_inventor.project_id").value(projectId));
- Query query = QueryBuilders.nested(i -> i.path("merge_inventor").query(q));
- queries.add(query);
- }
- Query query = QueryBuilders.bool(i -> i.should(queries));
- Query idQ = QueryBuilders.ids(i -> i.values(ids));
- Query parentQ = QueryBuilders.hasParent(i -> i.parentType("patent").query(idQ));
- Query bool = QueryBuilders.bool(i -> i.must(parentQ, query));
- builder.query(bool);
- builder.size(1000);
- builder.trackTotalHits(TrackHits.of(i -> i.enabled(true)));
- SearchResponse<Patent> response = client.search(builder.build(), Patent.class);
- List<Hit<Patent>> hits = response.hits().hits();
- List<String> idList = new ArrayList<>();
- for (Hit<Patent> hit : hits) {
- String id = hit.id();
- idList.add(id);
- }
- //删除子文档
- delete(idList);
- }
- //查询专利父文档
- private void searchPatent(List<String> ids, Integer type, Integer projectId) throws Exception {
- SearchRequest.Builder builder = new SearchRequest.Builder();
- //设置查询索引
- builder.index("patent");
- Query query = QueryBuilders.ids(i -> i.values(ids));
- builder.query(query);
- builder.size(1000);
- builder.trackTotalHits(TrackHits.of(i -> i.enabled(true)));
- SearchResponse<Patent> response = client.search(builder.build(), Patent.class);
- List<Hit<Patent>> hits = response.hits().hits();
- for (Hit<Patent> hit : hits) {
- String id = hit.id();
- Patent patent = hit.source();
- if (type == 0) {
- if (!CollectionUtils.isEmpty(patent.getApplicant())) {
- this.addChild(patent.getApplicant(), projectId, id, "merge_applicat",type);
- }
- if (!CollectionUtils.isEmpty(patent.getRightHolder())) {
- addChild(patent.getRightHolder(), projectId, id, "merge_right_holder",type);
- }
- } else {
- if (!CollectionUtils.isEmpty(patent.getInventor())) {
- addChild(patent.getInventor(), projectId, id, "merge_inventor",type);
- }
- }
- }
- }
- /**
- * 获取所有国家列表查询
- *
- * @return
- */
- public List<GetAllCountryDTO> getAllCountry() {
- List<GetAllCountryDTO> list = new ArrayList<>();
- List<SystemDict> countries = systemDictMapper.selectList(new LambdaQueryWrapper<SystemDict>()
- .eq(SystemDict::getType, "COUNTRIES"));
- if (!CollectionUtils.isEmpty(countries)) {
- countries.forEach(county -> {
- GetAllCountryDTO dto = new GetAllCountryDTO();
- dto.setLabel(county.getLabel());
- dto.setValue(county.getValue());
- dto.setType(county.getType());
- list.add(dto);
- });
- }
- return list;
- }
- /**
- * 获取国内省份列表查询
- *
- * @return
- */
- public List<GetAllCountryDTO> getAllProvince() {
- List<GetAllCountryDTO> list = new ArrayList<>();
- List<SystemDict> countries = systemDictMapper.selectList(new LambdaQueryWrapper<SystemDict>()
- .eq(SystemDict::getType, "PROVINCE"));
- if (!CollectionUtils.isEmpty(countries)) {
- countries.forEach(county -> {
- GetAllCountryDTO dto = new GetAllCountryDTO();
- dto.setLabel(county.getLabel());
- dto.setValue(county.getValue());
- dto.setType(county.getType());
- list.add(dto);
- });
- }
- return list;
- }
- /**
- * @param patent
- * @throws Exception
- */
- public String addChildToES(Patent patent, String id) throws Exception {
- IndexResponse indexResponse = client.index(i -> i
- .index("patent")
- .routing(id)
- //传入user对象
- .document(patent).refresh(Refresh.True)
- );
- return indexResponse.id();
- }
- //更新patent
- public Integer updateChild(Patent patent, String id, String routing) {
- UpdateRequest<Patent, Patent> req;
- req = UpdateRequest.of(
- b -> b.index("patent")
- .id(id)
- .routing(routing)
- .doc(patent).refresh(Refresh.True)
- );
- try {
- client.update(req, Patent.class);
- return 1;
- } catch (IOException e) {
- return -1;
- }
- }
- public Integer delete(List<String> ids) {
- Query query = QueryBuilders.ids(n -> n.values(ids));
- DeleteByQueryRequest request = DeleteByQueryRequest.of(i -> i.index("patent").refresh(true).query(query));
- try {
- client.deleteByQuery(request);
- return 1;
- } catch (IOException e) {
- throw new XiaoShiException("删除失败");
- }
- }
- /**
- * 删除申请人/权利人/发明人合并名称
- *
- * @param id
- * @return
- */
- public Integer delSingleMerge(String id, Integer type, String name) {
- String source = "";
- if (type == 0) {
- source = "if (ctx._source.merge_applicant != null) { ctx._source.merge_applicant.removeIf(item -> item.name == params.name); } if (ctx._source.merge_right_holder != null) { ctx._source.merge_right_holder .removeIf(item -> item.name == params.name); }";
- } else {
- source = "if (ctx._source.merge_inventor != null) { ctx._source.merge_inventor.removeIf(item -> item.name == params.name); }";
- }
- String finalSource = source;
- InlineScript inlineScript = InlineScript.of(i -> i.lang("painless").params("name", JsonData.of(name)).source(finalSource));
- Script script = Script.of(i -> i.inline(inlineScript));
- Query query = QueryBuilders.term(i -> i.field("_id").value(id));
- UpdateByQueryRequest request = UpdateByQueryRequest.of(i -> i.index("patent").script(script).query(query));
- try {
- client.updateByQuery(request);
- return 1;
- } catch (IOException e) {
- return -1;
- }
- }
- public Integer add(String id, Integer type, List<PatentMergePerson> mergePersonList) {
- Integer flag = null;
- for (PatentMergePerson mergePerson : mergePersonList) {
- Map<String, JsonData> map = new HashMap<>();
- map.put("name", JsonData.of(mergePerson.getName()));
- map.put("project_id", JsonData.of(mergePerson.getProjectId()));
- map.put("type", JsonData.of(mergePerson.getType()));
- map.put("order", JsonData.of(mergePerson.getOrder()));
- String source = "";
- if (type == 0) {
- source = "if (ctx._source.merge_applicant != null) {ctx._source.merge_applicant.add(params.data)} else { List list = new ArrayList();list.add(params.data);ctx._source.merge_applicant = list;}" +
- "if (ctx._source.merge_right_holder != null) { ctx._source.merge_right_holder.add(params.data)} else { List list = new ArrayList();list.add(params.data);ctx._source.merge_right_holder = list;}";
- } else {
- source = "if (ctx._source.merge_inventor != null) {ctx._source.merge_inventor.add(params.data)} else { List list = new ArrayList();list.add(params.data);ctx._source.merge_inventor = list;}";
- }
- String finalSource = source;
- InlineScript inlineScript = InlineScript.of(i -> i.lang("painless").params("data", JsonData.of(map)).source(finalSource));
- Script script = Script.of(i -> i.inline(inlineScript));
- Query query = QueryBuilders.term(i -> i.field("_id").value(id));
- UpdateByQueryRequest request = UpdateByQueryRequest.of(i -> i.index("patent").script(script).query(query));
- try {
- client.updateByQuery(request);
- flag += 1;
- } catch (IOException e) {
- flag += -1;
- }
- }
- return flag;
- }
- public Integer edit(String id, Integer type, String oldName, String newName) {
- String source = "";
- if (type == 0) {
- source = "if (ctx._source.merge_applicant != null) {for (item in ctx._source.merge_applicant) {if(item['name']==params.oldName){item['name']=params.newName}}}" +
- "if (ctx._source.merge_right_holder != null) {for (item in ctx._source.merge_right_holder) {if(item['name']==params.oldName){item['name']=params.newName}}}";
- } else {
- source = "if (ctx._source.merge_inventor != null) {for (item in ctx._source.merge_inventor) {if(item['name']==params.oldName){item['name']=params.newName}}}";
- }
- String finalSource = source;
- Map<String, JsonData> map = new HashMap<>();
- map.put("oldName", JsonData.of(oldName));
- map.put("newName", JsonData.of(newName));
- InlineScript inlineScript = InlineScript.of(i -> i.lang("painless").params(map).source(finalSource));
- Script script = Script.of(i -> i.inline(inlineScript));
- Query query = QueryBuilders.term(i -> i.field("_id").value(id));
- UpdateByQueryRequest request = UpdateByQueryRequest.of(i -> i.index("patent").script(script).query(query));
- try {
- client.updateByQuery(request);
- return 1;
- } catch (IOException e) {
- return -1;
- }
- }
- /**
- * 删除申请人/权利人/发明人合并名称
- *
- * @param patent
- * @param id
- * @return
- */
- public Integer delMergePerson(Patent patent, String id, Integer type, String name) {
- String source = "";
- if (type == 0) {
- source = "if (ctx._source.merge_applicant != null) { ctx._source.merge_applicant.removeIf(item -> item.name == params.name); } if (ctx._source.merge_right_holder != null) { ctx._source.merge_right_holder .removeIf(item -> item.name == params.name); }";
- } else {
- source = "if (ctx._source.merge_inventor != null) { ctx._source.merge_inventor.removeIf(item -> item.name == params.name); }";
- }
- String finalSource = source;
- InlineScript inlineScript = InlineScript.of(i -> i.lang("painless").params("name", JsonData.of(name)).source(finalSource));
- Script script = Script.of(i -> i.inline(inlineScript));
- Query query = QueryBuilders.term(i -> i.field("_id").value(id));
- UpdateByQueryRequest request = UpdateByQueryRequest.of(i -> i.index("patent").script(script).query(query));
- try {
- client.updateByQuery(request);
- return 1;
- } catch (IOException e) {
- return -1;
- }
- }
- }
|