egg实现登录鉴权（八）：sequelize联表查询

表之间的关联关系有三种：一对一，一对多，多对多。要进行多表操作，首先要建立表与表之间的关联关系，在sequelize中分别使用hasOne，hasMany和belongsToMany表示以上三中关系。例如我们要建立user表与role表的多对多的关联，可以这样写，首先要建立外键（当然也可以在两张表之间建立中间表）

本文主要完成多表操作简单的部分，熟悉在多表中建立关联关系并且可以完成简单的联表操作，以联表查询为例。

需求

• 登录成功时返回结果中要包含角色信息，以便展示对应的权限菜单
• 查询用户时返回结果中要包含角色名称

约定

• 登录（POST）

• • 传参:

• • • Body: nickname（必填）, password（必填）

• • 成功: {code:200,msg:'登录成功',token,roleid}
  • 失败: {code:400,msg:'登录失败'}

• 查询所有用户（GET）

• • 传参
  • 成功: {code:200,msg:'查询成功',data}
  • 失败: {code:400,msg:'查询失败'}

实现

• 数据库结构，user增加roleid

• 目录结构

• app/model/user.js

 

'use strict';

module.exports = app => {
const { STRING, INTEGER } = app.Sequelize;
const User = app.model.define(‘user’, {
id: { type: INTEGER, primaryKey: true, autoIncrement: true },
nickname: STRING(20),
password: STRING(50),
roleid: INTEGER,
}, {
timestamps: false,
});
User.associate = function() {
app.model.User.belongsTo(app.model.Role, { foreignKey: ‘roleid’, targetKey: ‘id’, as: ‘role’ });
};
return User;
};

 

 

• app/service/user.js

 

'use strict';

const Service = require(‘egg’).Service;
const crypto = require(‘crypto’);
// 设置默认密码
const DEFAULT_PWD = ‘123456’;
function toInt(str) {
if (typeof str === ‘number’) return str;
if (!str) return str;
return parseInt(str, 10) || 0;
}

class UserService extends Service {
// 查询user表，验证密码和花名
async validUser(nickname, password) {
const data = await this.ctx.model.User.findAll();
const pwd = this.getMd5Data(password);
for (const item of data) {
if (item.nickname === nickname && item.password === pwd) return item;
}
return false;
}
// 获取用户，不传id则查询所有
async getUser(id) {
const { ctx } = this;
const query = { limit: toInt(ctx.query.limit), offset: toInt(ctx.query.offset) };
if (id) {
return await ctx.model.User.findByPk(toInt(id));
}
return await ctx.model.User.findAll({ query,
attributes: [ ‘nickname’ ],
include: [{
model: ctx.model.Role,
as: ‘role’,
attributes: [ ‘name’ ],
}],
});
}
// 新增人员
async addUser(nickname) {
const { ctx } = this;
const password = this.getMd5Data(DEFAULT_PWD);
await ctx.model.User.create({ nickname, password });
}
// 修改密码
async editPwd(id, password) {
const { ctx } = this;
const user = await ctx.model.User.findByPk(id);
const newPwd = this.getMd5Data(password);
await user.update({ password: newPwd });
}
// 专门对数据进行md5加密的方法，输入明文返回密文
getMd5Data(data) {
return crypto.createHash(‘md5’).update(data).digest(‘hex’);
}
}
module.exports = UserService;

 

 

• app/controller/user.js

 

'use strict';

const Controller = require(‘egg’).Controller;

class UserController extends Controller {
// 登录
async login() {
const { ctx, app } = this;
const data = ctx.request.body;
// 判断该用户是否存在并且密码是否正确
const getUser = await ctx.service.user.validUser(data.nickname, data.password);
if (getUser) {
const token = app.jwt.sign({ nickname: data.nickname }, app.config.jwt.secret);
ctx.body = { code: 200, msg: ‘登录成功’, token, roleid: getUser.roleid };
} else {
ctx.body = { code: 400, msg: ‘登录失败’ };
}
}
// 获取所有用户
async index() {
const { ctx } = this;
const data = await ctx.service.user.getUser();
ctx.body = { code: 200, msg: ‘查询成功’, data };
}
// 通过id获取用户
async show() {
const { ctx } = this;
const data = await ctx.service.user.getUser(ctx.params.id);
ctx.body = { code: 200, msg: ‘查询成功’, data };
}
// 创建用户
async create() {
const { ctx } = this;
const { nickname } = ctx.request.body;
await ctx.service.user.addUser(nickname);
ctx.body = { code: 200, msg: ‘新增成功’ };
}
// 修改密码
async updatePwd() {
const { ctx } = this;
const { password } = ctx.request.body;
await ctx.service.user.editPwd(ctx.params.id, password);
ctx.body = { code: 200, msg: ‘修改成功’ };
}
}

module.exports = UserController;

 

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

 

测试

• 登录

• 查询

总结

本文主要完成多表的关联查询，先通过一个例子，用户角色之间的关联完成两张表之间的查询，熟悉多表之间的操作。

参考

• http://itbilu.com/nodejs/npm/EJarwPD8W.html
• https://sequelize.org/v5/manual/associations.html

分类: node系列

<div id="blog_post_info">

好文要顶 关注我 收藏该文

mingL
关注 - 4
粉丝 - 25

+加关注

0

0

<div class="clear"></div>
<div id="post_next_prev">

<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>
<br>
<a href="https://www.cnblogs.com/xingguozhiming/p/12712026.html" class="p_n_p_prefix">» </a> 下一篇：    <a href="https://www.cnblogs.com/xingguozhiming/p/12712026.html" title="发布于 2020-04-16 12:45">js实现将字符串首字母转换成大写</a>

转载：https://www.cnblogs.com/xingguozhiming/p/12340543.html