Engine: add server

trekjs · Sep 8, 2015 · 66344da · 66344da
1 parent bc92daa
commit 66344da
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/lib/Engine.js b/lib/Engine.js
@@ -266,9 +266,9 @@ var Engine = (function (_Koa) {
     return _co2['default'].call(this, function* () {
       // TODO: https
       if (!args[0]) args[0] = this.config.get('site.port');
-      var app = this.listen.apply(this, args);
-      this.logger.info(_chalk2['default'].green('%s application starting in %s on http://%s:%s'), Trek.version, Trek.env, app.address().address === '::' ? '127.0.0.1' : app.address().address, app.address().port);
-      this._httpServer = app;
+      this.server = this.listen.apply(this, args);
+      var address = this.server.address();
+      this.logger.info(_chalk2['default'].green('%s application starting in %s on http://%s:%s'), Trek.version, Trek.env, address.address === '::' ? '127.0.0.1' : address.address, address.port);
     })['catch'](function (e) {
       _this3.logger.error(_chalk2['default'].bold.red(`${ e.stack }`));
       _this3.logger.error(_chalk2['default'].red('boots failed.'));

diff --git a/src/Engine.js b/src/Engine.js
@@ -253,15 +253,15 @@ class Engine extends Koa {
     return co.call(this, function* () {
       // TODO: https
       if (!args[0]) args[0] = this.config.get('site.port');
-      let app = this.listen(...args);
+      this.server = this.listen(...args);
+      let address = this.server.address();
       this.logger.info(
         chalk.green('%s application starting in %s on http://%s:%s'),
         Trek.version,
         Trek.env,
-        app.address().address === '::' ? '127.0.0.1' : app.address().address,
-        app.address().port
+        address.address === '::' ? '127.0.0.1' : address.address,
+        address.port
       );
-      this._httpServer = app;
     }).catch((e) => {
       this.logger.error(chalk.bold.red(`${e.stack}`));
       this.logger.error(chalk.red('boots failed.'));