|
|
@@ -22,24 +22,36 @@ public class BaseController {
|
|
|
@Autowired
|
|
|
RedissonClient redissonClient;
|
|
|
|
|
|
- public String getUserTokenString(){
|
|
|
+ private String getToken(){
|
|
|
ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
|
|
|
+ if(attributes == null){
|
|
|
+ //内部调用的请求
|
|
|
+ return null;
|
|
|
+ }
|
|
|
HttpServletRequest request = attributes.getRequest();
|
|
|
- String authorizationHeader = request.getHeader("Authorization");
|
|
|
- if(Objects.isNull(authorizationHeader) || Objects.isNull(redissonClient.getBucket(authorizationHeader).get())){
|
|
|
+
|
|
|
+ return request.getHeader("Authorization");
|
|
|
+ }
|
|
|
+
|
|
|
+ private void valiedToken(String token){
|
|
|
+ if (Objects.isNull(token) || Objects.isNull(redissonClient.getBucket(token).get())) {
|
|
|
throw new TokenExpireException("用户登录信息过期");
|
|
|
}
|
|
|
- String userJsonStr = redissonClient.getBucket(authorizationHeader).get().toString();
|
|
|
- return userJsonStr;
|
|
|
+ }
|
|
|
+
|
|
|
+ public String getUserTokenString(){
|
|
|
+ String authorizationHeader = getToken();
|
|
|
+ valiedToken(authorizationHeader);
|
|
|
+ return redissonClient.getBucket(authorizationHeader).get().toString();
|
|
|
}
|
|
|
|
|
|
public String getUserName(){
|
|
|
- ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
|
|
|
- HttpServletRequest request = attributes.getRequest();
|
|
|
- String authorizationHeader = request.getHeader("Authorization");
|
|
|
- if(Objects.isNull(authorizationHeader) || Objects.isNull(redissonClient.getBucket(authorizationHeader).get())){
|
|
|
- throw new TokenExpireException("用户登录信息过期");
|
|
|
+ String authorizationHeader = getToken();
|
|
|
+ if(authorizationHeader == null){
|
|
|
+ //如果是内部请求的话 返回空值
|
|
|
+ return "";
|
|
|
}
|
|
|
+ valiedToken(authorizationHeader);
|
|
|
String userJsonStr = redissonClient.getBucket(authorizationHeader).get().toString();
|
|
|
JSONObject userInfo = JSONObject.parseObject(userJsonStr);
|
|
|
return userInfo.getString("username");
|
|
|
@@ -50,12 +62,12 @@ public class BaseController {
|
|
|
* @return
|
|
|
*/
|
|
|
public String getUserLoginSystem(){
|
|
|
- ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
|
|
|
- HttpServletRequest request = attributes.getRequest();
|
|
|
- String authorizationHeader = request.getHeader("Authorization");
|
|
|
- if(Objects.isNull(authorizationHeader) || Objects.isNull(redissonClient.getBucket(authorizationHeader).get())){
|
|
|
- throw new TokenExpireException("用户登录信息过期");
|
|
|
+ String authorizationHeader = getToken();
|
|
|
+ if(authorizationHeader == null){
|
|
|
+ //如果是内部请求的话 返回空值
|
|
|
+ return "";
|
|
|
}
|
|
|
+ valiedToken(authorizationHeader);
|
|
|
String userJsonStr = redissonClient.getBucket(authorizationHeader).get().toString();
|
|
|
JSONObject userInfo = JSONObject.parseObject(userJsonStr);
|
|
|
return userInfo.getString("system");
|
|
|
@@ -66,12 +78,13 @@ public class BaseController {
|
|
|
* @return
|
|
|
*/
|
|
|
public List<String> getUserPerms(){
|
|
|
- ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
|
|
|
- HttpServletRequest request = attributes.getRequest();
|
|
|
- String authorizationHeader = request.getHeader("Authorization");
|
|
|
- if(Objects.isNull(authorizationHeader) || Objects.isNull(redissonClient.getBucket(authorizationHeader).get())){
|
|
|
- throw new TokenExpireException("用户登录信息过期");
|
|
|
+ String authorizationHeader = getToken();
|
|
|
+ if(authorizationHeader == null){
|
|
|
+ //如果是内部请求的话 返回空值
|
|
|
+ return new ArrayList<>();
|
|
|
}
|
|
|
+ valiedToken(authorizationHeader);
|
|
|
+
|
|
|
List<String> userPerms = new ArrayList<>();
|
|
|
String userJsonStr = redissonClient.getBucket(authorizationHeader).get().toString();
|
|
|
JSONObject userInfo = JSONObject.parseObject(userJsonStr);
|
|
|
@@ -87,24 +100,24 @@ public class BaseController {
|
|
|
* @return
|
|
|
*/
|
|
|
public String getUserDeptId(){
|
|
|
- ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
|
|
|
- HttpServletRequest request = attributes.getRequest();
|
|
|
- String authorizationHeader = request.getHeader("Authorization");
|
|
|
- if(Objects.isNull(authorizationHeader) || Objects.isNull(redissonClient.getBucket(authorizationHeader).get())){
|
|
|
- throw new TokenExpireException("用户登录信息过期");
|
|
|
+ String authorizationHeader = getToken();
|
|
|
+ if(authorizationHeader == null){
|
|
|
+ //如果是内部请求的话 返回空值
|
|
|
+ return "";
|
|
|
}
|
|
|
+ valiedToken(authorizationHeader);
|
|
|
String userJsonStr = redissonClient.getBucket(authorizationHeader).get().toString();
|
|
|
JSONObject userInfo = JSONObject.parseObject(userJsonStr);
|
|
|
return userInfo.getString("deptId");
|
|
|
}
|
|
|
|
|
|
public String getUserSystemCode(){
|
|
|
- ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
|
|
|
- HttpServletRequest request = attributes.getRequest();
|
|
|
- String authorizationHeader = request.getHeader("Authorization");
|
|
|
- if(Objects.isNull(authorizationHeader) || Objects.isNull(redissonClient.getBucket(authorizationHeader).get())){
|
|
|
- throw new TokenExpireException("用户登录信息过期");
|
|
|
+ String authorizationHeader = getToken();
|
|
|
+ if(authorizationHeader == null){
|
|
|
+ //如果是内部请求的话 返回空值
|
|
|
+ return "";
|
|
|
}
|
|
|
+ valiedToken(authorizationHeader);
|
|
|
String userJsonStr = redissonClient.getBucket(authorizationHeader).get().toString();
|
|
|
JSONObject userInfo = JSONObject.parseObject(userJsonStr);
|
|
|
return userInfo.getString("system");
|
|
|
@@ -115,12 +128,12 @@ public class BaseController {
|
|
|
* @return
|
|
|
*/
|
|
|
public String getUserDataScope(){
|
|
|
- ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
|
|
|
- HttpServletRequest request = attributes.getRequest();
|
|
|
- String authorizationHeader = request.getHeader("Authorization");
|
|
|
- if(Objects.isNull(authorizationHeader) || Objects.isNull(redissonClient.getBucket(authorizationHeader).get())){
|
|
|
- throw new TokenExpireException("用户登录信息过期");
|
|
|
+ String authorizationHeader = getToken();
|
|
|
+ if(authorizationHeader == null){
|
|
|
+ //如果是内部请求的话 返回空值
|
|
|
+ return "";
|
|
|
}
|
|
|
+ valiedToken(authorizationHeader);
|
|
|
String userJsonStr = redissonClient.getBucket(authorizationHeader).get().toString();
|
|
|
JSONObject userInfo = JSONObject.parseObject(userJsonStr);
|
|
|
return userInfo.getString("dataScope");
|