fix: use co wrap generator for egg-bin run method

midwayjs · Dec 11, 2019 · 14cdb2d · 14cdb2d
1 parent fa108ef
commit 14cdb2d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git 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
@@ -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) {