六、从零模拟新浪微博-创建微博

代码仓库：https://github.com/changeclass/koa2-weibo

数据模型

/**
 * @description: 微博数据模型
 * @author: 小康
 * @url: https://xiaokang.me
 * @Date: 2020-12-18 16:05:16
 * @LastEditTime: 2020-12-18 16:05:17
 * @LastEditors: 小康
 */

const seq = require('../seq')
const { STRING, INTEGER, TEXT } = require('../type')

const Blog = seq.define('blog', {
  userId: {
    type: INTEGER,
    allowNull: false,
    comment: '用户ID'
  },
  content: {
    type: TEXT,
    allowNull: false,
    comment: '微博内容'
  },
  image: {
    type: STRING,
    comment: '图片地址'
  }
})

module.exports = Blog

在模型入口文件创建外键关系

/**
 * @description 数据模型入口文件
 * @author 小康
 */

const User = require('./User')
const Blog = require('./Blog')

// 创建外键
Blog.belongsTo(User, {
  foreignKey: 'userId'
})

module.exports = {
  User,
  Blog
}

创建微博API

1. 路由层

   /**
    * @description: 首页 api 路由
    * @author: 小康
    * @url: https://xiaokang.me
    * @Date: 2020-12-18 16:36:31
    * @LastEditTime: 2020-12-18 16:36:31
    * @LastEditors: 小康
    */
   
   const router = require('koa-router')()
   
   const { create } = require('../../controller/blog-home')
   const { loginCheck } = require('../../middlewares/loginChecks')
   router.prefix('/api/blog')
   
   // 创建微博
   router.post('/create', loginCheck, async (ctx, next) => {
     const { content, image } = ctx.request.body
     const { id: userId } = ctx.session.userInfo
     ctx.body = await create({ userId, content, image })
   })
   
   module.exports = router

2. 控制器层

   /**
    * @description: 首页 controller
    * @author: 小康
    * @url: https://xiaokang.me
    * @Date: 2020-12-18 16:46:39
    * @LastEditTime: 2020-12-18 16:46:39
    * @LastEditors: 小康
    */
   
   const { createBlogFailInfo } = require('../model/ErrorInfo')
   const { SuccessModel, ErrorModel } = require('../model/ResModel')
   const { createBlog } = require('../services/blog')
   
   /**
    * @author: 小康
    * @url: https://xiaokang.me
    * @param {string} userId
    * @param {string} content
    * @param {string} image
    * @description: 创建微博
    */
   async function create({ userId, content, image }) {
     // servers
     try {
       const blog = await createBlog({ userId, content, image })
       return new SuccessModel(blog)
     } catch (error) {
       console.error(error.message, error.stack)
       return new ErrorModel(createBlogFailInfo)
     }
   }
   
   module.exports = { create }

3. 服务层

   /**
    * @description: 微博servers
    * @author: 小康
    * @url: https://xiaokang.me
    * @Date: 2020-12-18 16:48:09
    * @LastEditTime: 2020-12-18 16:48:09
    * @LastEditors: 小康
    */
   const { Blog } = require('../db/model/index')
   
   /**
    * @author: 小康
    * @url: https://xiaokang.me
    * @description: 创建微博
    * @param {string} userId
    * @param {string} content
    * @param {string} image
    */
   async function createBlog({ userId, content, image }) {
     const result = await Blog.create({
       userId,
       content,
       image
     })
     return result.dataValues
   }
   
   module.exports = {
     createBlog
   }

防止xss攻击

安装插件

yarn add xss

对需要防止的内容进行包装即可

const xss = require('xss')
const blog = await createBlog({ userId, content: xss(content), image })