egg实现登录鉴权（六）：角色树的CRUD操作-一一网

前面已经完成角色树的存储并且能够查询并构建出树形数据，本文继续完成角色树的其他操作，可以插入，删除和修改等常规操作。

需求

查询不鉴权，增删改需要传token
能够通过id查询角色，返回该角色下面所有角色树。
插入新角色，可以另开一个组织架构，即该角色没有上级
修改角色和删除角色，修改和删除角色时要判断该角色是否存在，
修改角色分两种情况
- 修改角色名称，不改变结构
- 修改上级id，修改该角色上级时，该角色的下级默认一并带入；
删除角色时如果有下级则一并删除

约定

名次解释：
id:角色id
pid:角色上级id
name:角色名称

通过id查询角色树
- 参数：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

app/model/role.js

'use strict';

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;
};
复制代码

app/controller/role.js

'use strict';

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;
复制代码

app/service/role.js（重点）

'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相关暂时可不管）

'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', 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);
};
复制代码