fix: 积木报表的部分请求会报错

This commit is contained in:
gaibu 2023-01-06 17:47:10 +08:00
parent 24f0e4dd1f
commit 80ab569fa7

View File

@ -11,6 +11,8 @@ import cn.iocoder.yudao.module.system.api.oauth2.dto.OAuth2AccessTokenCheckRespD
import lombok.RequiredArgsConstructor; import lombok.RequiredArgsConstructor;
import org.jeecg.modules.jmreport.api.JmReportTokenServiceI; import org.jeecg.modules.jmreport.api.JmReportTokenServiceI;
import java.util.Objects;
/** /**
* {@link JmReportTokenServiceI} 实现类提供积木报表的 Token 校验用户信息的查询等功能 * {@link JmReportTokenServiceI} 实现类提供积木报表的 Token 校验用户信息的查询等功能
* *
@ -29,8 +31,40 @@ public class JmReportTokenServiceImpl implements JmReportTokenServiceI {
*/ */
@Override @Override
public Boolean verifyToken(String token) { public Boolean verifyToken(String token) {
Long userId = SecurityFrameworkUtils.getLoginUserId();
if (!Objects.isNull(userId)) {
return true;
}
return injectUserInfoByToken(token) != null;
}
/**
* 获得用户编号
* <p>
* 虽然方法名获得的是 username实际对应到项目中是用户编号
*
* @param token JmReport 前端传递的 token
* @return 用户编号
*/
@Override
public String getUsername(String token) {
Long userId = SecurityFrameworkUtils.getLoginUserId();
if (Objects.isNull(userId)) {
LoginUser user = injectUserInfoByToken(token);
return user == null ? null : String.valueOf(user.getId());
}
return String.valueOf(userId);
}
/**
* 注入用户信息
*
* @param token token
* @return 返回 token 对应的用户信息
*/
private LoginUser injectUserInfoByToken(String token) {
if (StrUtil.isEmpty(token)) { if (StrUtil.isEmpty(token)) {
return false; return null;
} }
// TODO 如下的实现不算特别优雅主要咱是不想搞的太复杂所以参考对应的 Filter 先实现了 // TODO 如下的实现不算特别优雅主要咱是不想搞的太复杂所以参考对应的 Filter 先实现了
@ -41,7 +75,7 @@ public class JmReportTokenServiceImpl implements JmReportTokenServiceI {
try { try {
OAuth2AccessTokenCheckRespDTO accessToken = oauth2TokenApi.checkAccessToken(token); OAuth2AccessTokenCheckRespDTO accessToken = oauth2TokenApi.checkAccessToken(token);
if (accessToken == null) { if (accessToken == null) {
return false; return null;
} }
user = new LoginUser().setId(accessToken.getUserId()).setUserType(accessToken.getUserType()) user = new LoginUser().setId(accessToken.getUserId()).setUserType(accessToken.getUserType())
.setTenantId(accessToken.getTenantId()).setScopes(accessToken.getScopes()); .setTenantId(accessToken.getTenantId()).setScopes(accessToken.getScopes());
@ -49,7 +83,7 @@ public class JmReportTokenServiceImpl implements JmReportTokenServiceI {
// do nothing如果报错说明认证失败则返回 false 即可 // do nothing如果报错说明认证失败则返回 false 即可
} }
if (user == null) { if (user == null) {
return false; return null;
} }
SecurityFrameworkUtils.setLoginUser(user, WebFrameworkUtils.getRequest()); SecurityFrameworkUtils.setLoginUser(user, WebFrameworkUtils.getRequest());
@ -57,21 +91,7 @@ public class JmReportTokenServiceImpl implements JmReportTokenServiceI {
// 目的基于 LoginUser 获得到的租户编号设置到 Tenant 上下文避免查询数据库时的报错 // 目的基于 LoginUser 获得到的租户编号设置到 Tenant 上下文避免查询数据库时的报错
TenantContextHolder.setIgnore(false); TenantContextHolder.setIgnore(false);
TenantContextHolder.setTenantId(user.getTenantId()); TenantContextHolder.setTenantId(user.getTenantId());
return true; return user;
}
/**
* 获得用户编号
*
* 虽然方法名获得的是 username实际对应到项目中是用户编号
*
* @param token JmReport 前端传递的 token
* @return 用户编号
*/
@Override
public String getUsername(String token) {
Long userId = SecurityFrameworkUtils.getLoginUserId();
return userId != null ? String.valueOf(userId) : null;
} }
} }