feat: 完成双token刷新

This commit is contained in:
ccnetcore
2025-06-29 15:18:30 +08:00
parent d4f00eb89f
commit 6a58af8dfb
7 changed files with 344 additions and 326 deletions

View File

@@ -14,6 +14,6 @@ namespace Yi.Framework.Rbac.Domain.Shared.Options
public string SecurityKey { get; set; } = "892u4j1803qj23jro0fjkf8bmsdf9nb9mf92834u23jdf923jrnmvasbceqwt347562tgdhdnsv9wevbnop"; public string SecurityKey { get; set; } = "892u4j1803qj23jro0fjkf8bmsdf9nb9mf92834u23jdf923jrnmvasbceqwt347562tgdhdnsv9wevbnop";
public long ExpiresMinuteTime { get; set; } = 120; public long ExpiresSecondTime { get; set; } = 600;
} }
} }

View File

@@ -19,9 +19,9 @@ namespace Yi.Framework.Rbac.Domain.Authorization
public class RefreshTokenMiddleware : IMiddleware, ITransientDependency public class RefreshTokenMiddleware : IMiddleware, ITransientDependency
{ {
private AccountManager _accountManager; private AccountManager _accountManager;
public RefreshTokenMiddleware(AccountManager accountManager) public RefreshTokenMiddleware(AccountManager accountManager)
{ {
_accountManager = accountManager; _accountManager = accountManager;
} }
@@ -30,22 +30,34 @@ namespace Yi.Framework.Rbac.Domain.Authorization
var refreshToken = context.Request.Headers["refresh_token"].ToString(); var refreshToken = context.Request.Headers["refresh_token"].ToString();
if (!string.IsNullOrEmpty(refreshToken)) if (!string.IsNullOrEmpty(refreshToken))
{ {
//每个用户的token刷新频率可以进行控制防止刷新token当访问token使用 //Jwt鉴权失败过期了再去找刷新token进行刷新处理不用每次都去刷新
var authResult = await context.AuthenticateAsync(TokenTypeConst.Refresh); var bearerAuthResult = await context.AuthenticateAsync("Bearer");
//token鉴权刷新成功 if (!bearerAuthResult.Succeeded)
if (authResult.Succeeded)
{ {
var userId = Guid.Parse(authResult.Principal.FindFirst(AbpClaimTypes.UserId).Value.ToString()); //每个用户的token刷新频率可以进行控制防止刷新token当访问token使用
var access_Token = await _accountManager.GetTokenByUserIdAsync(userId); var authResult = await context.AuthenticateAsync(TokenTypeConst.Refresh);
var refresh_Token = _accountManager.CreateRefreshToken(userId); //token鉴权刷新成功
context.Response.Headers["access_token"] = access_Token; if (authResult.Succeeded)
context.Response.Headers["refresh_token"] = refresh_Token; {
var userId = Guid.Parse(authResult.Principal.FindFirst(AbpClaimTypes.UserId).Value.ToString());
var access_Token = await _accountManager.GetTokenByUserIdAsync(userId);
var refresh_Token = _accountManager.CreateRefreshToken(userId);
context.Response.Headers["access_token"] = access_Token;
context.Response.Headers["refresh_token"] = refresh_Token;
//请求头替换,补充后续鉴权逻辑 //请求头替换,补充后续鉴权逻辑
context.Request.Headers["Authorization"] = "Bearer " + access_Token; context.Request.Headers["Authorization"] = "Bearer " + access_Token;
}
//刷新token 与 access_token都失效了
// else
// {
//context.Response.StatusCode = StatusCodes.Status401Unauthorized;
//return;
// }
} }
} }
await next(context); await next(context);
} }
} }
@@ -57,8 +69,6 @@ namespace Yi.Framework.Rbac.Domain.Authorization
{ {
app.UseMiddleware<RefreshTokenMiddleware>(); app.UseMiddleware<RefreshTokenMiddleware>();
return app; return app;
} }
} }
}
}

View File

