直接上代码(关键截取) 全在 src/app.tsx
修改
1:import 添加icon
import * as Icon from '@ant-design/icons';
2:添加处理图标
const loopMenuItem = (menus: MenuDataItem[]): MenuDataItem[] =>(
menus.map(({ icon, children, ...item }) => {
return {
...item,
icon: icon && React.createElement(Icon[icon]),
children: children && loopMenuItem(children),
}
})
);
3:根据动态处理路由方法包裹处理路由icon
![](https://img.haomeiwen.com/i13350377/ae3a4d45cc3676ef.png)
完整 src/app.ts代码
import React from 'react';
// @ts-ignore
import { BasicLayoutProps, Settings as LayoutSettings, MenuDataItem } from '@ant-design/pro-layout';
import { notification } from 'antd';
import { history, RequestConfig } from 'umi';
import RightContent from '@/components/RightContent';
import Footer from '@/components/Footer';
import { ResponseError } from 'umi-request';
import _ from 'lodash';
import { queryCurrent } from './services/user';
import { getMenuList } from './services/menu';
import defaultSettings from '../config/defaultSettings';
import * as Icon from '@ant-design/icons';
export async function getInitialState(): Promise<{
currentUser?: API.CurrentUser;
settings?: LayoutSettings;
menuList?: API.HttpRule;
}> {
// 如果是登录页面,不执行
if (history.location.pathname !== '/user/login') {
try {
const currentUser = await queryCurrent();
const menuList = await getMenuList(currentUser.userid || '');
return {
currentUser,
settings: defaultSettings,
menuList,
};
} catch (error) {
history.push('/user/login');
}
}
return {
settings: defaultSettings,
menuList: {},
};
}
const filterByMenuDate = (data: MenuDataItem[], keyWord: string): MenuDataItem[] =>
data
.map((item) => {
if (
(item.name && item.name.includes(keyWord)) ||
filterByMenuDate(item.children || [], keyWord).length > 0
) {
return {
...item,
children: filterByMenuDate(item.children || [], keyWord),
};
}
return undefined;
})
.filter((item) => item) as MenuDataItem[];
// 处理图标
const loopMenuItem = (menus: MenuDataItem[]): MenuDataItem[] =>(
menus.map(({ icon, children, ...item }) => {
return {
...item,
icon: icon && React.createElement(Icon[icon]),
children: children && loopMenuItem(children),
}
})
);
export const layout = ({
initialState,
}: {
initialState: {
settings?: LayoutSettings;
currentUser?: API.CurrentUser;
menuList?: API.HttpRule;
};
}): BasicLayoutProps => {
// 请求路由
const { routes } = _.isEmpty(initialState?.menuList)
? { routes: [] }
: initialState?.menuList.data;
return {
menuDataRender: () => loopMenuItem(routes),
rightContentRender: () => <RightContent />,
disableContentMargin: false,
footerRender: () => <Footer />,
onPageChange: () => {
// 如果没有登录,重定向到 login
if (!initialState?.currentUser?.userid && history.location.pathname !== '/user/login') {
history.push('/user/login');
}
},
menuHeaderRender: undefined,
...initialState?.settings,
};
};
const codeMessage = {
200: '服务器成功返回请求的数据。',
201: '新建或修改数据成功。',
202: '一个请求已经进入后台排队(异步任务)。',
204: '删除数据成功。',
400: '发出的请求有错误,服务器没有进行新建或修改数据的操作。',
401: '用户没有权限(令牌、用户名、密码错误)。',
403: '用户得到授权,但是访问是被禁止的。',
404: '发出的请求针对的是不存在的记录,服务器没有进行操作。',
405: '请求方法不被允许。',
406: '请求的格式不可得。',
410: '请求的资源被永久删除,且不会再得到的。',
422: '当创建一个对象时,发生一个验证错误。',
500: '服务器发生错误,请检查服务器。',
502: '网关错误。',
503: '服务不可用,服务器暂时过载或维护。',
504: '网关超时。',
};
/**
* 异常处理程序
*/
const errorHandler = (error: ResponseError) => {
const { response } = error;
if (response && response.status) {
const errorText = codeMessage[response.status] || response.statusText;
const { status, url } = response;
notification.error({
message: `请求错误 ${status}: ${url}`,
description: errorText,
});
}
if (!response) {
notification.error({
description: '您的网络发生异常,无法连接服务器',
message: '网络异常',
});
}
throw error;
};
export const request: RequestConfig = {
errorHandler,
};
结果
![](https://img.haomeiwen.com/i13350377/72d308c40ca26c9b.png)
网友评论