From 1db50ecdccd7a981888ee33c65e31b7f2f9f4733 Mon Sep 17 00:00:00 2001
From: MercuryZ <zdmisif@126.com>
Date: 星期四, 31 三月 2022 14:09:08 +0800
Subject: [PATCH] fix 操作日志

---
 ximon-admin/src/main/java/com/sandu/ximon/admin/controller/AdminController.java |   25 ++++++++++++++++++++-----
 1 files changed, 20 insertions(+), 5 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..0634fe4 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
@@ -9,6 +9,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;
@@ -21,9 +22,11 @@
 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.ClientService;
 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.Client;
 import com.sandu.ximon.dao.domain.Menu;
 import com.sandu.ximon.dao.domain.Role;
 import com.sandu.ximon.dao.enums.AdministratorEnums;
@@ -56,10 +59,11 @@
     private final MenuService menuService;
     private final PasswordEncoder passwordEncoder;
     private final AdminService adminService;
+    private final ClientService clientService;
     private final RoleService roleService;
     private final TokenProvider tokenProvider;
 
-
+    @Log("鍚庡彴鐢ㄦ埛鐧诲綍")
     @AnonymousAccess
     @PostMapping(value = "/login")
     public ResponseVO<Object> login(@RequestBody AdminLoginParam loginParam) {
@@ -74,7 +78,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());
@@ -105,13 +109,24 @@
     @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);

--
Gitblit v1.9.3