From 4a6a157465c21eadaeab6e6e03053111b6596fd3 Mon Sep 17 00:00:00 2001
From: zhaoxiaohao <913652501@qq.com>
Date: Fri, 24 Apr 2020 11:28:51 +0800
Subject: [PATCH] 修改@Loginuser注解
---
kidgrow-commons/kidgrow-common-spring-boot-starter/src/main/java/com/kidgrow/common/resolver/TokenArgumentResolver.java | 26 +++++++++++++++++---------
1 files changed, 17 insertions(+), 9 deletions(-)
diff --git a/kidgrow-commons/kidgrow-common-spring-boot-starter/src/main/java/com/kidgrow/common/resolver/TokenArgumentResolver.java b/kidgrow-commons/kidgrow-common-spring-boot-starter/src/main/java/com/kidgrow/common/resolver/TokenArgumentResolver.java
index d5b321a..0cad4b3 100644
--- a/kidgrow-commons/kidgrow-common-spring-boot-starter/src/main/java/com/kidgrow/common/resolver/TokenArgumentResolver.java
+++ b/kidgrow-commons/kidgrow-common-spring-boot-starter/src/main/java/com/kidgrow/common/resolver/TokenArgumentResolver.java
@@ -1,9 +1,11 @@
package com.kidgrow.common.resolver;
import cn.hutool.core.util.StrUtil;
+import com.alibaba.fastjson.JSON;
import com.kidgrow.common.annotation.LoginUser;
import com.kidgrow.common.constant.SecurityConstants;
import com.kidgrow.common.feign.UserService;
+import com.kidgrow.common.model.SysOrganization;
import com.kidgrow.common.model.SysRole;
import com.kidgrow.common.model.SysUser;
import com.kidgrow.common.utils.StringUtils;
@@ -65,6 +67,7 @@
String userId = request.getHeader(SecurityConstants.USER_ID_HEADER);
String username = request.getHeader(SecurityConstants.USER_HEADER);
String roles = request.getHeader(SecurityConstants.ROLE_HEADER);
+ String organizations = request.getHeader(SecurityConstants.USER_ORGS_HEADER);
if (StrUtil.isBlank(username)) {
log.warn("resolveArgument error username is empty");
return null;
@@ -76,16 +79,21 @@
user = new SysUser();
user.setId(Long.valueOf(userId));
user.setUsername(username);
+// List<SysRole> sysRoleList = new ArrayList<>();
+ if(StringUtils.isNotBlank(roles)){
+ List<SysRole> sysRoles = JSON.parseArray(roles, SysRole.class);
+// Arrays.stream(roles.split(",")).forEach(role -> {
+// SysRole sysRole = new SysRole();
+// sysRole.setCode(role);
+// sysRoleList.add(sysRole);
+// });
+ user.setRoles(sysRoles);
+ }
+ if(organizations!=null){
+ List<SysOrganization> sysOrganizations = JSON.parseArray(organizations, SysOrganization.class);
+ user.setOrganizations(sysOrganizations);
+ }
}
- List<SysRole> sysRoleList = new ArrayList<>();
- if(StringUtils.isNotBlank(roles)){
- Arrays.stream(roles.split(",")).forEach(role -> {
- SysRole sysRole = new SysRole();
- sysRole.setCode(role);
- sysRoleList.add(sysRole);
- });
- }
- user.setRoles(sysRoleList);
return user;
}
}
--
Gitblit v1.8.0