文档库--搜索中心添加收藏和订阅

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);