@@ -106,7 +106,7 @@ namespace Yi.Framework.Rbac.Domain.Managers
issuer: _jwtOptions.Issuer, issuer: _jwtOptions.Issuer,
audience: _jwtOptions.Audience, audience: _jwtOptions.Audience,
claims: claims, claims: claims,
expires: DateTime.Now.AddMinutes(_jwtOptions.ExpiresMinuteTime), expires: DateTime.Now.AddSeconds(_jwtOptions.ExpiresSecondTime),
notBefore: DateTime.Now, notBefore: DateTime.Now,
signingCredentials: creds); signingCredentials: creds);
string returnToken = new JwtSecurityTokenHandler().WriteToken(token); string returnToken = new JwtSecurityTokenHandler().WriteToken(token);
@@ -127,7 +127,7 @@ namespace Yi.Framework.Rbac.Domain.Managers
issuer: _refreshJwtOptions.Issuer, issuer: _refreshJwtOptions.Issuer,
audience: _refreshJwtOptions.Audience, audience: _refreshJwtOptions.Audience,
claims: claims, claims: claims,
expires: DateTime.Now.AddMinutes(_refreshJwtOptions.ExpiresMinuteTime), expires: DateTime.Now.AddSeconds(_refreshJwtOptions.ExpiresSecondTime),
notBefore: DateTime.Now, notBefore: DateTime.Now,
signingCredentials: creds); signingCredentials: creds);
string returnToken = new JwtSecurityTokenHandler().WriteToken(token); string returnToken = new JwtSecurityTokenHandler().WriteToken(token);

View File

