From 7b8d89b03cbf3b8cc98419462c2e616dcb70a5fe Mon Sep 17 00:00:00 2001
From: liuhaonan <konodioda2333@vip.qq.com>
Date: 星期三, 13 四月 2022 11:48:28 +0800
Subject: [PATCH] 设备汇总

---
 ximon-admin/src/main/java/com/sandu/ximon/admin/controller/AdminController.java |   62 +++++++++++++++++++++++--------
 1 files changed, 46 insertions(+), 16 deletions(-)

diff --git a/ximon-admin/src/main/java/com/sandu/ximon/admin/controller/AdminController.java b/ximon-admin/src/main/java/com/sandu/ximon/admin/controller/AdminController.java
index 7cce0e2..f6f7cf4 100644
--- a/ximon-admin/src/main/java/com/sandu/ximon/admin/controller/AdminController.java
+++ b/ximon-admin/src/main/java/com/sandu/ximon/admin/controller/AdminController.java
@@ -1,5 +1,6 @@
 package com.sandu.ximon.admin.controller;
 
+import cn.hutool.core.bean.BeanUtil;
 import cn.hutool.core.collection.CollectionUtil;
 import cn.hutool.core.map.MapBuilder;
 import cn.hutool.core.map.MapUtil;
@@ -9,6 +10,7 @@
 import com.sandu.common.domain.ResponseVO;
 import com.sandu.common.enums.AdminStatusStatus;
 import com.sandu.common.enums.ResponseStatusEnums;
+import com.sandu.common.log.Log;
 import com.sandu.common.object.BaseConditionVO;
 import com.sandu.common.security.LoginUserInfo;
 import com.sandu.common.security.annotation.AnonymousAccess;
@@ -16,16 +18,14 @@
 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.AdminDto;
+import com.sandu.ximon.admin.dto.ClientDto;
 import com.sandu.ximon.admin.param.AdminLoginParam;
 import com.sandu.ximon.admin.param.AdminParam;
 import com.sandu.ximon.admin.param.PwdParam;
 import com.sandu.ximon.admin.security.SecurityUtils;
-import com.sandu.ximon.admin.service.AdminService;
-import com.sandu.ximon.admin.service.MenuService;
-import com.sandu.ximon.admin.service.RoleService;
-import com.sandu.ximon.dao.domain.Admin;
-import com.sandu.ximon.dao.domain.Menu;
-import com.sandu.ximon.dao.domain.Role;
+import com.sandu.ximon.admin.service.*;
+import com.sandu.ximon.dao.domain.*;
 import com.sandu.ximon.dao.enums.AdministratorEnums;
 import lombok.AllArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
@@ -35,10 +35,7 @@
 import org.springframework.web.bind.annotation.*;
 
 import java.time.LocalDateTime;
