feat:添加头像上传业务

Author: pedro

.eslintrc.js

@@ -1,17 +1,17 @@
 module.exports = {
-  "extends": "standard",
-  "plugins": ["jest"],
-  "rules": {
-    "semi": ["warn", "always"],
-    "quotes": ["warn", "single"],
-    "eol-last": 0,
-    "jest/no-disabled-tests": "warn",
-    "jest/no-focused-tests": "error",
-    "jest/no-identical-title": "error",
-    "jest/prefer-to-have-length": "warn",
-    "jest/valid-expect": "error"
+  extends: 'standard',
+  plugins: ['jest'],
+  rules: {
+    semi: ['warn', 'always'],
+    quotes: ['warn', 'single'],
+    'eol-last': 0,
+    'jest/no-disabled-tests': 'warn',
+    'jest/no-focused-tests': 'error',
+    'jest/no-identical-title': 'error',
+    'jest/prefer-to-have-length': 'warn',
+    'jest/valid-expect': 'error'
   },
-  "env": {
-    "jest": true
+  env: {
+    jest: true
   }
-};
+};

app/api/cms/file.js

@@ -1,11 +1,11 @@
 'use strict';
 
-const { LinRouter, NotFound } = require('lin-mizar');
-const { ratelimit } = require('lin-mizar/lin/limiter');
+const { LinRouter, ParametersException } = require('lin-mizar');
+// const { ratelimit } = require('lin-mizar/lin/limiter');
 const { LocalUploader } = require('../../extensions/file/local-uploader');
-const redis = require('redis');
+// const redis = require('redis');
 
-const client = redis.createClient();
+// const client = redis.createClient();
 
 const file = new LinRouter({
   prefix: '/cms/file'
@@ -14,27 +14,27 @@ const file = new LinRouter({
 file.post('/', async ctx => {
   const files = await ctx.multipart();
   if (files.length < 1) {
-    throw new NotFound({ msg: '未找到符合条件的文件资源' });
+    throw new ParametersException({ msg: '未找到符合条件的文件资源' });
   }
   const uploader = new LocalUploader('app/assets');
-  await uploader.upload(files);
-  ctx.success();
+  const arr = await uploader.upload(files);
+  ctx.json(arr);
 });
 
-file.get(
-  '/',
-  ratelimit({
-    db: client,
-    duration: 30 * 1000,
-    max: 5,
-    // throw: true,
-    logging: true
-  }),
-  async ctx => {
-    ctx.body = {
-      msg: 'just a normal response.'
-    };
-  }
-);
+// file.get(
+//   '/',
+//   ratelimit({
+//     db: client,
+//     duration: 30 * 1000,
+//     max: 5,
+//     // throw: true,
+//     logging: true
+//   }),
+//   async ctx => {
+//     ctx.body = {
+//       msg: 'just a normal response.'
+//     };
+//   }
+// );
 
 module.exports = { file };

app/api/cms/user.js

@@ -15,7 +15,8 @@ const {
   RegisterValidator,
   LoginValidator,
   UpdateInfoValidator,
-  ChangePasswordValidator
+  ChangePasswordValidator,
+  AvatarUpdateValidator
 } = require('../../validators/user');
 
 const { UserDao } = require('../../dao/user');
@@ -147,4 +148,13 @@ user.linGet(
   }
 );
 
+user.put('/avatar', loginRequired, async ctx => {
+  const v = await new AvatarUpdateValidator().validate(ctx);
+  const avatar = v.get('body.avatar');
+  let user = ctx.currentUser;
+  user.avatar = avatar;
+  await user.save();
+  ctx.success({ msg: '更新头像成功' });
+});
+
 module.exports = { user };

app/app.js

@@ -4,6 +4,8 @@ const Koa = require('koa');
 const KoaBodyParser = require('koa-bodyparser');
 const cors = require('@koa/cors');
 const { config } = require('lin-mizar/lin/config');
+const mount = require('koa-mount');
+const serve = require('koa-static');
 
 function applyCors (app) {
   // 跨域
@@ -15,6 +17,11 @@ function applyBodyParse (app) {
   app.use(KoaBodyParser());
 }
 
+function applyStatic (app, prefix = '/assets') {
+  const assetsDir = config.getItem('file.storeDir', 'app/static');
+  app.use(mount(prefix, serve(assetsDir)));
+}
+
 function indexPage (app) {
   app.context.manager.loader.mainRouter.get('/', async ctx => {
     ctx.type = 'html';
@@ -30,6 +37,7 @@ async function createApp () {
   const app = new Koa();
   applyBodyParse(app);
   applyCors(app);
+  applyStatic(app);
   config.initApp(app);
   const { log, error, Lin, multipart } = require('lin-mizar');
   app.use(log);

app/config/setting.js

@@ -1,6 +1,8 @@
 'use strict';
 
 module.exports = {
+  port: 5000,
+  siteDomain: 'http://localhost:5000',
   countDefault: 10,
   pageDefault: 0,
   apiDir: 'app/api',

app/extensions/file/config.js

@@ -2,7 +2,7 @@
 
 module.exports = {
   file: {
-    storeDir: 'assets',
+    storeDir: 'app/assets',
     singleLimit: 1024 * 1024 * 2,
     totalLimit: 1024 * 1024 * 20,
     nums: 10,

app/extensions/file/local-uploader.js

@@ -1,17 +1,62 @@
 const { Uploader } = require('lin-mizar/lin/file');
+const { File } = require('lin-mizar');
+const { config } = require('lin-mizar/lin/config');
 const fs = require('fs');
+const path = require('path');
+const { cloneDeep } = require('lodash');
 
 class LocalUploader extends Uploader {
   /**
    * 处理文件流
    * @param {object[]} files 文件流数组
    */
   async upload (files) {
+    const arr = [];
     for (const stream of files) {
-      const filepath = this.getStorePath(stream.filename);
-      const target = fs.createWriteStream(filepath);
-      await stream.pipe(target);
+      // 由于stream的特性，当读取其中的数据时，它的buffer会被消费
+      // 所以此处深拷贝一份计算md5值
+      const tmpStream = cloneDeep(stream);
+      const md5 = this.generateMd5(tmpStream);
+      const siteDomain = config.getItem('siteDomain', 'http://localhost');
+      // 检查md5存在
+      const exist = await File.findOne({
+        where: {
+          md5: md5
+        }
+      });
+      if (exist) {
+        arr.push({
+          key: stream.fieldname,
+          id: exist.id,
+          url: `${siteDomain}/assets/${exist.path}`
+        });
+      } else {
+        const { absolutePath, relativePath, realName } = this.getStorePath(
+          stream.filename
+        );
+        const target = fs.createWriteStream(absolutePath);
+        await stream.pipe(target);
+        const ext = path.extname(realName);
+        // stream.filename tream.filedname stream.mimeType stream.readableLength
+        const saved = await File.createRecord(
+          {
+            path: relativePath,
+            // type: 1,
+            name: realName,
+            extension: ext,
+            size: stream.readableLength,
+            md5: md5
+          },
+          true
+        );
+        arr.push({
+          key: stream.fieldname,
+          id: saved.id,
+          url: `${siteDomain}/assets/${saved.path}`
+        });
+      }
     }
+    return arr;
   }
 }
 

app/starter.js

@@ -10,14 +10,17 @@ function applyConfig () {
   for (const file of files) {
     config.getConfigFromFile(`app/config/${file}`);
   }
+  // 加载其它配置文件
+  config.getConfigFromFile('app/extensions/file/config.js');
 }
 
 const run = async () => {
   applyConfig();
   const { createApp } = require('./app');
   const app = await createApp();
-  app.listen(5000, () => {
-    app.context.logger.start('listening at http://localhost:5000');
+  const port = config.getItem('port');
+  app.listen(port, () => {
+    app.context.logger.start(`listening at http://localhost:${port}`);
   });
 };
 // 启动应用

app/validators/user.js

@@ -81,9 +81,17 @@ class ChangePasswordValidator extends LinValidator {
   }
 }
 
+class AvatarUpdateValidator extends LinValidator {
+  constructor () {
+    super();
+    this.avatar = new Rule('isNotEmpty', '必须传入头像的url链接');
+  }
+}
+
 module.exports = {
   ChangePasswordValidator,
   UpdateInfoValidator,
   LoginValidator,
-  RegisterValidator
+  RegisterValidator,
+  AvatarUpdateValidator
 };

jest.config.js

@@ -2,9 +2,7 @@
 // https://jestjs.io/docs/en/configuration.html
 
 module.exports = {
-  coverageDirectory: "coverage",
-  testEnvironment: "node",
-  testPathIgnorePatterns: [
-    "/node_modules/"
-  ]
-};
+  coverageDirectory: 'coverage',
+  testEnvironment: 'node',
+  testPathIgnorePatterns: ['/node_modules/']
+};