添加事项同步数据

base-manager/src/main/java/com/mortals/xhx/module/site/service/impl/SiteBusinessServiceImpl.java

 package com.mortals.xhx.module.site.service.impl;
+import com.mortals.xhx.module.business.model.BusinessEntity;
+import com.mortals.xhx.module.business.model.BusinessQuery;
 import org.springframework.stereotype.Service;
 import com.mortals.framework.exception.AppException;
 import com.mortals.framework.model.Context;
@@ -22,6 +24,20 @@ import java.util.stream.Collectors;
 @Service("siteBusinessService")
 public class SiteBusinessServiceImpl extends AbstractCRUDServiceImpl<SiteBusinessDao, SiteBusinessEntity, Long> implements SiteBusinessService {
 
+
+    @Override
+    protected void findAfter(SiteBusinessEntity params, Context context, List<SiteBusinessEntity> list) throws AppException {
+        list.stream().peek(item->{
+            List<SiteBusinessEntity> childs = this.find(new SiteBusinessQuery().parentId(item.getId()));
+            if(!ObjectUtils.isEmpty(childs)){
+                item.setChildren(childs);
+            }else{
+                item.setChildren(null);
+            }
+        }).count();
+        super.findAfter(params, context, list);
+    }
+
     @Override
     protected void saveBefore(SiteBusinessEntity entity, Context context) throws AppException {
 

base-manager/src/main/java/com/mortals/xhx/module/site/web/SiteBusinessController.java

@@ -6,6 +6,8 @@ import com.mortals.framework.model.Context;
 import com.mortals.framework.util.StringUtils;
 import com.mortals.framework.web.BasePhpCRUDJsonMappingController;
 import com.mortals.xhx.base.system.param.service.ParamService;
+import com.mortals.xhx.common.code.IsBusinessEnum;
+import com.mortals.xhx.module.business.model.BusinessEntity;
 import com.mortals.xhx.module.site.model.SiteBusinessEntity;
 import com.mortals.xhx.module.site.model.SiteBusinessQuery;
 import com.mortals.xhx.module.site.model.SiteBusinessTreeSelect;
@@ -42,6 +44,12 @@ public class SiteBusinessController extends BasePhpCRUDJsonMappingController<Sit
         super.setModuleDesc("站点事项");
     }
 
+    @Override
+    protected void doListBefore(SiteBusinessEntity query, Map<String, Object> model, Context context) throws AppException {
+        query.setIsBusiness(IsBusinessEnum.一级业务.getValue());
+        super.doListBefore(query, model, context);
+    }
+
     @Override
     protected void init(Map<String, Object> model, Context context) {
         this.addDict(model, "isBusiness", paramService.getParamBySecondOrganize("SiteBusiness", "isBusiness"));
@@ -49,22 +57,6 @@ public class SiteBusinessController extends BasePhpCRUDJsonMappingController<Sit
         super.init(model, context);
     }
 
-    @Override
-    protected int doListAfter(SiteBusinessEntity query, Map<String, Object> model, Context context) throws AppException {
-        List<SiteBusinessEntity> businessEntities = (List<SiteBusinessEntity>) model.get(RESULT_KEY);
-        if (!ObjectUtils.isEmpty(businessEntities)) {
-            businessEntities.stream().peek(item -> {
-                List<SiteBusinessEntity> childs = this.service.find(new SiteBusinessQuery().parentId(item.getBusinessId()));
-                if (!ObjectUtils.isEmpty(childs)) {
-                    item.setChildren(childs);
-                } else {
-                    item.setChildren(null);
-                }
-            }).count();
-            model.put(RESULT_KEY, businessEntities);
-        }
-        return super.doListAfter(query, model, context);
-    }
 
     @PostMapping("list/exclude")
     public String excludeList(SiteBusinessForm form) {