diff --git a/lib/trader.js b/lib/trader.js index 43db0309..d7698e93 100644 --- a/lib/trader.js +++ b/lib/trader.js @@ -155,7 +155,7 @@ Trader.prototype.stopPolling = function () { clearInterval(this.rateInterval); }; -Trader.prototype.pollBalance = function () { +Trader.prototype.pollBalance = function (callback) { var self = this; self.logger.info('collecting balance'); @@ -170,9 +170,15 @@ Trader.prototype.pollBalance = function () { self.tradeExchange.balance(next); } }, function (err, balance) { + if (err) { + return callback && callback(err); + } + balance.timestamp = Date.now(); self.logger.info('Balance update:', balance); self.balance = balance; + + callback && callback(null, balance); }); };