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> mapList = new ArrayList<>(); for (Map stringObjectMap : list) { Map mapTemp = (Map) stringObjectMap.get("_source"); + //处理收藏(已收藏为1,未收藏为0) + LambdaQueryWrapper lambdaQueryWrapper = new LambdaQueryWrapper<>(); + lambdaQueryWrapper.in(OnlCgformCollection::getDocumentId, mapTemp.get("id")); + List collectList = iOnlCgformCollectionService.list(lambdaQueryWrapper); + if(collectList.size() > 0){ + mapTemp.put("collectFlag","1"); + }else{ + mapTemp.put("collectFlag","0"); + } + + //处理订阅 + LambdaQueryWrapper lambdaQueryWrapperTemp = new LambdaQueryWrapper<>(); + lambdaQueryWrapperTemp.in(OnlCgformSubscribe::getDocumentId, mapTemp.get("id")); + List readList = iOnlCgformSubscribeService.list(lambdaQueryWrapperTemp); + if(readList.size() > 0){ + mapTemp.put("subscribeFlag","1"); + }else{ + mapTemp.put("subscribeFlag","0"); + } + //格式化时间字段 solveTime(mapTemp); mapList.add(mapTemp);