egg實現登錄鑒權（五）：mysql表中存儲樹形結構數據

需求

• 在mysql表上實現樹形結構數據的存儲
• 實現樹形數據的查詢，返回json格式的樹形數據

實現

• mysql表中存儲樹形結構通常有四種方法，本文采用第一種即每條記錄加表示上級id的pid字段
• 數據庫表role結構，用戶名：root，密碼：123456，數據庫：test

• 代碼基本上和之前的一樣，主要實現一維的對象數組轉化為樹形數組
• config目錄下config.default.js和plugin.js都沒變
• 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);
};

 

• package.json

{
  "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" }

 

測試

• 獲取角色樹

參考

• https://www.zhihu.com/question/20417447
• https://www.jianshu.com/p/bdaffff30b31