xwl2023/Yi.Framework
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge branch 'abp' into ai-hub

Browse Source
This commit is contained in:
ccnetcore
2025-12-11 23:33:33 +08:00
parent 5ed79c6dd0 8a6e5abf48
commit cc812ba2cb
1 changed files with 20 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
Yi.Abp.Net8/src/Yi.Abp.Web/YiAbpWebModule.cs
View File

@@ -290,12 +290,20 @@ namespace Yi.Abp.Web
{
OnMessageReceived = messageContext =>
{
//优先Query中获取
//优先Query中获取，再去cookies中获取
var accessToken = messageContext.Request.Query["access_token"];
if (!string.IsNullOrEmpty(accessToken))
{
messageContext.Token = accessToken;
}
else
{
if (messageContext.Request.Cookies.TryGetValue("Token", out var cookiesToken))
{
messageContext.Token = cookiesToken;
}
}
return Task.CompletedTask;
}
};
@@ -315,11 +323,19 @@ namespace Yi.Abp.Web
{
OnMessageReceived = messageContext =>
{
var refreshToken = messageContext.Request.Query["refresh_token"];
if (!string.IsNullOrEmpty(refreshToken))
var headerRefreshToken = messageContext.Request.Headers["refresh_token"];
if (!string.IsNullOrEmpty(headerRefreshToken))
{
messageContext.Token = refreshToken;
messageContext.Token = headerRefreshToken;
return Task.CompletedTask;
}
var queryRefreshToken = messageContext.Request.Query["refresh_token"];
if (!string.IsNullOrEmpty(queryRefreshToken))
{
messageContext.Token = queryRefreshToken;
}
return Task.CompletedTask;
}
};