添加宜宾环境

base-manager/src/main/java/com/mortals/xhx/module/dept/service/impl/DeptServiceImpl.java

@@ -18,7 +18,10 @@ import com.mortals.xhx.module.dept.service.DeptService;
 import com.mortals.xhx.module.matters.model.MattersDeptEntity;
 import com.mortals.xhx.module.matters.model.MattersDeptQuery;
 import com.mortals.xhx.module.matters.service.MattersDeptService;
+import com.mortals.xhx.module.site.model.SiteBusinessEntity;
+import com.mortals.xhx.module.site.model.SiteBusinessQuery;
 import com.mortals.xhx.module.site.model.SiteEntity;
+import com.mortals.xhx.module.site.service.SiteBusinessService;
 import com.mortals.xhx.module.site.service.SiteService;
 import com.mortals.xhx.module.window.model.WindowBusinessEntity;
 import com.mortals.xhx.module.window.model.WindowBusinessQuery;
@@ -57,6 +60,8 @@ public class DeptServiceImpl extends AbstractCRUDCacheServiceImpl<DeptDao, DeptE
     @Autowired
     private BusinessService businessService;
     @Autowired
+    private SiteBusinessService siteBusinessService;
+    @Autowired
     private WindowBusinessService windowBusinessService;
 
 
@@ -136,8 +141,12 @@ public class DeptServiceImpl extends AbstractCRUDCacheServiceImpl<DeptDao, DeptE
         windowBusinessQuery.setWindowIdList(windowEntities.stream().map(WindowEntity::getId).collect(Collectors.toList()));
         List<WindowBusinessEntity> windowBusinessEntities = windowBusinessService.find(windowBusinessQuery);
 
+        SiteBusinessQuery siteBusinessQuery = new SiteBusinessQuery();
+        siteBusinessQuery.setBusinessIdList(windowBusinessEntities.stream().map(WindowBusinessEntity::getSiteBusinessId).collect(Collectors.toList()));
+
+        List<SiteBusinessEntity> siteBusinessEntities = siteBusinessService.find(siteBusinessQuery);
         BusinessQuery businessQuery = new BusinessQuery();
-        businessQuery.setIdList(windowBusinessEntities.stream().map(WindowBusinessEntity::getSiteBusinessId).collect(Collectors.toList()));
+        businessQuery.setIdList(siteBusinessEntities.stream().map(SiteBusinessEntity::getBusinessId).collect(Collectors.toList()));
         List<BusinessEntity> businessEntities = businessService.find(businessQuery);
 
         return Rest.ok(businessEntities);