From 684484fb93775edcfa19cfc7a43d0a748c8362be Mon Sep 17 00:00:00 2001
From: Junjie <xjj@123>
Date: 星期五, 12 四月 2024 10:59:30 +0800
Subject: [PATCH] #

---
 zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/system/controller/AuthController.java |   12 +++++++++++-
 1 files changed, 11 insertions(+), 1 deletions(-)

diff --git a/zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/system/controller/AuthController.java b/zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/system/controller/AuthController.java
index d2806d5..43c215f 100644
--- a/zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/system/controller/AuthController.java
+++ b/zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/system/controller/AuthController.java
@@ -8,9 +8,11 @@
 import com.zy.asrs.wcs.system.controller.param.LoginParam;
 import com.zy.asrs.wcs.system.controller.param.UpdatePasswordParam;
 import com.zy.asrs.wcs.system.controller.result.LoginResult;
+import com.zy.asrs.wcs.system.entity.Host;
 import com.zy.asrs.wcs.system.entity.Menu;
 import com.zy.asrs.wcs.system.entity.User;
 import com.zy.asrs.wcs.system.entity.UserLogin;
+import com.zy.asrs.wcs.system.service.HostService;
 import com.zy.asrs.wcs.system.service.RoleMenuService;
 import com.zy.asrs.wcs.system.service.UserLoginService;
 import com.zy.asrs.wcs.system.service.UserService;
@@ -40,6 +42,8 @@
     private UserLoginService userLoginService;
     @Resource
     private RoleMenuService roleMenuService;
+    @Resource
+    private HostService hostService;
 
     @PostMapping("/login")
     public R login(@RequestBody LoginParam param, HttpServletRequest request) {
@@ -55,9 +59,9 @@
         if (!userService.comparePassword(user.getPassword(), param.getPassword())) {
             return R.error("瀵嗙爜閿欒");
         }
-        userLoginService.saveAsync(user.getId(), UserLogin.TYPE_LOGIN, null, hostId, request);
         String accessToken = JwtUtil.buildToken(new JwtSubject(username, user.getHostId()),
                 configProperties.getTokenExpireTime(), configProperties.getTokenKey());
+        userLoginService.saveAsync(user.getId(), accessToken, UserLogin.TYPE_LOGIN, hostId, null, request);
         return R.ok("鐧诲綍鎴愬姛").add(new LoginResult(accessToken, user));
     }
 
@@ -72,6 +76,12 @@
         return R.ok().add(Utils.toTreeData(menus, 0L, Menu::getParentId, Menu::getId, Menu::setChildren));
     }
 
+    @GetMapping("/auth/host")
+    public R authHost() {
+        List<Host> list = hostService.list();
+        return R.ok().add(list);
+    }
+
     @PreAuthorize("hasAuthority('sys:auth:user')")
     @OperationLog
     @PutMapping("/auth/user")

--
Gitblit v1.9.1