舆情分析详情页调整

src/main/java/com/zhiwei/brandkbs2/es/EsQueryTools.java

@@ -211,10 +211,26 @@ public class EsQueryTools {
         return channelBoolQueryBuilder;
     }
 
+    /**
+     * source模糊匹配
+     * @param sourceKeyword
+     * @return
+     */
     public static BoolQueryBuilder assembleSourceQuery(String sourceKeyword) {
         return assembleFiledKeywordQuery("source", sourceKeyword);
     }
 
+    /**
+     * source精确匹配
+     * @param sourceKeyword
+     * @return
+     */
+    public static BoolQueryBuilder assembleSourceKeywordQuery(String sourceKeyword) {
+        BoolQueryBuilder queryBuilder = QueryBuilders.boolQuery();
+        queryBuilder.should(QueryBuilders.termQuery("source", sourceKeyword));
+        return queryBuilder;
+    }
+
     public static void platformMustNot(BoolQueryBuilder boolQueryBuilder, String... platformNames) {
         if (null == platformNames) {
             return;

src/main/java/com/zhiwei/brandkbs2/service/impl/MarkDataServiceImpl.java

@@ -2815,19 +2815,46 @@ public class MarkDataServiceImpl implements MarkDataService {
 
     @Override
     public PageVO<MarkFlowEntity> getYuqingAnalyzeDetail(MarkSearchDTO dto) {
-        //设置默认的时间、页码、排序方式等
-        defaultMarkSearch(dto);
-        PageVO<MarkFlowEntity> yuqingMarkList;
-        if (Objects.isNull(dto.getPlanId())){
-            yuqingMarkList = getYuqingMarkList(dto);
-        }else {
-            yuqingMarkList = getNonManualMarkList(dto);
+        try {
+            //设置默认的时间、页码、排序方式等
+            defaultMarkSearch(dto);
+            String[] indexes = esClientDao.getIndexes();
+            BoolQueryBuilder postFilter = yuqingAnalyzeQuery(dto);
+            FieldSortBuilder sort = null;
+            if (null != dto.getSorter()) {
+                for (Map.Entry<String, Object> entry : dto.getSorter().entrySet()) {
+                    String key = entry.getKey();
+                    if (null == entry.getValue() || entry.getValue().toString().contains("desc")) {
+                        sort = SortBuilders.fieldSort(key).order(SortOrder.DESC);
+                    } else {
+                        sort = SortBuilders.fieldSort(key).order(SortOrder.ASC);
+                    }
+                }
+            }
+            int from = (dto.getPage() - 1) * dto.getPageSize();
+            int size = dto.getPageSize();
+            SearchHits hits = esClientDao.searchHits(indexes, postFilter, null, null, sort, from, size, null);
+            // 总量
+            long total = hits.getTotalHits().value;
+            // 消息列表
+            List<MarkFlowEntity> flowEntityList;
+            if (Objects.nonNull(dto.getPlanId())) {
+                flowEntityList = getNonManualMarkFlowEntity(dto, hits);
+            } else {
+                flowEntityList = getMarkFlowEntity(dto, hits);
+            }
+            JSONObject jsonObject = new JSONObject();
+            // 高亮关键词
+            jsonObject.put("highlightWord", GlobalPojo.HIGHLIGHT_MAP.get(UserThreadLocal.getProjectId()));
+            // 列表数据最小mgid
+            MarkFlowEntity markFlowEntity = flowEntityList.stream()
+                    .min(Comparator.comparingLong(entity -> JSONObject.parseObject(JSONObject.toJSONString(entity.getData())).getLongValue("mgid"))).orElse(null);
+            jsonObject.put("pageGid", Objects.isNull(markFlowEntity) ? Long.MAX_VALUE : JSONObject.parseObject(JSONObject.toJSONString(markFlowEntity.getData())).getLongValue("mgid"));
+            return PageVO.createPageVo(total, dto.getPage(), dto.getPageSize(), flowEntityList).setInfo(jsonObject);
+        }catch (Exception e){
+            ExceptionCast.cast(CommonCodeEnum.FAIL, "新舆情分析详情页getYuqingAnalyzeDetail异常-", e);
         }
-        // 列表数据最小mgid
-        MarkFlowEntity markFlowEntity = yuqingMarkList.getList().stream()
-                .min(Comparator.comparingLong(entity -> JSONObject.parseObject(JSONObject.toJSONString(entity.getData())).getLongValue("mgid"))).orElse(null);
-        yuqingMarkList.getInfo().put("pageGid", Objects.isNull(markFlowEntity) ? Long.MAX_VALUE : JSONObject.parseObject(JSONObject.toJSONString(markFlowEntity.getData())).getLongValue("mgid"));
-        return yuqingMarkList;
+        return PageVO.createPageVo(0, dto.getPage(), dto.getPageSize(), Collections.emptyList());
     }
 
     @Override
@@ -3334,7 +3361,7 @@ public class MarkDataServiceImpl implements MarkDataService {
         // 新增channelEmotion
         BoolQueryBuilder postFilter = EsQueryTools.assembleCacheMapsQuery(projectId, contendId, dto.getPoliticsLevel(), dto.getField(), dto.getRegion(), dto.getMainBodyType(), dto.getChannelEmotion());
         if (Objects.nonNull(dto.getPlanId())){
-            postFilter = EsQueryTools.assembleCacheMapsPlanQuery(projectId, dto.getPlanId());
+            postFilter = EsQueryTools.assembleCacheMapsPlanQuery(projectId, dto.getPlanId(), dto.getPoliticsLevel());
         }
         // time
         postFilter.must(QueryBuilders.rangeQuery("time").gte(dto.getStartTime()).lt(dto.getEndTime()));
@@ -3367,7 +3394,7 @@ public class MarkDataServiceImpl implements MarkDataService {
         }
         // sourceKeyword
         if (StringUtils.isNotEmpty(dto.getSourceKeyword())) {
-            postFilter.must(EsQueryTools.assembleSourceQuery(dto.getSourceKeyword()));
+            postFilter.must(EsQueryTools.assembleSourceKeywordQuery(dto.getSourceKeyword()));
         }
         // hostKeyword
         if (StringUtils.isNotEmpty(dto.getHostKeyword())) {