v2.x

README.md

@@ -3,8 +3,8 @@ A wrapper for [log4js-node](https://github.com/nomiddlename/log4js-node) which s
 Log message is forked from Express (Connect) logger [file](https://github.com/nomiddlename/log4js-node/blob/master/lib/connect-logger.js).
 
 ## Note
-This branch is use to Koa v0.x & v1.x.
-To use [Koa v2.x](https://github.com/koajs/koa/tree/v2.x), please check the [v2.x](https://github.com/dominhhai/koa-log4js/tree/v2.x) branch.
+This branch is use to [Koa v2.x](https://github.com/koajs/koa/tree/v2.x).
+To use Koa v0.x & v1.x, please check the [master](https://github.com/dominhhai/koa-log4js/tree/master) branch.
 
 ## Installation
 
@@ -18,7 +18,7 @@ $ npm i --save koa-log4@1
 $ npm i --save koa-log4@2
 ```
 
-___The default logger is for [koa v2.x](https://github.com/koajs/koa/tree/v2.x)
+___The default logger is for [koa v2.x](https://github.com/koajs/koa/tree/v2.x)___
 ```
 $ npm i --save koa-log4
 ```

koa-logger.js

@@ -1,7 +1,8 @@
 'use strict'
 
+const co = require('co')
 const levels = require('log4js').levels
-var DEFAULT_FORMAT = ':remote-addr - -' +
+const DEFAULT_FORMAT = ':remote-addr - -' +
   ' ":method :url HTTP/:http-version"' +
   ' :status :content-length ":referrer"' +
   ' ":user-agent"'
@@ -47,13 +48,12 @@ function getKoaLogger (logger4js, options) {
   var fmt = options.format || DEFAULT_FORMAT
   var nolog = options.nolog ? createNoLogCondition(options.nolog) : null
 
-  return function *(next) {
-    var ctx = this
+  return co.wrap(function *(ctx, next) {
     // mount safety
-    if (ctx.request._logging) return yield next
+    if (ctx.request._logging) return yield next()
 
 		// nologs
-    if (nolog && nolog.test(ctx.originalUrl)) return yield next
+    if (nolog && nolog.test(ctx.originalUrl)) return yield next()
     if (thislogger.isLevelEnabled(level) || options.level === 'auto') {
       var start = new Date()
       var writeHead = ctx.response.writeHead
@@ -78,7 +78,7 @@ function getKoaLogger (logger4js, options) {
         }
       }
 
-      yield next
+      yield next()
 			// hook on end request to emit the log entry of the HTTP request.
       ctx.response.responseTime = new Date() - start
 			// status code response level handling
@@ -101,8 +101,8 @@ function getKoaLogger (logger4js, options) {
     }
 
     // ensure next gets always called
-    yield next
-  }
+    yield next()
+  })
 }
 
 /**

package.json

@@ -1,6 +1,6 @@
 {
   "name": "koa-log4",
-  "version": "1.0.0",
+  "version": "2.0.0",
   "description": "log4js-node support Koa-middleware",
   "main": "index.js",
   "scripts": {
@@ -22,6 +22,7 @@
   },
   "homepage": "https://github.com/dominhhai/koa-log4js#readme",
   "dependencies": {
+    "co": "^4.6.0",
     "log4js": "^0.6.35"
   },
   "devDependencies": {