Madiator2011/lbry-sdk
1
0
Fork 0
forked from LBRYCommunity/lbry-sdk
Code Pull requests Activity

support callbacks to component setups

Browse source
This commit is contained in:
Jack Robison 2018-04-02 16:49:48 -04:00
parent eb11da9b19
commit b808d08eb3
No known key found for this signature in database
GPG key ID: DF25C68FE0239BB2
1 changed files with 19 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
lbrynet/daemon/ComponentManager.py
View file

@ -50,15 +50,27 @@ class ComponentManager(object):
@classmethod
@defer.inlineCallbacks
def setup(cls):
def setup(cls, **callbacks):
"""
Start Components in sequence sorted by requirements
:return: (defer.Deferred)
"""
for component_name, cb in callbacks.iteritems():
if not callable(cb):
raise ValueError("%s is not callable" % cb)
cls.get_component(component_name)
def _setup(component):
if component.component_name in callbacks:
d = component._setup()
d.addCallback(callbacks[component.component_name])
return d
return component.setup()
stages = cls.sort_components()
for stage in stages:
yield defer.DeferredList([component._setup() for component in stage])
yield defer.DeferredList([_setup(component) for component in stage])
@classmethod
@defer.inlineCallbacks
@ -75,13 +87,15 @@ class ComponentManager(object):
@classmethod
def all_components_running(cls, *component_names):
"""
Check if components are running
:return: (bool) True if all specified components are running
"""
c = {component.component_name: component for component in cls.components}
components = {component.component_name: component for component in cls.components}
for component in component_names:
if component not in c:
if component not in components:
raise NameError("%s is not a known Component" % component)
if not c[component].running:
if not components[component].running:
return False
return True