diff --git a/lib/updateProcessors/MessageUpdateProcessor.js b/lib/updateProcessors/MessageUpdateProcessor.js index 72a5203..3b33c86 100755 --- a/lib/updateProcessors/MessageUpdateProcessor.js +++ b/lib/updateProcessors/MessageUpdateProcessor.js @@ -54,7 +54,19 @@ class MessageUpdateProcessor extends BaseUpdateProcessor { controller.controller.localization = this._dataSource.localization try { - controller.controller[controller.handler](controller.controller.before(scope)) + let beforeMiddlewareValue = controller.controller.before(scope) + + if (beforeMiddlewareValue === false) { + if (typeof controller.controller.rejected !== 'undefined') { + controller.controller.rejected(null, scope) + } + } else if (beforeMiddlewareValue instanceof Promise) { + beforeMiddlewareValue + .then(scope => controller.controller[controller.handler](scope)) + .catch(err => controller.controller.rejected(err, scope)) + } else { + controller.controller[controller.handler](beforeMiddlewareValue) + } } catch (e) { this._dataSource.logger.error({ @@ -71,7 +83,7 @@ class MessageUpdateProcessor extends BaseUpdateProcessor { }) } - scope = null + //scope = null return }