FengChaoYu il y a 1 mois
Parent
commit
9569d99fc2

+ 15 - 0
mall-server-api/src/main/java/com/gree/mall/manager/logic/contract/WebsitContractSignLogic.java

@@ -726,10 +726,25 @@ public class WebsitContractSignLogic {
     }
 
     public WebsitContractSignRecordCount count() {
+        AdminUserCom adminUser = commonLogic.getAdminUser();
+        String websitId = null;
+        if (adminUser.getType().equals(0) && adminUser.getAdminWebsitIds().size() == 1) {
+            final AdminWebsit websit = adminWebsitService.lambdaQuery()
+                    .eq(AdminWebsit::getCompanyWechatId, adminUser.getCompanyWechatId())
+                    .eq(AdminWebsit::getWebsitId, adminUser.getAdminWebsitIds().get(0))
+                    .one();
+            if (websit.getLevel() == 2) {
+                websitId = websit.getWebsitId();
+            }
+        }
         final Integer count = websitContractSignRecordService.lambdaQuery()
+                .eq(StringUtils.isNotBlank(websitId), WebsitContractSignRecord::getWebsitId, websitId)
+                .in(StringUtils.isBlank(websitId) && CollectionUtil.isNotEmpty(adminUser.getAdminWebsitIds()), WebsitContractSignRecord::getWebsitId, adminUser.getAdminWebsitIds())
                 .eq(WebsitContractSignRecord::getStatus, SignContractStatusEnum.WAIT_SIGN.getKey())
                 .count();
         final Integer count1 = websitContractSignRecordService.lambdaQuery()
+                .eq(StringUtils.isNotBlank(websitId), WebsitContractSignRecord::getWebsitId, websitId)
+                .in(StringUtils.isBlank(websitId) && CollectionUtil.isNotEmpty(adminUser.getAdminWebsitIds()), WebsitContractSignRecord::getWebsitId, adminUser.getAdminWebsitIds())
                 .eq(WebsitContractSignRecord::getStatus, SignContractStatusEnum.WAIT.getKey())
                 .count();
         WebsitContractSignRecordCount recordCount = new WebsitContractSignRecordCount();