Skip to content
项目
群组
代码片段
帮助
当前项目
正在载入...
登录 / 注册
切换导航面板
Y
yudao-cloud
项目
项目
详情
活动
周期分析
仓库
仓库
文件
提交
分支
标签
贡献者
图表
比较
统计图
议题
0
议题
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
CI / CD
CI / CD
流水线
作业
日程
统计图
Wiki
Wiki
代码片段
代码片段
成员
成员
折叠边栏
关闭边栏
活动
图像
聊天
创建新问题
作业
提交
问题看板
Open sidebar
hblj
yudao-cloud
Commits
1b84f3f1
提交
1b84f3f1
authored
7月 26, 2022
作者:
YunaiV
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
bugfix:解决用户 Token 过期时,因为 Mono.empty() 导致返回空的 Response 的问题
上级
f1cf5f9f
隐藏空白字符变更
内嵌
并排
正在显示
1 个修改的文件
包含
19 行增加
和
5 行删除
+19
-5
TokenAuthenticationFilter.java
...ao/gateway/filter/security/TokenAuthenticationFilter.java
+19
-5
没有找到文件。
yudao-gateway/src/main/java/cn/iocoder/yudao/gateway/filter/security/TokenAuthenticationFilter.java
浏览文件 @
1b84f3f1
...
@@ -34,9 +34,19 @@ import java.util.function.Function;
...
@@ -34,9 +34,19 @@ import java.util.function.Function;
@Component
@Component
public
class
TokenAuthenticationFilter
implements
GlobalFilter
,
Ordered
{
public
class
TokenAuthenticationFilter
implements
GlobalFilter
,
Ordered
{
/**
* CommonResult<OAuth2AccessTokenCheckRespDTO> 对应的 TypeReference 结果,用于解析 checkToken 的结果
*/
private
static
final
TypeReference
<
CommonResult
<
OAuth2AccessTokenCheckRespDTO
>>
CHECK_RESULT_TYPE_REFERENCE
private
static
final
TypeReference
<
CommonResult
<
OAuth2AccessTokenCheckRespDTO
>>
CHECK_RESULT_TYPE_REFERENCE
=
new
TypeReference
<
CommonResult
<
OAuth2AccessTokenCheckRespDTO
>>()
{};
=
new
TypeReference
<
CommonResult
<
OAuth2AccessTokenCheckRespDTO
>>()
{};
/**
* 空的 LoginUser 的结果
*
* TODO 芋艿:用于解决 getLoginUser 返回 Mono.empty() 的时候,会导致后续的 flatMap 无法进行处理的问题。先暂时这么解决,寻找更优解 ing
*/
private
static
final
LoginUser
LOGIN_USER_EMPTY
=
new
LoginUser
();
private
final
WebClient
webClient
;
private
final
WebClient
webClient
;
/**
/**
...
@@ -76,14 +86,18 @@ public class TokenAuthenticationFilter implements GlobalFilter, Ordered {
...
@@ -76,14 +86,18 @@ public class TokenAuthenticationFilter implements GlobalFilter, Ordered {
}
}
// 情况二,如果有 Token 令牌,则解析对应 userId、userType、tenantId 等字段,并通过 通过 Header 转发给服务
// 情况二,如果有 Token 令牌,则解析对应 userId、userType、tenantId 等字段,并通过 通过 Header 转发给服务
return
getLoginUser
(
exchange
,
token
).
flatMap
(
user
->
{
// 重要说明:defaultIfEmpty 作用,保证 Mono.empty() 情况,可以继续执行 `flatMap 的 chain.filter(exchange)` 逻辑,避免返回给前端空的 Response!!
if
(
user
==
null
)
{
return
getLoginUser
(
exchange
,
token
).
defaultIfEmpty
(
LOGIN_USER_EMPTY
).
flatMap
(
user
->
{
// 1. 无用户,直接 filter 继续请求
if
(
user
==
LOGIN_USER_EMPTY
)
{
return
chain
.
filter
(
exchange
);
return
chain
.
filter
(
exchange
);
}
}
// 设置登录用户
// 2.1 有用户,则设置登录用户
SecurityFrameworkUtils
.
setLoginUser
(
exchange
,
user
);
SecurityFrameworkUtils
.
setLoginUser
(
exchange
,
user
);
// 将 user 并设置到 login-user 的请求头,使用 json 存储值
// 2.2 将 user 并设置到 login-user 的请求头,使用 json 存储值
ServerWebExchange
newExchange
=
exchange
.
mutate
().
request
(
builder
->
SecurityFrameworkUtils
.
setLoginUserHeader
(
builder
,
user
)).
build
();
ServerWebExchange
newExchange
=
exchange
.
mutate
()
.
request
(
builder
->
SecurityFrameworkUtils
.
setLoginUserHeader
(
builder
,
user
)).
build
();
return
chain
.
filter
(
newExchange
);
return
chain
.
filter
(
newExchange
);
});
});
}
}
...
...
编写
预览
Markdown
格式
0%
重试
或
添加新文件
添加附件
取消
您添加了
0
人
到此讨论。请谨慎行事。
请先完成此评论的编辑!
取消
请
注册
或者
登录
后发表评论