Merge branch 'master' of https://github.com/ZWkang/javascript-demo

Author: ZWkang

units/simple-server-router-test.js

@@ -0,0 +1,70 @@
+var Koa = require('koa')
+  , http = require('http')
+  , request = require('supertest')
+  , should = require('should')
+  , Route = require('./simple-server-router.js')
+
+async function writeCtxBodyAuthorName (ctx, next) {
+    ctx.body = {
+        Author: 'ZWkang'
+    }
+    return await next()
+}
+async function writeCtxBodyMuti (ctx, next) {
+    ctx.body.muti = true
+    return await next()
+}
+describe('Route#middlerware', function() {
+    it('test single Route case', function(done) {
+        var app = new Koa();
+        var router = new Route();
+        
+        router.register(/\/singleMiddler/, writeCtxBodyAuthorName)
+        
+        app.use(router.createMiddlerware());
+        request(http.createServer(app.callback()))
+        .get('/singleMiddler')
+        .expect(200)
+        .end(function(err, res) {
+            if (err) return done(err);
+            res.should.have.property('body');
+            res.body.should.have.property('Author', 'ZWkang');
+            done();
+        });
+    });
+    it('test muti Routes case', function(done) {
+        var app = new Koa();
+        var router = new Route({
+            muti: true
+        });
+        
+        router.register(/\/singleMiddler/, writeCtxBodyAuthorName)
+        router.register(/\/singleMiddlersTest/, writeCtxBodyMuti)
+        app.use(router.createMiddlerware());
+        request(http.createServer(app.callback()))
+        .get('/singleMiddlersTest')
+        .expect(200)
+        .end(function(err, res) {
+            if (err) return done(err);
+            res.should.have.property('body');
+            res.body.should.have.property('Author', 'ZWkang');
+            res.body.should.have.property('muti', true);
+            done();
+        });
+    });
+    it('test default should be 404 status code', function(done) {
+        var app = new Koa();
+        var router = new Route();
+        
+        router.register(/\/singleMiddler/, writeCtxBodyAuthorName)
+        router.register(/\/singleMiddlersTest/, writeCtxBodyMuti)
+        app.use(router.createMiddlerware());
+        request(http.createServer(app.callback()))
+        .get('/iwant404')
+        .expect(404)
+        .end(function(err, res) {
+            if (err) return done(err);
+            done();
+        });
+    });
+})

units/simple-server-router.js

@@ -0,0 +1,55 @@
+const debug = require('debug')('kang-route');
+const compose = require('koa-compose');
+
+/**
+ * use
+ * 1. require('simple-server-router.js')
+ * 2. const routers = new Route({
+ *      muti: true| false // 是否重复匹配路由
+ * })
+ * 
+ * 3. routers.register(/test/, async (ctx, next) { ctx.body = {name: 'kang'}})
+ * 4. app.use(routers.createMiddlerware())
+ */
+const defaultOptions = {
+    muti: false
+}
+function Route(options) {
+    this.middlerwares = []
+    this.options = options || defaultOptions
+}
+
+Route.prototype.register = function (regex, middlerware) {
+    this.middlerwares.push({
+        regex,
+        middlerware
+    })
+}
+Route.prototype.createMiddlerware = function () {
+    const { muti } = this.options
+    return async (ctx, next) => {
+        const { req } = ctx
+        const len = this.middlerwares.length
+        const url = ctx.url
+        const middlerwares = []
+        for(let i = 0; i < len; i++) {
+            const middlerware = this.middlerwares[i]
+            const isRegexp = middlerware['regex'].exec(url)
+            if(isRegexp) {
+                middlerwares.push(middlerware['middlerware'])
+                if(i === len-1 || !muti) {
+                    return compose(middlerwares)(ctx,next)
+                }
+            }
+        }
+        noop(ctx, next)
+        return await next()
+    }
+}
+
+function noop (ctx, next) {
+    ctx.status = 404
+    ctx.res.end()
+}
+
+module.exports = Route