实现权限级菜单、localStorageSWR缓存

This commit is contained in:
2025-06-19 09:03:52 +08:00
parent 00e6ffe12a
commit 3ac2a164a5
5 changed files with 115 additions and 80 deletions

View File

@@ -25,29 +25,52 @@ export default function ConsoleMenuLayout({
const router = useRouter();
const getInitialData = () => {
if (!window || !window.localStorage) return null;
const cache = localStorage.getItem(USER_ME_CACHE_KEY);
if (!cache) return;
try {
const user = JSON.parse(cache);
if (!user || !user.userId) throw new Error();
return user;
} catch (error) {
localStorage.removeItem(USER_ME_CACHE_KEY);
}
return undefined;
}
const USER_ME_CACHE_KEY = 'user-me-cache';
const { data: user, isLoading, error } = useSWR(
'/api/user/me',
() => UserApi.me(),
async () => {
const data = await UserApi.me();
localStorage.setItem(USER_ME_CACHE_KEY, JSON.stringify(data));
return data;
},
{
onError: (error) => {
if (error.statusCode === 401) {
localStorage.removeItem('token');
localStorage.removeItem(USER_ME_CACHE_KEY);
toast.info('登录凭证已失效,请重新登录');
router.replace('/console/login');
}
}
},
fallbackData: getInitialData(),
revalidateIfStale: false,
revalidateOnFocus: false,
revalidateOnReconnect: false,
}
);
if (!isLoading && !error && !user) {
router.replace('/console/login');
localStorage.removeItem('token');
localStorage.removeItem(USER_ME_CACHE_KEY);
toast.error('账户状态异常,请重新登录');
}
return (
<SidebarProvider>
<AppSidebar user={user} isUserLoading={isLoading} />