From 3358bfb341bc3213e65aaae63d3753edf79404cc Mon Sep 17 00:00:00 2001
From: thinker0 <thinker0@linecorp.com>
Date: Sat, 20 Jan 2018 15:36:19 +0900
Subject: [PATCH] Add Aurora Master Proxy

---
 tellapart/aurproxy/source/__init__.py       |  1 +
 tellapart/aurproxy/source/sources/aurora.py | 11 +++++++++++
 2 files changed, 12 insertions(+)

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)