From 239a0ff14b0cfd8d46ba1a9113900cd5a95f4616 Mon Sep 17 00:00:00 2001
From: zhn <947514737@qq.com>
Date: Fri, 25 Mar 2022 22:31:45 +0800
Subject: [PATCH] =?UTF-8?q?=E6=96=87=E6=A1=A3=E5=BA=93--=E6=90=9C=E7=B4=A2?=
 =?UTF-8?q?=E4=B8=AD=E5=BF=83=E6=B7=BB=E5=8A=A0=E6=94=B6=E8=97=8F=E5=92=8C?=
 =?UTF-8?q?=E8=AE=A2=E9=98=85?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../impl/DocumentSearchServiceImpl.java       | 29 +++++++++++++++++++
 1 file changed, 29 insertions(+)

diff --git a/jero-boot/jero-boot-modules/src/main/java/com/jero/modules/searchcenter/service/impl/DocumentSearchServiceImpl.java b/jero-boot/jero-boot-modules/src/main/java/com/jero/modules/searchcenter/service/impl/DocumentSearchServiceImpl.java
index a7da908b9..5b16aa34f 100644
--- a/jero-boot/jero-boot-modules/src/main/java/com/jero/modules/searchcenter/service/impl/DocumentSearchServiceImpl.java
+++ b/jero-boot/jero-boot-modules/src/main/java/com/jero/modules/searchcenter/service/impl/DocumentSearchServiceImpl.java
@@ -3,6 +3,7 @@ package com.jero.modules.searchcenter.service.impl;
 import com.alibaba.fastjson.JSON;
 import com.alibaba.fastjson.JSONArray;
 import com.alibaba.fastjson.JSONObject;
+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.jero.common.constant.enums.ModuleEnum;
@@ -10,6 +11,8 @@ import com.jero.common.constant.enums.YesOrNoEnum;
 import com.jero.common.es.JeroElasticsearchTemplate;
 import com.jero.generater.modules.online.cgform.entity.OnlCgformField;
 import com.jero.generater.modules.online.cgform.service.impl.OnlCgformFieldServiceImpl;
+import com.jero.modules.collection.entity.OnlCgformCollection;
+import com.jero.modules.collection.service.IOnlCgformCollectionService;
 import com.jero.modules.document.enums.FieldTypeEnum;
 import com.jero.modules.document.enums.SearchEnum;
 import com.jero.modules.document.service.IBussDocumentLibraryEOService;
@@ -17,6 +20,8 @@ import com.jero.modules.document.service.impl.BussDocumentLibraryEOServiceImpl;
 import com.jero.modules.oss.entity.OSSFile;
 import com.jero.modules.oss.service.IOSSFileService;
 import com.jero.modules.searchcenter.vo.SearchVO;
+import com.jero.modules.subscribe.entity.OnlCgformSubscribe;
+import com.jero.modules.subscribe.service.IOnlCgformSubscribeService;
 import com.jero.modules.system.entity.SysCategory;
 import com.jero.modules.system.entity.SysDictItem;
 import com.jero.modules.system.service.impl.SysCategoryServiceImpl;
@@ -54,6 +59,10 @@ public class DocumentSearchServiceImpl implements IDocumentSearchService {
     private BussDocumentLibraryEOServiceImpl bussDocumentLibraryEOService;
     @Autowired
     private IOSSFileService iOSSFileService;
+    @Autowired
+    private IOnlCgformCollectionService iOnlCgformCollectionService;
+    @Autowired
+    private IOnlCgformSubscribeService iOnlCgformSubscribeService;
 
 
     @Override
@@ -170,6 +179,26 @@ public class DocumentSearchServiceImpl implements IDocumentSearchService {
         List<Map<String, Object>> mapList = new ArrayList<>();
         for (Map<String, Object> stringObjectMap : list) {
             Map<String, Object> mapTemp = (Map<String, Object>) stringObjectMap.get("_source");
+            //处理收藏(已收藏为1,未收藏为0)
+            LambdaQueryWrapper<OnlCgformCollection> lambdaQueryWrapper = new LambdaQueryWrapper<>();
+            lambdaQueryWrapper.in(OnlCgformCollection::getDocumentId, mapTemp.get("id"));
+            List<OnlCgformCollection> collectList = iOnlCgformCollectionService.list(lambdaQueryWrapper);
+            if(collectList.size() > 0){
+                mapTemp.put("collectFlag","1");
+            }else{
+                mapTemp.put("collectFlag","0");
+            }
+
+            //处理订阅
+            LambdaQueryWrapper<OnlCgformSubscribe> lambdaQueryWrapperTemp = new LambdaQueryWrapper<>();
+            lambdaQueryWrapperTemp.in(OnlCgformSubscribe::getDocumentId, mapTemp.get("id"));
+            List<OnlCgformSubscribe> readList = iOnlCgformSubscribeService.list(lambdaQueryWrapperTemp);
+            if(readList.size() > 0){
+                mapTemp.put("subscribeFlag","1");
+            }else{
+                mapTemp.put("subscribeFlag","0");
+            }
+
             //格式化时间字段
             solveTime(mapTemp);
             mapList.add(mapTemp);