@@ -190,25 +190,21 @@ namespace Yi.Framework.Rbac.Domain.Managers
{ {
//此处优先从缓存中获取 //此处优先从缓存中获取
UserRoleMenuDto output = null; UserRoleMenuDto output = null;
var tokenExpiresMinuteTime = var tokenExpiresSecondTime =
LazyServiceProvider.GetRequiredService<IOptions<JwtOptions>>().Value.ExpiresMinuteTime; LazyServiceProvider.GetRequiredService<IOptions<JwtOptions>>().Value.ExpiresSecondTime;
var cacheData = await _userCache.GetOrAddAsync(new UserInfoCacheKey(userId), var cacheData = await _userCache.GetOrAddAsync(new UserInfoCacheKey(userId),
async () => async () =>
{ {
var user = await _userRepository.GetUserAllInfoAsync(userId); var user = await _userRepository.GetUserAllInfoAsync(userId);
var data = EntityMapToDto(user); var data = EntityMapToDto(user);
//系统用户数据被重置,老前端访问重新授权 //系统用户数据被重置,老前端访问重新授权
if (data is null)
{
throw new AbpAuthorizationException();
}
//data.Menus.Clear(); //data.Menus.Clear();
output = data; output = data ?? throw new AbpAuthorizationException();
return new UserInfoCacheItem(data); return new UserInfoCacheItem(data);
}, },
() => new DistributedCacheEntryOptions () => new DistributedCacheEntryOptions
{ AbsoluteExpirationRelativeToNow = TimeSpan.FromMinutes(tokenExpiresMinuteTime) }); { AbsoluteExpirationRelativeToNow = TimeSpan.FromSeconds(tokenExpiresSecondTime) });
if (cacheData is not null) if (cacheData is not null)
{ {

View File

@@ -1,99 +1,117 @@
import axios from "axios"; import axios from "axios";
import router from "@/router"; import router from "@/router";
import { ElMessage } from "element-plus"; import {ElMessage} from "element-plus";
import { config } from "@/config/axios/config"; import {config} from "@/config/axios/config";
import { Session } from "@/utils/storage"; import {Session} from "@/utils/storage";
import useAuths from "@/hooks/useAuths"; import useAuths from "@/hooks/useAuths";
const { VITE_APP_ENV_NAME } = import.meta.env; const {VITE_APP_ENV_NAME} = import.meta.env;
const { getToken, removeToken } = useAuths(); const {getToken,getRefreshToken, removeToken, setToken, setRefreshToken} = useAuths();
const { base_url, request_timeout, pre_interface } = config; const {base_url, request_timeout, pre_interface} = config;
export const PATH_URL = base_url[VITE_APP_ENV_NAME]; export const PATH_URL = base_url[VITE_APP_ENV_NAME];
// 配置新建一个 axios 实例 // 配置新建一个 axios 实例
const service = axios.create({ const service = axios.create({
baseURL: PATH_URL, // api 的 base_url baseURL: PATH_URL, // api 的 base_url
timeout: request_timeout, // 请求超时时间 timeout: request_timeout, // 请求超时时间
headers: { "Content-Type": "application/json" }, headers: {"Content-Type": "application/json"},
hideerror: false, //是否在底层显示错误信息 hideerror: false, //是否在底层显示错误信息
isFinish: false, isFinish: false,
}); });
// 添加请求拦截器 // 添加请求拦截器
service.interceptors.request.use( service.interceptors.request.use(
(config) => { (config) => {
// 在发送请求之前做些什么 token // 在发送请求之前做些什么 token
const token = getToken(); const token = getToken();
if (token) { if (token) {
config.headers["Authorization"] = `Bearer ${token}`; config.headers["Authorization"] = `Bearer ${token}`;
}
const refreshToken = getRefreshToken();
if (refreshToken) {
config.headers["refresh_token"] = `${refreshToken}`;
}
if (Session.get("tenantId")) {
config.headers["TenantId"] = Session.get("tenantId");
}
return config;
},
(error) => {
// 对请求错误做些什么
return Promise.reject(error);
} }
if (Session.get("tenantId")) {
config.headers["TenantId"] = Session.get("tenantId");
}
return config;
},
(error) => {
// 对请求错误做些什么
return Promise.reject(error);
}
); );
// 添加响应拦截器 // 添加响应拦截器
service.interceptors.response.use( service.interceptors.response.use(
(response) => { (response) => {
const { config } = response; const {config} = response;
config.isFinish = true; config.isFinish = true;
return Promise.resolve(response); //后端返回双token替换
}, if (response.headers["refresh_token"]) {
(error) => { setToken(response.headers["access_token"]);
const { config } = error; setRefreshToken(response.headers["refresh_token"]);
// 对响应错误做点什么 // //然后修改config重新请求
if (error.message.indexOf("timeout") != -1) { // config.headers["Authorization"] = `Bearer ${getToken()}`;
ElMessage({ // service.request(config);
type: "error",
message: "网络超时",
});
} else if (error.message == "Network Error") {
ElMessage({
type: "error",
message: "网络连接错误",
});
} else {
const res = error.response || {};
const status = Number(res.status) || 200;
const message = res?.data?.error?.message;
if (status === 401) {
ElMessageBox.confirm("该功能需要登陆后享有,是否立即登录?", "提示", {
confirmButtonText: "确认",
cancelButtonText: "取消",
type: "warning",
}).then(() => {
removeToken();
router.push("/login");
});
return;
}
if (status !== 200) {
if (status >= 500) {
ElMessage({
type: "error",
message: "网络开小差了,请稍后再试",
});
return Promise.reject(new Error(message));
} }
// 避开找不到后端接口的提醒
if (status !== 404) { return Promise.resolve(response);
ElMessage({ },
type: "error", (error) => {
message, const {config} = error;
}); // 对响应错误做点什么
if (error.message.indexOf("timeout") !== -1) {
ElMessage({
type: "error",
message: "网络超时",
});
} else if (error.message === "Network Error") {
ElMessage({
type: "error",
message: "网络连接错误",
});
} else {
//处理状态码和消息
const res = error.response || {};
const status = Number(res.status) || 200;
const message = res?.data?.error?.message;
if (status === 401) {
ElMessageBox.confirm("该功能需要登陆后享有,是否立即登录?", "提示", {
confirmButtonText: "确认",
cancelButtonText: "取消",
type: "warning",
}).then(() => {
removeToken();
router.push("/login");
});
return;
}
//处理非200
if (status !== 200) {
if (status >= 500) {
ElMessage({
type: "error",
message: "网络开小差了,请稍后再试",
});
return Promise.reject(new Error(message));
}
// 避开找不到后端接口的提醒
if (status !== 404) {
ElMessage({
type: "error",
message,
});
}
}
} }
} config.isFinish = true;
return Promise.reject(error.response);
} }
config.isFinish = true;
return Promise.reject(error.response);
}
); );
// 导出 axios 实例 // 导出 axios 实例

View File

@@ -14,213 +14,210 @@ export const AUTH_USER = "AUTH_USER";
export default function useAuths(opt) { export default function useAuths(opt) {
const defaultOpt = {
loginUrl: "/login", // 登录页跳转url 默认: /login
loginReUrl: "", // 登录页登陆成功后带重定向redirect=的跳转url 默认为空
homeUrl: "/index", // 主页跳转url 默认: /index
otherQuery: {}, // 成功登录后携带的除redirect外其他参数
};
const defaultOpt = { let option = {
loginUrl: "/login", // 登录页跳转url 默认: /login ...defaultOpt,
loginReUrl: "", // 登录页登陆成功后带重定向redirect=的跳转url 默认为空 ...opt,
homeUrl: "/index", // 主页跳转url 默认: /index };
otherQuery: {}, // 成功登录后携带的除redirect外其他参数
};
let option = { // 获取token
...defaultOpt, const getToken = () => {
...opt, var token = Local.get(TokenKey);
}; return token;
};
// 获取token
const getRefreshToken = () => {
return Local.get(RefreshTokenKey);
};
// 获取token const isLogin = computed(() => {
const getToken = () => { return getToken() ? true : false
var token= Local.get(TokenKey);
return token;
};
// 获取token
const getRefreshToken = () => {
return Local.get(RefreshTokenKey);
};
const isLogin=computed(()=>{
return getToken()? true : false
});
const currentUserInfo=computed(()=>{
return useUserStore();
});
// 存储token到cookies
const setToken = (token) => {
if (token == null) {
return false;
}
Local.set(TokenKey, token);
return true;
};
// 存储RefreshToken到cookies
const setRefreshToken = (token) => {
if (token == null) {
return false;
}
Local.set(RefreshTokenKey, token);
return true;
};
// 退出登录
const logoutFun = async () => {
let flag = true;
try {
await userLogout().then((res) => {
useUserStore().updateToken(null);
ElMessage({
message: "退出成功",
type: "info",
duration: 2000,
});
});
} catch (error) {
flag = await ElMessageBox.confirm(
`退出登录失败,是否强制退出?`,
"提示",
{
confirmButtonText: "确 定",
cancelButtonText: "取 消",
type: "warning",
}
)
.then(() => {
useUserStore().updateToken(null);
return true;
})
.catch(() => {
//取消
return false;
});
}
if (flag) {
clearStorage();
}
};
// 清空本地存储的信息
const clearStorage = () => {
Session.clear();
Local.clear();
removeToken();
};
// 用户名密码登录
const loginFun = async (params) => {
try {
const res = await userLogin(params);
ElMessage({
message: `您好${params.userName},登录成功!`,
type: "success",
});
await loginSuccess(res);
return res;
} catch (error) {
const { data } = error;
if (error.status === 403 && data.error?.message === "验证码错误") {
useUserStore().updateCodeImage();
}
}
};
// 获取用户基本信息、角色、菜单权限
const getUserInfo = async () => {
try {
let { data } = await getUserDetailInfo();
// useUserStore
// store.dispatch("updateUserInfo", result);
return data;
} catch (error) {
return {};
}
};
// 删除token
const removeToken = () => {
// console.log("token发生改变22清除清除")
Local.remove(TokenKey);
return true;
};
// 登录成功之后的操作
const loginSuccess = async (res) => {
const { token,refreshToken } = res.data;
setToken(token);
setRefreshToken(refreshToken);
useUserStore().updateToken(token);
try {
// 存储用户信息
await useUserStore().getInfo(); // 用户信息
// 登录成功后 路由跳转
// 如果有记录当前跳转页面
const currentPath = Session.get("currentPath");
if (currentPath) {
router.replace(currentPath);
} else {
router.replace({
path: option.loginReUrl ? option.loginReUrl : option.homeUrl,
query: option.otherQuery,
});
}
} catch (error) {
removeToken();
return false;
}
};
// 注册
const registerFun = async (params) => {
// try {
await userRegister(params);
ElMessage({
message: `恭喜!${params.userName},注册成功!请登录!`,
type: "success",
});
// } catch (error) {
// console.log(error);
// }
};
// 找回密码
const retrievePasswordFun = async (params) => {
// try {
const {data}=await userRetrievePassword(params);
ElMessage({
message: `恭喜!账号:${data},找回成功!密码已重置,请登录!`,
type: "success",
duration: 8000
}); });
// } catch (error) {
// console.log(error);
// }
};
return { const currentUserInfo = computed(() => {
getToken, return useUserStore();
getRefreshToken, });
setToken,
setRefreshToken, // 存储token到cookies
removeToken, const setToken = (token) => {
loginFun, if (token == null) {
getUserInfo, return false;
logoutFun, }
retrievePasswordFun, Local.set(TokenKey, token);
clearStorage,
registerFun, return true;
loginSuccess, };
isLogin, // 存储RefreshToken到cookies
currentUserInfo const setRefreshToken = (token) => {
}; if (token == null) {
return false;
}
Local.set(RefreshTokenKey, token);
return true;
};
// 退出登录
const logoutFun = async () => {
let flag = true;
try {
await userLogout().then((res) => {
useUserStore().updateToken(null);
ElMessage({
message: "退出成功",
type: "info",
duration: 2000,
});
});
} catch (error) {
flag = await ElMessageBox.confirm(
`退出登录失败,是否强制退出?`,
"提示",
{
confirmButtonText: "确 定",
cancelButtonText: "取 消",
type: "warning",
}
)
.then(() => {
useUserStore().updateToken(null);
return true;
})
.catch(() => {
//取消
return false;
});
}
if (flag) {
clearStorage();
}
};
// 清空本地存储的信息
const clearStorage = () => {
Session.clear();
Local.clear();
removeToken();
};
// 用户名密码登录
const loginFun = async (params) => {
try {
const res = await userLogin(params);
ElMessage({
message: `您好${params.userName},登录成功!`,
type: "success",
});
await loginSuccess(res);
return res;
} catch (error) {
const {data} = error;
if (error.status === 403 && data.error?.message === "验证码错误") {
useUserStore().updateCodeImage();
}
}
};
// 获取用户基本信息、角色、菜单权限
const getUserInfo = async () => {
try {
let {data} = await getUserDetailInfo();
// useUserStore
// store.dispatch("updateUserInfo", result);
return data;
} catch (error) {
return {};
}
};
// 删除token
const removeToken = () => {
// console.log("token发生改变22清除清除")
Local.remove(TokenKey);
return true;
};
// 登录成功之后的操作
const loginSuccess = async (res) => {
const {token, refreshToken} = res.data;
setToken(token);
setRefreshToken(refreshToken);
useUserStore().updateToken(token);
try {
// 存储用户信息
await useUserStore().getInfo(); // 用户信息
// 登录成功后 路由跳转
// 如果有记录当前跳转页面
const currentPath = Session.get("currentPath");
if (currentPath) {
router.replace(currentPath);
} else {
router.replace({
path: option.loginReUrl ? option.loginReUrl : option.homeUrl,
query: option.otherQuery,
});
}
} catch (error) {
removeToken();
return false;
}
};
// 注册
const registerFun = async (params) => {
// try {
await userRegister(params);
ElMessage({
message: `恭喜!${params.userName},注册成功!请登录!`,
type: "success",
});
// } catch (error) {
// console.log(error);
// }
};
// 找回密码
const retrievePasswordFun = async (params) => {
// try {
const {data} = await userRetrievePassword(params);
ElMessage({
message: `恭喜!账号:${data},找回成功!密码已重置,请登录!`,
type: "success",
duration: 8000
});
// } catch (error) {
// console.log(error);
// }
};
return {
getToken,
getRefreshToken,
setToken,
setRefreshToken,
removeToken,
loginFun,
getUserInfo,
logoutFun,
retrievePasswordFun,
clearStorage,
registerFun,
loginSuccess,
isLogin,
currentUserInfo
};
} }

View File

@@ -1,10 +1,10 @@
import axios from 'axios'; import axios from 'axios';
import { getToken } from '@/utils/auth' import {getToken} from '@/utils/auth'
export let isRelogin = { show: false };
export let isRelogin = {show: false};
// import JsonBig from 'json-bigint' // import JsonBig from 'json-bigint'
const myaxios = axios.create({ const myaxios = axios.create({
baseURL:import.meta.env.VITE_APP_BASEAPI, baseURL: import.meta.env.VITE_APP_BASEAPI,
timeout: 50000, timeout: 50000,
// transformResponse: [data => { // transformResponse: [data => {
// try { // try {
@@ -22,12 +22,11 @@ const myaxios = axios.create({
}) })
// 请求拦截器 // 请求拦截器
myaxios.interceptors.request.use(function (config) { myaxios.interceptors.request.use(function (config) {
if (getToken()) { if (getToken()) {
config.headers['Authorization'] = 'Bearer ' + getToken() // 让每个请求携带自定义token 请根据实际情况自行修改 config.headers['Authorization'] = 'Bearer ' + getToken() // 让每个请求携带自定义token 请根据实际情况自行修改
} }
return config; return config;
}, function (error) { }, function (error) {
@@ -37,27 +36,25 @@ myaxios.interceptors.request.use(function (config) {
// 响应拦截器 // 响应拦截器
myaxios.interceptors.response.use(function (response) { myaxios.interceptors.response.use(function (response) {
//业务错误 //业务错误
if(response.data.statusCode==403) if (response.data.statusCode == 403) {
{
ElMessage.error(response.data.errors) ElMessage.error(response.data.errors)
} }
return response; return response;
}, function (error) { }, function (error) {
const code = error.response.status; const code = error.response.status;
const msg = error.message; const msg = error.message;
//业务异常+应用异常,统一处理 //业务异常+应用异常,统一处理
switch(code) switch (code) {
{ case 401:
case 401: ElMessage.error('登录已过期')
ElMessage.error('登录已过期') break;
break; case 403:
case 403: ElMessage.error(msg)
ElMessage.error(msg) break;
break; case 500:
case 500: ElMessage.error(msg)
ElMessage.error(msg) break;
break; }
}
return Promise.reject(error); return Promise.reject(error);
}); });