package com.ruoyi.service.impl;
|
|
import com.ruoyi.common.core.domain.entity.EsModel;
|
import com.ruoyi.domain.ZAuthority;
|
import com.ruoyi.domain.ZInfoUser;
|
import com.ruoyi.mapper.EsRepository;
|
import com.ruoyi.service.EsService;
|
|
import com.ruoyi.service.ZAuthorityService;
|
import com.ruoyi.service.ZInfoUserService;
|
import org.elasticsearch.index.query.BoolQueryBuilder;
|
import org.elasticsearch.index.query.MultiMatchQueryBuilder;
|
import org.elasticsearch.index.query.QueryBuilders;
|
import org.springframework.data.domain.PageRequest;
|
import org.springframework.data.elasticsearch.core.ElasticsearchRestTemplate;
|
import org.springframework.data.elasticsearch.core.SearchHit;
|
import org.springframework.data.elasticsearch.core.SearchHits;
|
import org.springframework.data.elasticsearch.core.mapping.IndexCoordinates;
|
import org.springframework.data.elasticsearch.core.query.NativeSearchQuery;
|
import org.springframework.data.elasticsearch.core.query.NativeSearchQueryBuilder;
|
import org.springframework.data.elasticsearch.core.query.Query;
|
import org.springframework.stereotype.Service;
|
|
import javax.annotation.Resource;
|
import java.util.ArrayList;
|
import java.util.Iterator;
|
import java.util.List;
|
import java.util.Objects;
|
import java.util.stream.Collectors;
|
|
import static com.ruoyi.constant.MenuAuthority.EQUIPMENT_LIST;
|
|
@Service
|
public class EsServiceImpl implements EsService {
|
@Resource
|
private EsRepository esRepository;
|
@Resource
|
private ElasticsearchRestTemplate elasticsearchRestTemplate;
|
|
@Resource
|
private ZInfoUserService zInfoUserService;
|
|
@Resource
|
private ZAuthorityService zAuthorityService;
|
|
@Override
|
public int insertTable(EsModel esModel) {
|
try {
|
esRepository.save(esModel);
|
// esRepository.searchSimilar()
|
} catch (Exception e) {
|
System.out.println(e);
|
// throw new Exception("ES数据同步失败，请联系管理员处理！");
|
}
|
return 0;
|
}
|
|
// 全文检索
|
@Override
|
public List<EsModel> findByCondition(String con) {
|
List<EsModel> esm = new ArrayList<>();
|
// MatchQueryBuilder matchQueryBuilder = QueryBuilders.matchQuery("title", title);
|
//多字段进行匹配
|
// MultiMatchQueryBuilder multiMatchQueryBuilder = QueryBuilders.multiMatchQuery(con, "ctName", "ctContent","remark",
|
// "by1","by2","by3","by4");
|
|
|
//要查自己家庭的
|
ZInfoUser myself = zInfoUserService.getMyself();
|
Long familyId = myself.getFamilyId();
|
//也要查别人授权的
|
List<ZAuthority> authority = zAuthorityService.getAuthority();
|
List<Long> idList = authority.stream().filter(auth -> auth.getAuthority().toString().equals(EQUIPMENT_LIST)).map(ZAuthority::getFid).collect(Collectors.toList());
|
//加上自己家庭的id
|
idList.add(familyId);
|
|
|
BoolQueryBuilder boolQueryBuilder = QueryBuilders.boolQuery();
|
boolQueryBuilder.should(QueryBuilders.matchQuery("ctName",con));
|
boolQueryBuilder.should(QueryBuilders.matchQuery("ctContent",con));
|
boolQueryBuilder.should(QueryBuilders.matchQuery("remark",con));
|
boolQueryBuilder.should(QueryBuilders.matchQuery("by1",con));
|
boolQueryBuilder.should(QueryBuilders.matchQuery("by2",con));
|
boolQueryBuilder.should(QueryBuilders.matchQuery("by3",con));
|
boolQueryBuilder.should(QueryBuilders.matchQuery("by4",con));
|
boolQueryBuilder.should(QueryBuilders.matchQuery("by6",con));
|
boolQueryBuilder.should(QueryBuilders.matchQuery("by7",con));
|
|
|
NativeSearchQueryBuilder nativeSearchQueryBuilder = new NativeSearchQueryBuilder()
|
.withQuery(boolQueryBuilder);
|
NativeSearchQuery nativeSearchQuery = nativeSearchQueryBuilder.build();
|
|
// 执行查询
|
SearchHits<EsModel> searchHits = elasticsearchRestTemplate.search(nativeSearchQuery,EsModel.class);
|
searchHits.getSearchHits().forEach(personSearchHit -> {
|
EsModel content = personSearchHit.getContent();
|
esm.add(content);
|
// System.out.println(content);
|
});
|
// articleList.forEach(a-> System.out.println(a))
|
|
return esm.stream().filter(esModel -> {
|
return idList.contains(esModel.getFid());
|
}).collect(Collectors.toList());
|
|
}
|
|
//根据mysql中的id查找信息
|
@Override
|
public EsModel findByCtId(Integer ctId, String ctTableName) {
|
// NativeSearchQuery searchQuery = new NativeSearchQueryBuilder()
|
// .withQuery(QueryBuilders.matchQuery("ctId", ctId)) // 自定义查询条件
|
// .withQuery(QueryBuilders.matchQuery("ctTableName", ctTableName)) // 自定义查询条件
|
// .build();
|
BoolQueryBuilder boolQueryBuilder = QueryBuilders.boolQuery();
|
boolQueryBuilder.must(QueryBuilders.matchQuery("ctId", ctId));
|
boolQueryBuilder.must(QueryBuilders.matchQuery("ctTableName", ctTableName));
|
|
NativeSearchQueryBuilder nativeSearchQueryBuilder = new NativeSearchQueryBuilder()
|
.withQuery(boolQueryBuilder);
|
NativeSearchQuery nativeSearchQuery = nativeSearchQueryBuilder.build();
|
|
SearchHit<EsModel> esModelSearchHit = elasticsearchRestTemplate.searchOne(nativeSearchQuery, EsModel.class);
|
if(esModelSearchHit != null){
|
return esModelSearchHit.getContent();
|
}
|
return null;
|
|
}
|
|
@Override
|
public void deleteById(String id) {
|
esRepository.deleteById(id);
|
}
|
|
@Override
|
public void deleteAllData() {
|
esRepository.deleteAll();
|
}
|
|
@Override
|
public List<EsModel> getAll() {
|
|
ArrayList<EsModel> esModels = new ArrayList<>();
|
|
// 构建查询请求
|
SearchHits<EsModel> searchHits = elasticsearchRestTemplate.search(Query.findAll(), EsModel.class,IndexCoordinates.of("allsearchdata"));
|
|
Iterator<SearchHit<EsModel>> iterator = searchHits.stream().iterator();
|
while (iterator.hasNext()){
|
esModels.add(iterator.next().getContent());
|
}
|
// 返回查询结果
|
return esModels;
|
}
|
}
|
