wulin/wl_management
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

增加获取用户日志

Browse Source
This commit is contained in:
wulin 2020-11-20 16:54:24 +08:00
parent 921fc6d7fb
commit a342994b3f
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/main/java/com/lz/modules/sys/oauth2/OAuth2Realm.java
View File

@ -14,6 +14,7 @@ import com.lz.modules.sys.entity.SysUserEntity;
import com.lz.modules.sys.entity.SysUserEntityDto;
import com.lz.modules.sys.entity.SysUserTokenEntity;
import com.lz.modules.sys.service.ShiroService;
import lombok.extern.slf4j.Slf4j;
import org.apache.shiro.authc.*;
import org.apache.shiro.authz.AuthorizationInfo;
import org.apache.shiro.authz.SimpleAuthorizationInfo;
@ -30,6 +31,7 @@ import java.util.Set;
* @author Mark sunlightcs@gmail.com
*/
@Component
@Slf4j
public class OAuth2Realm extends AuthorizingRealm {
@Autowired
private ShiroService shiroService;
@ -74,8 +76,10 @@ public class OAuth2Realm extends AuthorizingRealm {
SysUserEntity user = null;
StaffEntity staffEntity = staffService.selectStaffById(tokenEntity.getUserId());
if(tokenEntity.getType() == 0){ //如果是系统用户
log.info("token判定为系统用户");
user = shiroService.queryUser(tokenEntity.getUserId());
if(staffEntity != null){
log.info("查询到系统用户信息");
user.setAvatar(staffEntity.getAvatar());
}