From 39969f50f23ae92c5fc25005d8d367eea0570c31 Mon Sep 17 00:00:00 2001
From: jiangping <jp@doumee.com>
Date: 星期一, 18 十二月 2023 18:21:31 +0800
Subject: [PATCH] Merge remote-tracking branch 'origin/master'

---
 server/dmvisit_service/src/main/java/com/doumee/service/business/impl/DeviceRoleServiceImpl.java |   20 ++++++++++++++++----
 1 files changed, 16 insertions(+), 4 deletions(-)

diff --git a/server/dmvisit_service/src/main/java/com/doumee/service/business/impl/DeviceRoleServiceImpl.java b/server/dmvisit_service/src/main/java/com/doumee/service/business/impl/DeviceRoleServiceImpl.java
index 02c9be2..8c9e5ba 100644
--- a/server/dmvisit_service/src/main/java/com/doumee/service/business/impl/DeviceRoleServiceImpl.java
+++ b/server/dmvisit_service/src/main/java/com/doumee/service/business/impl/DeviceRoleServiceImpl.java
@@ -15,12 +15,14 @@
 import com.doumee.dao.business.DeviceRoleMapper;
 import com.doumee.dao.business.model.Device;
 import com.doumee.dao.business.model.DeviceRole;
+import com.doumee.dao.business.model.Member;
 import com.doumee.dao.web.response.DeviceRoleVO;
 import com.doumee.service.business.DeviceRoleService;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
+import com.doumee.service.business.MemberService;
 import io.swagger.models.auth.In;
 import org.apache.commons.collections.CollectionUtils;
 import org.apache.commons.lang3.StringUtils;
@@ -45,6 +47,8 @@
 
     @Autowired
     private DeviceRoleMapper deviceRoleMapper;
+    @Autowired
+    private MemberService memberService;
 
 
     @Override
@@ -235,9 +239,6 @@
         if (pageWrap.getModel().getStatus() != null) {
             queryWrapper.lambda().eq(DeviceRole::getStatus, pageWrap.getModel().getStatus());
         }
-
-        queryWrapper.select("t_aa.*," +
-                "(SELECT count(DISTINCT `KEY`) FROM `test` WHERE `test`.`key` = `t_aa`.id ) ");
         for(PageWrap.SortData sortData: pageWrap.getSorts()) {
             if (sortData.getDirection().equalsIgnoreCase(PageWrap.DESC)) {
                 queryWrapper.orderByDesc(sortData.getProperty());
@@ -245,7 +246,18 @@
                 queryWrapper.orderByAsc(sortData.getProperty());
             }
         }
-        return PageData.from(deviceRoleMapper.selectPage(page, queryWrapper));
+        IPage<DeviceRole> newPage = deviceRoleMapper.selectPage(page, queryWrapper);
+        if(null != newPage && newPage.getRecords().size() > 0) {
+            List<DeviceRole> deviceRoleList = newPage.getRecords();
+            for (DeviceRole dto:deviceRoleList) {
+                Member member = new Member();
+                member.setIsdeleted(Constants.ZERO);
+                member.setRoleId(dto.getId());
+                dto.setMemberNum((int) memberService.count(member));
+            }
+            newPage.setRecords(deviceRoleList);
+        }
+        return PageData.from(newPage);
     }
 
     @Override

--
Gitblit v1.9.3