From d835d1b51f832889929cdf69010034a30ef44d02 Mon Sep 17 00:00:00 2001 From: Junjie <xjj@123> Date: 星期四, 17 十月 2024 13:57:29 +0800 Subject: [PATCH] # --- zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/system/controller/AuthController.java | 16 +++++++++++++--- 1 files changed, 13 insertions(+), 3 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..e1c1286 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,14 +8,16 @@ 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; import com.zy.asrs.wcs.utils.JwtUtil; -import com.zy.asrs.wcs.utils.Utils; +import com.zy.asrs.wcs.utils.CommonUtils; import org.springframework.security.access.prepost.PreAuthorize; import org.springframework.web.bind.annotation.*; @@ -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)); } @@ -69,7 +73,13 @@ @GetMapping("/auth/menu") public R userMenu() { List<Menu> menus = roleMenuService.listMenuByUserId(getLoginUserId(), Menu.TYPE_MENU); - return R.ok().add(Utils.toTreeData(menus, 0L, Menu::getParentId, Menu::getId, Menu::setChildren)); + return R.ok().add(CommonUtils.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')") -- Gitblit v1.9.1