Update Koa example for Koa 2 with async/await (vercel#1317)

aranajhonny · timneutkens · commit 68d48cb004ac · 2017-02-28T21:49:58.000+01:00
diff --git a/examples/custom-server-koa/package.json b/examples/custom-server-koa/package.json
@@ -6,8 +6,8 @@
     "start": "NODE_ENV=production node server.js"
   },
   "dependencies": {
-    "koa": "^1.2.4",
-    "koa-router": "^5.4.0",
+    "koa": "^2.0.1",
+    "koa-router": "^7.1.0",
     "next": "^2.0.0-beta",
     "react": "^15.4.2",
     "react-dom": "^15.4.2"
diff --git a/examples/custom-server-koa/server.js b/examples/custom-server-koa/server.js
@@ -11,26 +11,24 @@ app.prepare()
   const server = new Koa()
   const router = new Router()
 
-  router.get('/a', function *() {
-    app.render(this.req, this.res, '/b', this.query)
-    this.respond = false
+  router.get('/a', async ctx => {
+    await app.render(ctx.req, ctx.res, '/b', ctx.query)
+    ctx.respond = false
   })
 
-  router.get('/b', function *() {
-    app.render(this.req, this.res, '/a', this.query)
-    this.respond = false
+  router.get('/b', async ctx => {
+    await app.render(ctx.req, ctx.res, '/a', ctx.query)
+    ctx.respond = false
   })
 
-  router.get('*', function *() {
-    handle(this.req, this.res)
-    this.respond = false
+  router.get('*', async ctx => {
+    await handle(ctx.req, ctx.res)
+    ctx.respond = false
   })
 
-  server.use(function *(next) {
-    // Koa doesn't seems to set the default statusCode.
-    // So, this middleware does that
-    this.res.statusCode = 200
-    yield next
+  server.use(async (ctx, next) => {
+    ctx.res.statusCode = 200
+    await next()
   })
 
   server.use(router.routes())
