egg实现登录鉴权（七）：权限管理

‘use strict‘;
module.exports = app => {
const { INTEGER, STRING } = app.Sequelize;
const Menu = app.model.define(‘menu‘, {
id: { type: INTEGER, primaryKey: true, autoIncrement: true },
name: STRING(50),
pid: INTEGER,
}, {
timestamps: false,
});
return Menu;
};

‘use strict‘;
const Controller = require(‘egg‘).Controller;
class menuController extends Controller {
async index() {
const { ctx } = this;
const { isMenu } = ctx.request.body;
ctx.body = await ctx.service.menu.getMenu(isMenu);
}
async update() {
const { ctx } = this;
const { name, pid } = ctx.request.body;
await ctx.service.menu.editMenu(ctx.params.id, name, pid);
if (ctx.status === 404) {
ctx.body = { code: 400, msg: ‘编辑失败‘ };
} else {
ctx.body = { code: 200, msg: ‘编辑成功‘ };
}
}
}
module.exports = menuController;

‘use strict‘;
const Service = require(‘egg‘).Service;
function toInt(str) {
if (typeof str === ‘number‘) return str;
if (!str) return str;
return parseInt(str, 10) || 0;
}
class MenuService extends Service {
// 构建菜单权限树
// 如果id为空，则构建所有的数据
// id不为空，则构建以id为根结点的树
buildTree(id, data, isMenu) {
const res = [];
if (id) {
for (const item of data) {
if (toInt(item.id) === toInt(id)) {
item.children = getNode(id);
res.push(item);
}
}
} else {
for (const item of data) {
if (!item.pid) {
item.children = getNode(item.id);
res.push(item);
}
}
}
// 传入根结点id 递归查找所有子节点
function getNode(id) {
const node = [];
for (const item of data) {
if (toInt(item.pid) === toInt(id) && (isMenu === ‘true‘ ? item.children : true)) {
item.children = getNode(item.id);
node.push(item);
}
}
if (node.length === 0) return;
return node;
}
return res;
}
// 获取所有子节点集合
getChildrenIds(treeData) {
const res = [];
function getIds(treeData, res) {
for (const item of treeData) {
res.push(item.id);
if (item.children) { getIds(item.children, res); }
}
}
getIds(treeData, res);
return res;
}
// 查询角色并构建菜单树
async getMenu(isMenu) {
const { ctx } = this;
const query = { limit: toInt(ctx.query.limit), offset: toInt(ctx.query.offset) };
const data = await ctx.model.Menu.findAll({ query, raw: true });
return this.buildTree(null, data, isMenu);
}
// 根据id查询角色
async getMenuById(id) {
const { ctx } = this;
return await ctx.model.Menu.findByPk(toInt(id));
}
// 编辑菜单
async editMenu(id, name, pid) {
const { ctx } = this;
const menu = await this.getMenuById(toInt(id));
if (!menu) {
ctx.status = 404;
return;
}
await menu.update({ name: name || menu.name, pid: pid || menu.pid });
ctx.status = 200;
}
}
module.exports = MenuService;

‘use strict‘;
/**
* @param {Egg.Application} app - egg application
*/
module.exports = app => {
const { router, controller, jwt } = app;
router.get(‘/‘, controller.home.index);
router.post(‘/user/login‘, controller.user.login);
// 查询用户
router.get(‘/user‘, controller.user.index);
router.get(‘/user/:id‘, jwt, controller.user.show);
// 新增
router.put(‘/user‘, jwt, controller.user.create);
// 修改密码
router.post(‘/user/:id‘, jwt, controller.user.updatePwd);
// 获取角色
router.get(‘/role‘, controller.role.index);
router.get(‘/role/:id‘, controller.role.show);
// 插入角色
router.put(‘/role‘, jwt, controller.role.create);
// 修改角色
router.post(‘/role/:id‘, jwt, controller.role.update);
// 删除角色
router.delete(‘/role/:id‘, jwt, controller.role.remove);
// 获取菜单
router.get(‘/menu‘, controller.menu.index);
// 编辑菜单
router.post(‘/menu/:id‘, jwt, controller.menu.update);
};