TokenController.java
2.46 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
package com.huaheng.auth.controller;
import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import com.huaheng.common.core.utils.StringUtils;
import com.huaheng.common.log.enums.BusinessType;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.springframework.web.bind.annotation.*;
import com.huaheng.auth.form.LoginBody;
import com.huaheng.auth.service.SysLoginService;
import com.huaheng.common.core.domain.R;
import com.huaheng.common.security.service.TokenService;
import com.huaheng.system.api.model.LoginUser;
import com.huaheng.common.log.annotation.Log;
/**
* token 控制
*/
@RestController
@Api(value = "Token", tags="Token")
public class TokenController {
@Resource
private TokenService tokenService;
@Resource
private SysLoginService sysLoginService;
@ApiOperation(value = "用户登录")
@PostMapping("login")
@Log(title = "用户登录", businessType= BusinessType.GRANT)
public R<?> login(@RequestBody LoginBody form) {
// 用户登录
LoginUser userInfo = sysLoginService.login(form.getUsername(), form.getPassword(), form.getWarehouseCode());
// 获取登录token
return R.ok(tokenService.createToken(userInfo));
}
@ApiOperation(value = "退出")
@DeleteMapping("logout")
public R<?> logout(HttpServletRequest request) {
LoginUser loginUser = tokenService.getLoginUser(request);
if (StringUtils.isNotNull(loginUser)) {
String username = loginUser.getUsername();
// 删除用户缓存记录
tokenService.delLoginUser(loginUser.getToken());
// 记录用户退出日志
sysLoginService.logout(username);
}
return R.ok();
}
@ApiOperation(value = "刷新token")
@PostMapping("refresh")
public R<?> refresh(HttpServletRequest request) {
LoginUser loginUser = tokenService.getLoginUser(request);
if (StringUtils.isNotNull(loginUser)) {
// 刷新令牌有效期
return R.ok(tokenService.refreshToken(loginUser));
}
return R.ok();
}
/**
* 切换仓库
* @param warehouseCode 仓库编码
* @return
*/
@PostMapping("/switchWarehouse")
public R<?> switchWarehouse(@RequestParam("warehouseCode") String warehouseCode) {
LoginUser userInfo = sysLoginService.switchWarehouse(warehouseCode);
return R.ok(tokenService.createToken(userInfo));
}
}