From 1662cd2f5c686973a9a596865dfe412fc5c8ae6d Mon Sep 17 00:00:00 2001 From: Junjie <540245094@qq.com> Date: 星期三, 20 十二月 2023 16:02:15 +0800 Subject: [PATCH] # --- zy-asrs-common/src/main/java/com/zy/asrs/common/web/AuthController.java | 46 ++++++++++++++++++++++++++++++++++++++++------ 1 files changed, 40 insertions(+), 6 deletions(-) diff --git a/zy-asrs-common/src/main/java/com/zy/asrs/common/web/AuthController.java b/zy-asrs-common/src/main/java/com/zy/asrs/common/web/AuthController.java index df4a984..0ec42a1 100644 --- a/zy-asrs-common/src/main/java/com/zy/asrs/common/web/AuthController.java +++ b/zy-asrs-common/src/main/java/com/zy/asrs/common/web/AuthController.java @@ -44,6 +44,8 @@ private ResourceService resourceService; @Autowired private RoleResourceService roleResourceService; + @Autowired + private HostService hostService; @RequestMapping("/login.action") @ManagerAuth(value = ManagerAuth.Auth.NONE, memo = "鐧诲綍") @@ -79,11 +81,47 @@ userLogin.setToken(token); userLogin.setCreateTime(new Date()); userLogin.setSystem(system); + if (user.getRoleId() == 2) { + userLogin.setHostId(hostService.getTop1().getId()); + } userLoginService.save(userLogin); Map<String, Object> res = new HashMap<>(); res.put("username", user.getUsername()); res.put("token", token); return R.ok(res); + } + + @RequestMapping("/show/host.action") + @ManagerAuth + public R showHosts() { + Long hostId = getHostId(); + String hostName = null; + if (hostId != null) { + Host host = hostService.getById(hostId); + if (host != null) { + hostName = host.getName(); + } + } + return R.ok().add(Cools + .add("root", getUser().getRoleId() == 2) + .add("host", getHostId() == null) + .add("hostName", hostName) + ); + } + + @RequestMapping(value = "/root/change/host/auth") + @ManagerAuth + public R rootChangeHost(@RequestParam Long hostId) { + UserLogin userLogin = userLoginService.getOne(new LambdaQueryWrapper<UserLogin>().eq(UserLogin::getUserId, getUserId()).eq(UserLogin::getSystem, String.valueOf(LoginSystemType.WMS))); + if (userLogin != null) { + userLogin.setHostId(hostId); + if (!userLoginService.updateById(userLogin)) { + throw new CoolException("淇敼鍟嗘埛澶辫触"); + } + return R.ok(); + } else { + return R.error(); + } } @RequestMapping("/code/switch.action") @@ -130,10 +168,6 @@ oneLevel = resourceService.list(new LambdaQueryWrapper<Resource>().eq(Resource::getLevel, 1).orderByAsc(Resource::getSort)); resourceWrapper = new LambdaQueryWrapper<Resource>().eq(Resource::getLevel, 2).eq(Resource::getStatus, 1).orderByAsc(Resource::getSort); } else { - // 婵�娲荤爜楠岃瘉 - if (!SystemProperties.SYSTEM_ACTIVATION) { - return R.ok(); - } oneLevel = resourceService.list(new LambdaQueryWrapper<Resource>().eq(Resource::getLevel, 1).eq(Resource::getStatus, 1).orderByAsc(Resource::getSort)); // 鑾峰彇褰撳墠鐢ㄦ埛鐨勬墍鏈変簩绾ц彍鍗� user = userService.getById(getUserId()); @@ -158,11 +192,11 @@ // 鏄惁鎷ユ湁鏌ョ湅鏉冮檺 if (getUserId() != 9527) { Resource view = resourceService.getOne(new LambdaQueryWrapper<Resource>().eq(Resource::getResourceId, resource.getId()).like(Resource::getCode, "#view")); - if (!Cools.isEmpty(view)){ + if (!Cools.isEmpty(view)) { RoleResource param = new RoleResource(); param.setResourceId(view.getId()); param.setRoleId(user.getRoleId()); - if (null == roleResourceService.getOne(new LambdaQueryWrapper<>(param))){ + if (null == roleResourceService.getOne(new LambdaQueryWrapper<>(param))) { continue; } } -- Gitblit v1.9.1