-import java.util.Collection;
-import java.util.HashMap;
-import java.util.List;
-import java.util.Map;
+import java.util.*;
 import java.util.stream.Collectors;
 
 /**
@@ -56,10 +53,12 @@
     private final MenuService menuService;
     private final PasswordEncoder passwordEncoder;
     private final AdminService adminService;
+    private final ClientService clientService;
     private final RoleService roleService;
     private final TokenProvider tokenProvider;
+    private final AdminRoleRelationService adminRoleRelationService;
 
-
+    @Log("鍚庡彴鐢ㄦ埛鐧诲綍")
     @AnonymousAccess
     @PostMapping(value = "/login")
     public ResponseVO<Object> login(@RequestBody AdminLoginParam loginParam) {
@@ -74,7 +73,7 @@
             return ResponseUtil.error(ResponseStatusEnums.DISABLE_ACCOUNT.getCode(), ResponseStatusEnums.DISABLE_ACCOUNT.getMessage());
         }
 
-        Collection<GrantedAuthority> grantedAuthorities = roleService.mapToGrantedAuthorities(admin.getId());
+        Collection<GrantedAuthority> grantedAuthorities = roleService.mapToGrantedAuthorities(admin.getId(), true);
 
         LoginUserInfo loginUserInfo = new LoginUserInfo();
         loginUserInfo.setUserId(admin.getId());
@@ -102,16 +101,28 @@
         return ResponseUtil.success(authInfo);
     }
 
+    @Log("鍚庡彴鐢ㄦ埛淇℃伅")
     @GetMapping(value = "/info")
     public ResponseVO<Object> getUserInfo() {
         Long userId = SecurityUtils.getUserId();
-        Admin admin = adminService.getById(userId);
-        List<Role> roles = roleService.listByAdminId(admin.getId());
+        List<Role> roles;
+        MapBuilder<Object, Object> builder;
+        if (AdministratorEnums.ADMIN.equals(SecurityUtils.getAdministratorIdentity())
+                || AdministratorEnums.CUSTOMER.equals(SecurityUtils.getAdministratorIdentity())) {
+            Admin admin = adminService.getById(userId);
+            roles = roleService.listByAdminId(admin.getId());
+            builder = MapUtil.builder().put("admin", admin);
+
+        } else {
+            Client client = clientService.getById(userId);
+            roles = roleService.listByClientId(client.getId());
+            builder = MapUtil.builder().put("admin", client);
+
+        }
         if (CollectionUtil.isEmpty(roles)) {
             return ResponseUtil.fail("鐢ㄦ埛娌℃湁瑙掕壊锛屾棤娉曡幏鍙栨暟鎹�");
         }
         List<Menu> menus = menuService.listByRoleIds(roles.stream().map(Role::getId).collect(Collectors.toList()));
-        MapBuilder<Object, Object> builder = MapUtil.builder().put("admin", admin);
         if (CollectionUtil.isNotEmpty(menus)) {
             List<Menu> collect = menus.stream().filter(menu -> menu.getType() == 0 || menu.getType() == 1).collect(Collectors.toList());
             builder.put("menus", collect);
@@ -119,6 +130,7 @@
         return ResponseUtil.success(builder.build());
     }
 
+    @Log("鍚庡彴鐢ㄦ埛鍒楄〃")
     @GetMapping(value = "/list")
     public ResponseVO<Object> listAdmin(BaseConditionVO baseConditionVO, @RequestParam(value = "keyword", required = false) String keyword) {
         PageHelper.startPage(baseConditionVO.getPageNo(), baseConditionVO.getPageSize());
@@ -130,9 +142,24 @@
                     });
         }
         List<Admin> list = adminService.list(wrapper);
-        return ResponseUtil.successPage(list);
+
+        List<AdminDto> adminDtoList = new ArrayList<>();
+        for (Admin demo : list) {
+            AdminDto adminDto = new AdminDto();
+            BeanUtil.copyProperties(demo, adminDto);
+            AdminRoleRelation one = adminRoleRelationService.getOne(Wrappers.lambdaQuery(AdminRoleRelation.class)
+                    .eq(AdminRoleRelation::getAdminId, demo.getId()));
+
+            if (one != null) {
+                adminDto.setRoleId(one.getRoleId());
+            }
+            adminDtoList.add(adminDto);
+
+        }
+        return ResponseUtil.successPage(adminDtoList);
     }
 
+    @Log("鍚庡彴鐢ㄦ埛娉ㄥ唽")
     @PostMapping("/register")
     public ResponseVO<Object> register(@Validated @RequestBody AdminParam param) {
         boolean result = adminService.register(param);
@@ -148,6 +175,7 @@
     /**
      * 淇敼绠$悊鍛樹俊鎭�
      */
+    @Log("鍚庡彴鐢ㄦ埛淇敼淇℃伅")
     @PostMapping("/update/{adminId}")
     public ResponseVO<Object> updateAdmin(@RequestBody AdminParam param, @PathVariable Long adminId) {
         boolean result = adminService.updateAdmin(adminId, param);
@@ -161,6 +189,7 @@
     /**
      * 淇敼鎴戠殑瀵嗙爜
      */
+    @Log("鍚庡彴鐢ㄦ埛淇敼瀵嗙爜")
     @PutMapping("/updateMyPassword")
     public ResponseVO<Object> updateMyPassword(@Validated @RequestBody PwdParam param) {
         boolean result = adminService.updateMyPassword(param);
@@ -181,6 +210,7 @@
 //        return ResponseUtil.success(null);
 //    }
 
+    @Log("鍚庡彴鐢ㄦ埛鍒犻櫎")
     @PostMapping("/del/{adminId}")
     public ResponseVO<Object> deleteAdmin(@PathVariable Long adminId) {
         boolean result = adminService.deleteAdmin(adminId);

--
Gitblit v1.9.3