以 menu 表为例:
1.app/model/menu.js
'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, icon: { type: STRING, allowNull: true }, }, { timestamps: false, }); return Menu; };
2.app/service/menu.js
'use strict'; const Service = require('egg').Service; class MenuService extends Service { // 获取所有子节点集合 // 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, app } = this; const query = { limit: app.toInt(ctx.query.limit), offset: app.toInt(ctx.query.offset) }; try { const data = await ctx.model.Menu.findAll({ query, raw: true }); return ctx.service.tools.buildTree(null, data, isMenu); } catch(err) { console.log(err); return null; } } // 根据id查询角色 async getMenuById(id) { const { ctx, app } = this; return await ctx.model.Menu.findByPk(app.toInt(id)); } // 编辑菜单 async editMenu(id, name, pid) { const { ctx, app } = this; const menu = await this.getMenuById(app.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;
3.app/controller/menu.js
'use strict'; const Controller = require('egg').Controller; class MenuController extends Controller { async lists() { const { ctx } = this; const { isMenu } = ctx.request.body; const data = await ctx.service.menu.getMenu(isMenu); if(data){ ctx.body = { code: 200, message: '查询菜单数据成功', data } }else{ ctx.body = { code: 500, message: '查询菜单数据失败', data: {} } } } 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;
4.app/router.js
// 获取菜单 router.get('/menu', controller.menu.lists); // 编辑菜单 router.post('/menu/:id', jwt, controller.menu.update);
.