egg实现登录鉴权（五）：mysql表中存储树形结构数据需求在mysql表上实现树形结构数据的存储实现树形数据的查询

需求

在mysql表上实现树形结构数据的存储
实现树形数据的查询，返回json格式的角色树数组

实现

mysql表中存储树形结构通常有四种方法，本文采用第一种即每条记录加上级id:pid字段
数据库表role结构，用户名：root，密码：123456，数据库：test

代码基本上和之前的一样，主要实现一维的对象数组转化为树形数组
config目录下config.default.js和plugin.js以及user那一支都没变
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;
    const data = await ctx.model.Role.findAll({ raw: true });
    ctx.body = ctx.service.role.buildTree(data);
  }
}
module.exports = RoleController;

app/service/role.js

'use strict';
const Service = require('egg').Service;
class RoleService extends Service {
  // 构建树形结构数据
  buildTree(data) {
    const res = [];
    // 找出所有根结点
    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 (item.pid === id) {
          item.children = getNode(item.id);
          node.push(item);
        }
      }
      if (node.length === 0) return;
      return node;
    }
    return res;
  }
}
module.exports = RoleService;

app/router.js

'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);
};

测试

获取角色树

egg实现登录鉴权（五）：mysql表中存储树形结构数据

需求

实现

测试

参考