Show More
@@ -212,6 +212,10 b' class HubFactory(RegistrationFactory):' | |||||
212 |
|
212 | |||
213 | """) |
|
213 | """) | |
214 |
|
214 | |||
|
215 | registration_timeout = Integer(0, config=True, | |||
|
216 | help="Engine registration timeout in seconds [default: max(10," | |||
|
217 | "5*heartmonitor.period)]" ) | |||
|
218 | ||||
215 | # not configurable |
|
219 | # not configurable | |
216 | db = Instance('IPython.parallel.controller.dictdb.BaseDB') |
|
220 | db = Instance('IPython.parallel.controller.dictdb.BaseDB') | |
217 | heartmonitor = Instance('IPython.parallel.controller.heartmonitor.HeartMonitor') |
|
221 | heartmonitor = Instance('IPython.parallel.controller.heartmonitor.HeartMonitor') | |
@@ -337,10 +341,13 b' class HubFactory(RegistrationFactory):' | |||||
337 | url = util.disambiguate_url(self.client_url('task')) |
|
341 | url = util.disambiguate_url(self.client_url('task')) | |
338 | r.connect(url) |
|
342 | r.connect(url) | |
339 |
|
343 | |||
|
344 | # convert seconds to msec | |||
|
345 | registration_timeout = 1000*self.registration_timeout | |||
|
346 | ||||
340 | self.hub = Hub(loop=loop, session=self.session, monitor=sub, heartmonitor=self.heartmonitor, |
|
347 | self.hub = Hub(loop=loop, session=self.session, monitor=sub, heartmonitor=self.heartmonitor, | |
341 | query=q, notifier=n, resubmit=r, db=self.db, |
|
348 | query=q, notifier=n, resubmit=r, db=self.db, | |
342 | engine_info=self.engine_info, client_info=self.client_info, |
|
349 | engine_info=self.engine_info, client_info=self.client_info, | |
343 | log=self.log) |
|
350 | log=self.log, registration_timeout=registration_timeout) | |
344 |
|
351 | |||
345 |
|
352 | |||
346 | class Hub(SessionFactory): |
|
353 | class Hub(SessionFactory): | |
@@ -410,7 +417,8 b' class Hub(SessionFactory):' | |||||
410 | """ |
|
417 | """ | |
411 |
|
418 | |||
412 | super(Hub, self).__init__(**kwargs) |
|
419 | super(Hub, self).__init__(**kwargs) | |
413 |
self.registration_timeout = |
|
420 | if self.registration_timeout <= 0: | |
|
421 | self.registration_timeout = max(10000, 5*self.heartmonitor.period) | |||
414 |
|
422 | |||
415 | # register our callbacks |
|
423 | # register our callbacks | |
416 | self.query.on_recv(self.dispatch_query) |
|
424 | self.query.on_recv(self.dispatch_query) |
General Comments 0
You need to be logged in to leave comments.
Login now