Skip to content

Commit

Permalink
feat(route): allow to inject middlewares to routeMap
Browse files Browse the repository at this point in the history
  • Loading branch information
a632079 committed Jun 14, 2020
1 parent 40c1f75 commit b206713
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 5 deletions.
4 changes: 2 additions & 2 deletions routes.js
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
'use strict'
module.exports = (router, controller) => {
module.exports = (router, middlewares, controller) => {
// Route Map
/* router.get('/', async (ctx, next) => {
ctx.body = {
Expand All @@ -15,7 +15,7 @@ module.exports = (router, controller) => {
const nconf = require('nconf')
const os = require('os')
let memoryUsage = 0
for (let v of Object.values(process.memoryUsage())) {
for (const v of Object.values(process.memoryUsage())) {
memoryUsage += parseInt(v)
}
memoryUsage = memoryUsage / (1024 * 1024)
Expand Down
11 changes: 8 additions & 3 deletions src/route.js
Original file line number Diff line number Diff line change
@@ -1,11 +1,12 @@
'use strict'
const fs = require('fs')
const path = require('path')
const winston = require('winston')

class route {
constructor () {
const Controller = require(path.join(__dirname, '../', './src/controller'))
const Controller = require('./controller')
this.controller = new Controller()
this.middlewares = require('./middleware').fetch(false) // TODO: Support Dev Routes
return this.routes()
}

Expand All @@ -14,7 +15,11 @@ class route {
const controller = await this.controller
// RouteMap
const Router = require('koa-router')
return require(path.join(__dirname, '../', './routes'))(new Router(), controller)
if (!fs.existsSync(path.join(__dirname, '../', './routes.js'))) {
winston.error('[route] can\'t find the route file, program exiting.')
process.exitCode(1)
}
return require(path.join(__dirname, '../', './routes'))(new Router(), this.middlewares, controller)
} catch (err) {
winston.error(err)
process.exit(1)
Expand Down

0 comments on commit b206713

Please sign in to comment.