From 810d81a44d78bc3e7c42c2f116cc9968ab04baef Mon Sep 17 00:00:00 2001
From: Pu Zhibing <393733352@qq.com>
Date: 星期五, 07 二月 2025 15:57:28 +0800
Subject: [PATCH] 修改门店首页统计bug

---
 ruoyi-service/ruoyi-account/src/main/java/com/ruoyi/account/controller/AppUserController.java |   26 ++++++++++++++++++++++----
 1 files changed, 22 insertions(+), 4 deletions(-)

diff --git a/ruoyi-service/ruoyi-account/src/main/java/com/ruoyi/account/controller/AppUserController.java b/ruoyi-service/ruoyi-account/src/main/java/com/ruoyi/account/controller/AppUserController.java
index ac4feb3..e9b6927 100644
--- a/ruoyi-service/ruoyi-account/src/main/java/com/ruoyi/account/controller/AppUserController.java
+++ b/ruoyi-service/ruoyi-account/src/main/java/com/ruoyi/account/controller/AppUserController.java
@@ -993,13 +993,27 @@
         if (cd.getStartTime().isAfter(cd.getEndTime())) {
             return R.fail("开始时间不能大于结束时间");
         }
+        Long userid = tokenService.getLoginUser().getUserid();
+        SysUser sysUser = sysUserClient.getSysUser(userid).getData();
         Map<String, Map<Integer, BigDecimal>> dailyVipCommissions = new TreeMap<>();
         getDatesBetween(cd.getStartTime(), cd.getEndTime()).forEach(date -> {
             for (int i = 1; i < 8; i++){
-                List<BalanceChangeRecord> list = balanceChangeRecordService.list(new LambdaQueryWrapper<BalanceChangeRecord>()
+                LambdaQueryWrapper<BalanceChangeRecord> queryWrapper = new LambdaQueryWrapper<BalanceChangeRecord>()
                         .between(BalanceChangeRecord::getCreateTime, date, date.plusDays(1))
-                        .eq(BalanceChangeRecord::getChangeType,4)
-                        .eq(BalanceChangeRecord::getVipId, i));
+                        .eq(BalanceChangeRecord::getChangeType, 4)
+                        .eq(BalanceChangeRecord::getVipId, i);
+                if(sysUser.getRoleType() == 2){
+                    Integer objectId = sysUser.getObjectId();
+                    List<AppUser> list = appUserService.list(new LambdaQueryWrapper<AppUser>().eq(AppUser::getShopId, objectId).eq(AppUser::getDelFlag, 0).eq(AppUser::getStatus, 1));
+                    List<Long> appUserIds = list.stream().map(AppUser::getId).collect(Collectors.toList());
+                    Set<Long> data = orderClient.getAppUserByShoppingShop(objectId).getData();
+                    appUserIds.addAll(data);
+                    if(appUserIds.size() == 0){
+                        appUserIds.add(-1L);
+                    }
+                    queryWrapper.in(BalanceChangeRecord::getAppUserId, appUserIds);
+                }
+                List<BalanceChangeRecord> list = balanceChangeRecordService.list(queryWrapper);
                 BigDecimal reduce = list.stream().map(BalanceChangeRecord::getChangeAmount).reduce(BigDecimal.ZERO, BigDecimal::add);
                 String dateKey = date.format(DateTimeFormatter.ofPattern("yyyy-MM-dd"));
                 dailyVipCommissions.computeIfAbsent(dateKey, k -> new HashMap<>())
@@ -1007,7 +1021,11 @@
             }
 
             for (int i = 1; i < 4; i++){
-                R<List<ShopBalanceStatement>> r = shopBalanceStatementClient.getShopBalanceStatementList2(i, date);
+                Integer objectId = 0;
+                if(sysUser.getRoleType() == 2){
+                    objectId = sysUser.getObjectId();
+                }
+                R<List<ShopBalanceStatement>> r = shopBalanceStatementClient.getShopBalanceStatementList2(i, objectId, date);
                 if (!CollectionUtils.isEmpty(r.getData())){
                     List<ShopBalanceStatement> list = r.getData();
                     BigDecimal reduce = list.stream().map(ShopBalanceStatement::getVariableAmount).reduce(BigDecimal.ZERO, BigDecimal::add);

--
Gitblit v1.7.1