From 67485b16168ee64a57296eefd8f74e7a07374e06 Mon Sep 17 00:00:00 2001
From: zhanzhiqin <sandu123456>
Date: 星期四, 17 三月 2022 18:21:37 +0800
Subject: [PATCH] fix

---
 ximon-admin/src/main/java/com/sandu/ximon/admin/controller/ClientController.java |   26 ++++++++++++++++++++++----
 ximon-admin/src/main/java/com/sandu/ximon/admin/dto/ClientDto.java               |   12 ++++++++++++
 2 files changed, 34 insertions(+), 4 deletions(-)

diff --git a/ximon-admin/src/main/java/com/sandu/ximon/admin/controller/ClientController.java b/ximon-admin/src/main/java/com/sandu/ximon/admin/controller/ClientController.java
index 19fef03..071225b 100644
--- a/ximon-admin/src/main/java/com/sandu/ximon/admin/controller/ClientController.java
+++ b/ximon-admin/src/main/java/com/sandu/ximon/admin/controller/ClientController.java
@@ -1,5 +1,6 @@
 package com.sandu.ximon.admin.controller;
 
+import cn.hutool.core.bean.BeanUtil;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.github.pagehelper.PageHelper;
@@ -12,13 +13,16 @@
 import com.sandu.common.security.token.TokenProvider;
 import com.sandu.common.util.IpUtil;
 import com.sandu.common.util.ResponseUtil;
+import com.sandu.ximon.admin.dto.ClientDto;
 import com.sandu.ximon.admin.param.AddClientPrarm;
 import com.sandu.ximon.admin.param.AdminLoginParam;
 import com.sandu.ximon.admin.param.UpdateClientPrarm;
 import com.sandu.ximon.admin.security.SecurityUtils;
+import com.sandu.ximon.admin.service.ClientRoleRelationService;
 import com.sandu.ximon.admin.service.ClientService;
 import com.sandu.ximon.admin.service.RoleService;
 import com.sandu.ximon.dao.domain.Client;
+import com.sandu.ximon.dao.domain.ClientRoleRelation;
 import com.sandu.ximon.dao.enums.AdministratorEnums;
 import lombok.AllArgsConstructor;
 import org.springframework.security.core.GrantedAuthority;
@@ -27,9 +31,7 @@
 import org.springframework.web.bind.annotation.*;
 
 import java.time.LocalDateTime;
-import java.util.Collection;
-import java.util.HashMap;
-import java.util.Map;
+import java.util.*;
 import java.util.stream.Collectors;
 
 /**
@@ -44,6 +46,7 @@
     private final ClientService clientService;
     private final PasswordEncoder passwordEncoder;
     private final RoleService roleService;
+    private final ClientRoleRelationService clientRoleRelationService;
     private final TokenProvider tokenProvider;
 
     @PostMapping("/add")
@@ -89,7 +92,22 @@
                     .or(clientLambdaQueryWrapper -> clientLambdaQueryWrapper.like(Client::getMobile, keyword))
                     .or(clientLambdaQueryWrapper -> clientLambdaQueryWrapper.like(Client::getLinkMan, keyword));
         }
-        return ResponseUtil.successPage(clientService.list(wrapper));
+        List<Client> clientList = clientService.list(wrapper);
+        List<ClientDto> clientDtoList = new ArrayList<>();
+
+        for (Client demo : clientList) {
+            ClientDto clientDto = new ClientDto();
+            BeanUtil.copyProperties(demo, clientDto);
+            ClientRoleRelation one = clientRoleRelationService.getOne(Wrappers.lambdaQuery(ClientRoleRelation.class)
+                    .eq(ClientRoleRelation::getClientId, demo.getId()));
+
+            if (one != null) {
+                clientDto.setRoleId(one.getRoleId());
+            }
+            clientDtoList.add(clientDto);
+
+        }
+        return ResponseUtil.successPage(clientDtoList);
     }
 
     @AnonymousAccess
diff --git a/ximon-admin/src/main/java/com/sandu/ximon/admin/dto/ClientDto.java b/ximon-admin/src/main/java/com/sandu/ximon/admin/dto/ClientDto.java
new file mode 100644
index 0000000..84d361b
--- /dev/null
+++ b/ximon-admin/src/main/java/com/sandu/ximon/admin/dto/ClientDto.java
@@ -0,0 +1,12 @@
+package com.sandu.ximon.admin.dto;
+
+import com.sandu.ximon.dao.domain.Client;
+import lombok.Data;
+
+@Data
+public class ClientDto extends Client {
+    /**
+     * 瑙掕壊ID
+     */
+    private Long roleId;
+}

--
Gitblit v1.9.3