41 lines
1.2 KiB
Python
41 lines
1.2 KiB
Python
import dns.resolver
|
|
|
|
|
|
# Resolve service from mesos-dns SRV record
|
|
# return dict {"servicename": [{"name": "service.f.q.d.n", "port": 9999}]}
|
|
def resolve(app, conf):
|
|
hosts = {}
|
|
services = app['services']
|
|
domain = conf['domain']
|
|
group = None
|
|
|
|
if app['env'].get('SUROK_DISCOVERY_GROUP') is not None:
|
|
group = app['env']['SUROK_DISCOVERY_GROUP']
|
|
|
|
for service in services:
|
|
hosts[service['name']] = []
|
|
|
|
# Check group configuration
|
|
if group is not None:
|
|
pass
|
|
else:
|
|
# Load group from service config
|
|
# /etc/surok/conf.d/service_conf.json
|
|
group = service['group']
|
|
|
|
try:
|
|
for rdata in dns.resolver.query('_' +
|
|
service['name'] + '.' +
|
|
group + '._tcp.' +
|
|
domain, 'SRV'):
|
|
info = str(rdata).split()
|
|
server = {'name': info[3], 'port': info[2]}
|
|
|
|
hosts[service['name']].append(server)
|
|
except Exception as e:
|
|
print(str(e) + ": Could not resolve " +
|
|
service['name'] + '.' +
|
|
group + '._tcp.' + domain)
|
|
|
|
return hosts
|