diff --git a/.gitignore b/.gitignore index ffa90d4..6e332f6 100644 --- a/.gitignore +++ b/.gitignore @@ -1,2 +1,4 @@ surok/__pycache__ __pycache__ +*.pyc +*.swp diff --git a/conf.d/1.json b/conf.d/1.json index 0967ef4..152ae26 100644 --- a/conf.d/1.json +++ b/conf.d/1.json @@ -1 +1,12 @@ -{} +{ + "services": [ + { + "name": "kioskservice", + "group": "production.romania" + } + ], + "conf_name": "test", + "template": "templates/nginx_proxy.jj2", + "dest": "emailsender", + "reload_cmd": "/bin/echo reload nginx" +} diff --git a/conf/surok.json b/conf/surok.json index 96046a7..58e5694 100644 --- a/conf/surok.json +++ b/conf/surok.json @@ -1,4 +1,7 @@ { "marathon": "10.0.1.199:8080", - "confd": "conf.d" + "confd": "conf.d", + "domain": "marathon.mesos", + "wait_time": 20, + "lock_dir": "/var/tmp" } diff --git a/surok.py b/surok.py index 1417e3a..1cf77e8 100755 --- a/surok.py +++ b/surok.py @@ -9,7 +9,7 @@ from surok.discovery import resolve from surok.system import reload_conf # Load base configurations -f = open('/etc/surok/conf/surok.json', 'r') +f = open('conf/surok.json', 'r') conf = json.loads(f.read()) print(conf) f.close() @@ -36,7 +36,7 @@ while 1: for app in confs: app_conf = load_app_conf(app) app_hosts = resolve(app_conf, conf) - my = {'app': app_conf['name'], 'hosts': app_hosts} + my = { 'hosts': app_hosts } service_conf = gen(my, app_conf['template']) print(reload_conf(service_conf, app_conf)) diff --git a/surok/discovery.py b/surok/discovery.py index b43a298..6d8297a 100644 --- a/surok/discovery.py +++ b/surok/discovery.py @@ -2,10 +2,12 @@ import dns.resolver def resolve(app, conf): hosts = [] + services = app['services'] domain = conf['domain'] - for rdata in dns.resolver.query('_' + app['name'] + '.' + app['group'] + '._tcp.' + domain, 'SRV'): - info = str(rdata).split() - server = { 'name': info[3], 'port': info[2] } - hosts.append(server) + for service in services: + for rdata in dns.resolver.query('_' + service['name'] + '.' + service['group'] + '._tcp.' + domain, 'SRV'): + info = str(rdata).split() + server = { service['name']: { 'name': info[3], 'port': info[2] } } + hosts.append(server) return hosts diff --git a/surok/system.py b/surok/system.py index 686fca8..bd1bb7e 100644 --- a/surok/system.py +++ b/surok/system.py @@ -28,14 +28,14 @@ def write_lock(name, service_conf): def reload_conf(service_conf, app_conf): # Check old config - if get_old(app_conf['name'], service_conf) != 1: + if get_old(app_conf['conf_name'], service_conf) != 1: print('Write new configuration') f = open(app_conf['dest'], 'w') f.write(service_conf) f.close() - write_lock(app_conf['name'], service_conf) + write_lock(app_conf['conf_name'], service_conf) # Reload conf stdout = os.popen(app_conf['reload_cmd']).read() diff --git a/templates/nginx_proxy.jj2 b/templates/nginx_proxy.jj2 new file mode 100644 index 0000000..4abe0b3 --- /dev/null +++ b/templates/nginx_proxy.jj2 @@ -0,0 +1,11 @@ + {% for host in my['hosts'] %} + server {{host['kioskservice']['name']}}:{{host['kioskservice']['port']}}; + {% endfor %} + +server { + listen 0.0.0.0:80; + server_name _; + + location / { + } +}