From b33d31dd72a691330aab49a2d4b6bed694e398e5 Mon Sep 17 00:00:00 2001
From: Jon Eyrick <social@nextlocal.net>
Date: Wed, 27 Dec 2017 11:05:23 -0800
Subject: [PATCH] Added websocket reconnect to `trades` function

---
 node-binance-api.js | 17 ++++++++++++-----
 1 file changed, 12 insertions(+), 5 deletions(-)

diff --git a/node-binance-api.js b/node-binance-api.js
index dc7c8f76..1462facb 100644
--- a/node-binance-api.js
+++ b/node-binance-api.js
@@ -262,11 +262,15 @@ module.exports = function() {
 	};
 	const depthData = function(data) { // Used for /depth endpoint
 		let bids = {}, asks = {}, obj;
-		for ( obj of data.bids ) {
-			bids[obj[0]] = parseFloat(obj[1]);
+		if ( typeof data.bids !== "undefined" ) {
+			for ( obj of data.bids ) {
+				bids[obj[0]] = parseFloat(obj[1]);
+			}
 		}
-		for ( obj of data.asks ) {
-			asks[obj[0]] = parseFloat(obj[1]);
+		if ( typeof data.asks !== "undefined" ) {
+			for ( obj of data.asks ) {
+				asks[obj[0]] = parseFloat(obj[1]);
+			}
 		}
 		return {bids:bids, asks:asks};
 	}
@@ -606,7 +610,10 @@ module.exports = function() {
 			},
 			trades: function(symbols, callback) {
 				for ( let symbol of symbols ) {
-					subscribe(symbol.toLowerCase()+"@aggTrade", callback);
+					let reconnect = function() {
+						if ( options.reconnect ) subscribe(symbol.toLowerCase()+"@aggTrade", callback, reconnect);
+					};
+					subscribe(symbol.toLowerCase()+"@aggTrade", callback, reconnect);
 				}
 			},
 			chart: function chart(symbols, interval, callback) {