[后端] 优化生产环境插件

Author: smallpath

server/build/blogpack.prod.config.js

@@ -25,14 +25,21 @@ const CheckAuthPlugin = require('../plugins/beforeRestful/checkAuth')
 
 config.plugins.push(
   // beforeUseRoutes
-  new RatelimitPlugin(),
+  new RatelimitPlugin({
+    duration: 1000,
+    errorMessage: 'Slow Down Your Request.',
+    id: ctx => ctx.ip,
+    max: 10
+  }),
   new BodyParserPlugin(),
   new LogTimePlugin(),
   new RestcPlugin(),
+
   // beforeServerStart
   new InitUserPlugin(),
   new InstallThemePlugin(),
   new InitOptionPlugin(),
+
   // beforeRestful
   new CheckAuthPlugin()
 )

server/plugins/beforeServerStart/initOption.js

@@ -3,8 +3,6 @@ const options = require('../../conf/option')
 const models = require('../../model/mongo')
 
 module.exports = class {
-  constructor() {}
-
   async beforeServerStart() {
     for (const option of options) {
       let key = option.key

server/plugins/beforeServerStart/initUser.js

@@ -3,8 +3,6 @@ const config = require('../../conf/config')
 const models = require('../../model/mongo')
 
 module.exports = class {
-  constructor() {}
-
   async beforeServerStart() {
     const count = await models.user.find().count().exec()
     if (count !== 0) return

server/plugins/beforeServerStart/installTheme.js

@@ -5,8 +5,6 @@ const resolve = file => path.resolve(__dirname, file)
 const models = require('../../model/mongo')
 
 module.exports = class {
-  constructor() {}
-
   async beforeServerStart() {
     const prefix = '../../theme'
     let fileArr = fs.readdirSync(resolve(prefix))

server/plugins/beforeUseRoutes/bodyParser.js

@@ -1,8 +1,6 @@
 const bodyParser = require('koa-bodyparser')
 
 module.exports = class {
-  constructor() {}
-
   async beforeUseRoutes({ app }) {
     app.use(bodyParser())
   }

server/plugins/beforeUseRoutes/logTime.js

@@ -1,10 +1,6 @@
 const log = require('../../utils/log')
 
 module.exports = module.exports = class {
-  constructor(options) {
-    this.options = options
-  }
-
   async beforeUseRoutes({ app, redis }) {
     app.use(async (ctx, next) => {
       const start = new Date()

server/plugins/beforeUseRoutes/ratelimit.js

@@ -1,15 +1,14 @@
 const ratelimit = require('koa-ratelimit')
 
 module.exports = class {
-  constructor() {}
+  constructor(options) {
+    this.options = options
+  }
 
   async beforeUseRoutes({ app, redis }) {
-    app.use(ratelimit({
-      db: redis,
-      duration: 1000,
-      errorMessage: 'Slow Down Your Request.',
-      id: ctx => ctx.ip,
-      max: 10
-    }))
+    const config = Object.assign({}, this.options, {
+      db: redis
+    })
+    app.use(ratelimit(config))
   }
 }

server/plugins/beforeUseRoutes/restc.js

@@ -1,8 +1,6 @@
 const restc = require('restc')
 
 module.exports = class {
-  constructor() {}
-
   async beforeUseRoutes({ app }) {
     app.use(restc.koa2())
   }