Added websocket pinger to keep connection alive

ophis · Jul 20, 2017 · d2f8831 · d2f8831
1 parent 26ba493
commit d2f8831
Showing 1 changed file with 5 additions and 7 deletions.
diff --git a/gdax/websocket_client.py b/gdax/websocket_client.py
@@ -27,11 +27,6 @@ def __init__(self, url="wss://ws-feed.gdax.com", products=None, message_type="su
         self.api_secret = api_secret
         self.api_passphrase = api_passphrase
 
-    def _auth_message(self, method, path, options=None):
-
-
-        return auth
-
     def start(self):
         def _go():
             self._connect()
@@ -73,6 +68,9 @@ def _connect(self):
     def _listen(self):
         while not self.stop:
             try:
+                if int(time.time() % 30) == 0:
+                    # Set a 30 second ping to keep connection alive
+                    self.ws.ping("keepalive")
                 msg = json.loads(self.ws.recv())
             except Exception as e:
                 self.on_error(e)
@@ -101,7 +99,7 @@ def on_message(self, msg):
         print(msg)
 
     def on_error(self, e):
-        return
+        print(e)
 
 if __name__ == "__main__":
     import gdax
@@ -126,7 +124,7 @@ def on_close(self):
     wsClient.start()
     print(wsClient.url, wsClient.products)
     # Do some logic with the data
-    while wsClient.message_count < 500:
+    while wsClient.message_count < 10000:
         print("\nMessageCount =", "%i \n" % wsClient.message_count)
         time.sleep(1)