站点事项增加是否已收藏字段

knowledge-base/src/main/java/com/mortals/xhx/module/site/model/vo/SiteMatterVo.java

 package com.mortals.xhx.module.site.model.vo;
 import com.mortals.framework.model.BaseEntityLong;
 import com.mortals.xhx.module.site.model.SiteMatterEntity;
+import lombok.Data;
+
 import java.util.ArrayList;
 import java.util.List;
 /**
@@ -9,16 +11,13 @@ import java.util.List;
 * @author zxfei
 * @date 2023-03-10
 */
+@Data
 public class SiteMatterVo extends BaseEntityLong {
 
     /** 材料数量 */
-    private Integer datumCount;
+    private Integer datumCount = 0;
 
-    public Integer getDatumCount() {
-        return datumCount;
-    }
+    /** 是否已添加收藏 */
+    private Integer isCollect = 0;
 
-    public void setDatumCount(Integer datumCount) {
-        this.datumCount = datumCount;
-    }
 }
\ No newline at end of file

knowledge-base/src/main/java/com/mortals/xhx/module/site/service/impl/SiteMatterServiceImpl.java

 package com.mortals.xhx.module.site.service.impl;
 import com.mortals.framework.model.PageInfo;
+import com.mortals.framework.util.DataUtil;
 import com.mortals.xhx.module.matter.model.MatterDatumEntity;
 import com.mortals.xhx.module.matter.model.MatterDatumFileEntity;
 import com.mortals.xhx.module.matter.model.MatterDatumFileQuery;
 import com.mortals.xhx.module.matter.model.MatterDatumQuery;
 import com.mortals.xhx.module.matter.service.MatterDatumService;
+import com.mortals.xhx.module.user.model.UserCollectEntity;
+import com.mortals.xhx.module.user.model.UserCollectQuery;
+import com.mortals.xhx.module.user.service.UserCollectService;
 import org.apache.commons.collections4.CollectionUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
@@ -31,6 +35,8 @@ public class SiteMatterServiceImpl extends AbstractCRUDServiceImpl<SiteMatterDao
 
     @Autowired
     private MatterDatumService matterDatumService;
+    @Autowired
+    private UserCollectService userCollectService;
 
     @Override
     protected void findAfter(SiteMatterEntity params, PageInfo pageInfo, Context context, List<SiteMatterEntity> list) throws AppException {
@@ -42,6 +48,23 @@ public class SiteMatterServiceImpl extends AbstractCRUDServiceImpl<SiteMatterDao
                 item.setDatumCount(0);
             }
         }).count();
+        if(context!=null&&context.getUser()!=null){
+            List<UserCollectEntity> userCollectEntities = userCollectService.find(new UserCollectQuery().userId(context.getUser().getId()).collectType(1));
+            Map<Long,Long> matterMap = new HashMap<>();
+            if(CollectionUtils.isNotEmpty(userCollectEntities)){
+                for(UserCollectEntity userCollectEntity:userCollectEntities){
+                    Long matterId = DataUtil.converStr2Long(userCollectEntity.getBusinessId(),0);
+                    matterMap.put(matterId,matterId);
+                }
+            }else {
+                matterMap.put(0l,0l);
+            }
+            list.stream().peek(item->{
+                if(matterMap.containsKey(item.getMatterId())){
+                    item.setIsCollect(1);
+                }
+            }).count();
+        }
         super.findAfter(params, context, list);
     }
 
@@ -55,6 +78,23 @@ public class SiteMatterServiceImpl extends AbstractCRUDServiceImpl<SiteMatterDao
                 item.setDatumCount(0);
             }
         }).count();
+        if(context!=null&&context.getUser()!=null){
+            List<UserCollectEntity> userCollectEntities = userCollectService.find(new UserCollectQuery().userId(context.getUser().getId()).collectType(1));
+            Map<Long,Long> matterMap = new HashMap<>();
+            if(CollectionUtils.isNotEmpty(userCollectEntities)){
+                for(UserCollectEntity userCollectEntity:userCollectEntities){
+                    Long matterId = DataUtil.converStr2Long(userCollectEntity.getBusinessId(),0);
+                    matterMap.put(matterId,matterId);
+                }
+            }else {
+                matterMap.put(0l,0l);
+            }
+            list.stream().peek(item->{
+                if(matterMap.containsKey(item.getMatterId())){
+                    item.setIsCollect(1);
+                }
+            }).count();
+        }
         super.findAfter(params, context, list);
     }