#

Junjie

2024-04-12 684484fb93775edcfa19cfc7a43d0a748c8362be

 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) {
@@ -57,7 +61,7 @@
        }
        String accessToken = JwtUtil.buildToken(new JwtSubject(username, user.getHostId()),
                configProperties.getTokenExpireTime(), configProperties.getTokenKey());
        userLoginService.saveAsync(user.getId(), accessToken, UserLogin.TYPE_LOGIN, null, request);
        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")