From 14cdb2df140f413d29f9254a0ec2e9cc7957cfca Mon Sep 17 00:00:00 2001
From: Harry Chen <czy88840616@gmail.com>
Date: Mon, 9 Dec 2019 15:11:56 +0800
Subject: [PATCH] fix: use co wrap generator for egg-bin run method

---
 packages/midway-bin/lib/cmd/debug.js | 3 ++-
 packages/midway-bin/lib/cmd/dev.js   | 3 ++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/packages/midway-bin/lib/cmd/debug.js b/packages/midway-bin/lib/cmd/debug.js
index 353fa477b51..f3187a9c90e 100644
--- a/packages/midway-bin/lib/cmd/debug.js
+++ b/packages/midway-bin/lib/cmd/debug.js
@@ -1,5 +1,6 @@
 'use strict';
 const resolver = require('../util').resolveModule;
+const co = require('co');
 
 class DebugCommand extends require('egg-bin').DebugCommand {
   constructor(rawArgv) {
@@ -11,7 +12,7 @@ class DebugCommand extends require('egg-bin').DebugCommand {
     if (!context.argv.framework) {
       context.argv.framework = this.findFramework('midway') || this.findFramework('midway-mirror');
     }
-    await super.run(context);
+    await co(super.run(context));
   }
 
   findFramework(module) {
diff --git a/packages/midway-bin/lib/cmd/dev.js b/packages/midway-bin/lib/cmd/dev.js
index 74a9d71107a..45ef79fd509 100644
--- a/packages/midway-bin/lib/cmd/dev.js
+++ b/packages/midway-bin/lib/cmd/dev.js
@@ -1,5 +1,6 @@
 'use strict';
 const resolver = require('../util').resolveModule;
+const co = require('co');
 
 class DevCommand extends require('egg-bin/lib/cmd/dev') {
   constructor(rawArgv) {
@@ -12,7 +13,7 @@ class DevCommand extends require('egg-bin/lib/cmd/dev') {
     if (!context.argv.framework) {
       context.argv.framework = this.findFramework('midway') || this.findFramework('midway-mirror');
     }
-    await super.run(context);
+    await co(super.run(context));
   }
 
   findFramework(module) {