[ARVADOS] updated: a2c147e469bc4ccc0e788f2e72b2d94aa81ba368

[Git user]

Summary of changes:
 sdk/python/arvados/events.py        | 22 +++++++++++++++-------
 sdk/python/tests/test_websockets.py |  6 +++---
 2 files changed, 18 insertions(+), 10 deletions(-)

       via  a2c147e469bc4ccc0e788f2e72b2d94aa81ba368 (commit)
      from  7b2cd50f499a1f9c5b21ab6aa641f14701402885 (commit)

Those revisions listed above that are new to this repository have
not appeared on any other notification email; so we list those
revisions in full, below.


commit a2c147e469bc4ccc0e788f2e72b2d94aa81ba368
Author: radhika <radhika at curoverse.com>
Date:   Tue Apr 5 12:00:19 2016 -0400

    7658: update EventClient.on_closed to retry on connect errors.

diff --git a/sdk/python/arvados/events.py b/sdk/python/arvados/events.py
index fcc3feb..57bf3ae 100644
--- a/sdk/python/arvados/events.py
+++ b/sdk/python/arvados/events.py
@@ -15,7 +15,7 @@ _logger = logging.getLogger('arvados.events')
 
 
 class _EventClient(WebSocketClient):
-    def __init__(self, url, filters, on_event, last_log_id, on_closed=None):
+    def __init__(self, url, filters, on_event, last_log_id, on_closed):
         ssl_options = {'ca_certs': arvados.util.ca_certs_path()}
         if config.flag_is_true('ARVADOS_API_HOST_INSECURE'):
             ssl_options['cert_reqs'] = ssl.CERT_NONE
@@ -27,7 +27,11 @@ class _EventClient(WebSocketClient):
         # will be attempted -- and it might not be the right one. See
         # ws4py's WebSocketBaseClient.__init__.
         super(_EventClient, self).__init__(url, ssl_options=ssl_options)
-        self.filters = filters
+
+        if filters:
+            self.filters = filters
+        else:
+            self.filters = [[]]
         self.on_event = on_event
         self.last_log_id = last_log_id
         self._closing_lock = threading.RLock()
@@ -78,7 +82,6 @@ class EventClient(object):
         self.last_log_id = last_log_id
         self.is_closed = False
         self.subscriptions = {}
-        self.subscriptions[str(filters)] = filters
         self.ec = _EventClient(url, filters, self.on_event, last_log_id, self.on_closed)
 
     def connect(self):
@@ -106,11 +109,16 @@ class EventClient(object):
 
     def on_closed(self):
         if self.is_closed == False:
-            filters = []
-            for s in self.subscriptions:
-                filters.append(self.subscriptions[s])
             self.ec = _EventClient(self.url, self.filters, self.on_event, self.last_log_id, self.on_closed)
-            self.ec.connect()
+            while True:
+              try:
+                  self.ec.connect()
+                  for s in self.subscriptions:
+                      self.ec.subscribe(self.subscriptions[s], self.last_log_id)
+                  break
+              except:
+                  _logger.warn("Failed to reconnect to websockets on %s. Will retry after 5s." % endpoint)
+                  time.sleep(5)
 
 
 class PollClient(threading.Thread):
diff --git a/sdk/python/tests/test_websockets.py b/sdk/python/tests/test_websockets.py
index 2049f56..98d84ad 100644
--- a/sdk/python/tests/test_websockets.py
+++ b/sdk/python/tests/test_websockets.py
@@ -144,15 +144,15 @@ class WebsocketTest(run_test_server.TestCaseWithServers):
         with self.assertRaises(Queue.Empty):
             self.assertEqual(events.get(True, 2), None)
 
-        # create one more obj
-        human2 = arvados.api('v1').humans().create(body={}).execute()
-
         # close (im)properly
         if close_unexpected:
             self.ws.close_connection()
         else:
             self.ws.close()
 
+        # create one more obj
+        human2 = arvados.api('v1').humans().create(body={}).execute()
+
         # (un)expect the object creation event
         if close_unexpected:
             log_object_uuids = []

-----------------------------------------------------------------------


hooks/post-receive
--