diff --git a/tellapart/aurproxy/source/__init__.py b/tellapart/aurproxy/source/__init__.py index 9aa39de..338822c 100644 --- a/tellapart/aurproxy/source/__init__.py +++ b/tellapart/aurproxy/source/__init__.py @@ -16,6 +16,7 @@ from .source import ProxySource from .sources.api import ApiSource from .sources.aurora import AuroraProxySource +from .sources.aurora import AuroraMasterProxySource from .sources.mesos import MesosMasterProxySource from .sources.static import StaticProxySource from .sources.serverset import ServerSetSource diff --git a/tellapart/aurproxy/source/sources/aurora.py b/tellapart/aurproxy/source/sources/aurora.py index edbb950..d7de4cd 100644 --- a/tellapart/aurproxy/source/sources/aurora.py +++ b/tellapart/aurproxy/source/sources/aurora.py @@ -17,6 +17,7 @@ from .serverset import ServerSetSource _DEFAULT_ANNOUNCER_SERVERSET_PATH = '/aurora/' +_DEFAULT_AURORA_SCHEDULER_PATH = '/aurora/scheduler' def get_service_discovery_path(job, announcer_serverset_path): @@ -42,3 +43,13 @@ def __init__(self, announcer_serverset_path ) super(AuroraProxySource, self).__init__(serverset_path, zk_servers, **kw) + + +class AuroraMasterProxySource(ServerSetSource): + + def __init__(self, + zk_servers, + aurora_master_path=_DEFAULT_AURORA_SCHEDULER_PATH, + **kw): + + super(AuroraMasterProxySource, self).__init__(aurora_master_path, zk_servers, **kw)