前面已经完成角色树的存储并且能够查询并构建出树形数据,本文继续完成角色树的其他操作,可以插入,删除和修改等常规操作。
需求
- 查询不鉴权,增删改需要传token
- 能够通过id查询角色,返回该角色下面所有角色树。
- 插入新角色,可以另开一个组织架构,即该角色没有上级
- 修改角色和删除角色,修改和删除角色时要判断该角色是否存在,
- 修改角色分两种情况
- 修改角色名称,不改变结构
- 修改上级id,修改该角色上级时,该角色的下级默认一并带入;
约定
名次解释:
id:角色id
pid:角色上级id
name:角色名称
- 参数:id(可选),不传则查询所有
- 成功返回:角色树,json格式数组
- 失败返回:{code:400,msg:'查询失败'}
- 参数:pid(可选),name(必选),pid为空则该角色为顶级
- 成功返回:{code:200,msg:'新增成功'}
- 参数:id(必选),name(可选),pid(可选)
- 成功返回:{code:200,msg:'修改成功'}
- 失败返回:{code:400,msg:'修改失败'}
- 参数:id(必选)
- 成功返回:{code:200,msg:'删除成功'}
- 失败返回:{code:400,msg:'删除失败'}
实现
注意力转移到角色表,user表和config都和之前一样,没有改动。难点在同个id查询角色树和删除角色时要同时删除子集
- 数据库,用户名:root,密码:123456,表:test
![image.png image.png](https://i-blog.csdnimg.cn/blog_migrate/03dfe480844b26cf552f115d6923d4d8.png)
- 目录结构
![image.png image.png](https://i-blog.csdnimg.cn/blog_migrate/656e59ad80c24fff4e243ce306fcea5f.png)
- app/model/role.js
module.exports = app => {
const { INTEGER, STRING } = app.Sequelize;
const Role = app.model.define(‘role’, {
id: { type: INTEGER, primaryKey: true, autoIncrement: true },
name: STRING(50),
pid: INTEGER,
}, {
timestamps: false,
});
return Role;
};
const Controller = require(‘egg’).Controller;
class RoleController extends Controller {
async index() {
const { ctx } = this;
ctx.body = await ctx.service.role.getRole();
}
async show() {
const { ctx } = this;
ctx.body = await ctx.service.role.getRole(ctx.params.id);
}
// 插入角色
async create() {
const { ctx } = this;
const { name, pid } = ctx.request.body;
await ctx.service.role.addRole(name, pid);
ctx.body = { code: 200, msg: ‘新增成功’ };
}
// 更新角色
async update() {
const { ctx } = this;
const { name, pid } = ctx.request.body;
await ctx.service.role.editRole(ctx.params.id, name, pid);
if (ctx.status === 404) {
ctx.body = { code: 400, msg: ‘修改失败’ };
} else {
ctx.body = { code: 200, msg: ‘修改成功’ };
}
}
// 移除角色
async remove() {
const { ctx } = this;
await ctx.service.role.removeRole(ctx.params.id);
if (ctx.status === 404) {
ctx.body = { code: 400, msg: ‘删除失败’ };
} else {
ctx.body = { code: 200, msg: ‘删除成功’ };
}
}
}
module.exports = RoleController;
'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 RoleService extends Service {
// 构建树形结构数据
// 如果id为空,则构建所有的数据
// id不为空,则构建以id为根结点的树
buildTree(id, data) {
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)) {
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 getRole(id) {
const { ctx } = this;
const query = { limit: toInt(ctx.query.limit), offset: toInt(ctx.query.offset) };
const data = await ctx.model.Role.findAll({ query, raw: true });
return this.buildTree(id, data);
}
// 根据id查询角色
async getRoleById(id) {
const { ctx } = this;
return await ctx.model.Role.findByPk(toInt(id));
}
// 插入角色
async addRole(name, pid) {
const { ctx } = this;
await ctx.model.Role.create({ name, pid });
}
// 修改角色
async editRole(id, name, pid) {
const { ctx } = this;
const role = await this.getRoleById(toInt(id));
if (!role) {
ctx.status = 404;
return;
}
await role.update({ name: name || role.name, pid: pid || role.pid });
ctx.status = 200;
}
// 删除角色
async removeRole(id) {
const { ctx } = this;
const roleTree = await this.getRole(toInt(id));
const role = await this.getRoleById(toInt(id));
if (!role) {
ctx.status = 404;
return;
}
const ids = this.getChildrenIds(roleTree);
for (const i of ids) {
const r = await this.getRoleById(toInt(i));
r.destroy();
}
ctx.status = 200;
}
}
module.exports = RoleService;
- app/router.js(角色路由,user相关暂时可不管)
/
- @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’, jwt, 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);
};
{
"name": "jwt",
"version": "1.0.0",
"description": "",
"private": true,
"egg": { "declarations": true }, "dependencies": { "egg": "^2.15.1", "egg-cors": "^2.2.3", "egg-jwt": "^3.1.7", "egg-scripts": "^2.11.0", "egg-sequelize": "^5.2.0", "mysql2": "^2.0.2" }, "devDependencies": { "autod": "^3.0.1", "autod-egg": "^1.1.0", "egg-bin": "^4.11.0", "egg-ci": "^1.11.0", "egg-mock": "^3.21.0", "eslint": "^5.13.0", "eslint-config-egg": "^7.1.0" }, "engines": { "node": ">=10.0.0" }, "scripts": { "start": "egg-scripts start --daemon --title=egg-server-jwt", "stop": "egg-scripts stop --title=egg-server-jwt", "dev": "egg-bin dev", "debug": "egg-bin debug", "test": "npm run lint -- --fix && npm run test-local", "test-local": "egg-bin test", "cov": "egg-bin cov", "lint": "eslint .", "ci": "npm run lint && npm run cov", "autod": "autod" }, "ci": { "version": "10" }, "repository": { "type": "git", "url": "" }, "author": "", "license": "MIT" }
测试
![image.png image.png](https://i-blog.csdnimg.cn/blog_migrate/e5bfe09c808e03145980797c7275dc44.png)
![image.png image.png](https://i-blog.csdnimg.cn/blog_migrate/f92a6202ef0033bc8d474b3bafe06868.png)
![image.png image.png](https://i-blog.csdnimg.cn/blog_migrate/01ca073fd3b280a153176e0cb409d26f.png)
![image.png image.png](https://i-blog.csdnimg.cn/blog_migrate/9b0031feab53d40cf14b322d531d88f7.png)
![image.png image.png](https://i-blog.csdnimg.cn/blog_migrate/2f2a1d475810c875bbe70d40857b4623.png)
总结
- 通过id获取子节点
- 遍历树形json获取所有子节点id集合
- 循环遍历ids,删除当前节点的所有子节点
<div id="blog_post_info">
<div class="clear"></div>
<div id="post_next_prev">
<a href="https://www.cnblogs.com/xingguozhiming/p/12146454.html" class="p_n_p_prefix">« </a> 上一篇: <a href="https://www.cnblogs.com/xingguozhiming/p/12146454.html" title="发布于 2020-01-03 19:15">egg实现登录鉴权(五):mysql表中存储树形结构数据</a>
<br>
<a href="https://www.cnblogs.com/xingguozhiming/p/12172597.html" class="p_n_p_prefix">» </a> 下一篇: <a href="https://www.cnblogs.com/xingguozhiming/p/12172597.html" title="发布于 2020-01-09 17:24">egg实现登录鉴权(七):权限管理</a>
转载:https://www.cnblogs.com/xingguozhiming/p/12157019.html