blob: 2f1efd80062c4ebf4a0115e951d5820d4daefed1 [file] [log] [blame]
khenb95fe9a2016-10-05 11:15:25 -07001#
Zsolt Haraszti3eb27a52017-01-03 21:56:48 -08002# Copyright 2017 the original author or authors.
khenb95fe9a2016-10-05 11:15:25 -07003#
4# Licensed under the Apache License, Version 2.0 (the "License");
5# you may not use this file except in compliance with the License.
6# You may obtain a copy of the License at
7#
8# http://www.apache.org/licenses/LICENSE-2.0
9#
10# Unless required by applicable law or agreed to in writing, software
11# distributed under the License is distributed on an "AS IS" BASIS,
12# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13# See the License for the specific language governing permissions and
14# limitations under the License.
15#
16
17"""
18Some consul related convenience functions
19"""
20
khenb95fe9a2016-10-05 11:15:25 -070021from structlog import get_logger
khenb95fe9a2016-10-05 11:15:25 -070022from consul import Consul
23from random import randint
24
25log = get_logger()
26
Khen Nursimulu441dedd2016-10-05 14:44:26 -070027
khenb95fe9a2016-10-05 11:15:25 -070028def get_endpoint_from_consul(consul_endpoint, service_name):
29 """Look up, from consul, the service name specified by service-name
30 """
Khen Nursimulu90fc35d2017-01-09 08:42:04 -050031 log.debug('getting-service-endpoint', consul=consul_endpoint,
32 service=service_name)
33
khenb95fe9a2016-10-05 11:15:25 -070034 host = consul_endpoint.split(':')[0].strip()
35 port = int(consul_endpoint.split(':')[1].strip())
36
37 consul = Consul(host=host, port=port)
38 _, services = consul.catalog.service(service_name)
39
40 if len(services) == 0:
Khen Nursimulu441dedd2016-10-05 14:44:26 -070041 raise Exception(
42 'Cannot find service {} in consul'.format(service_name))
khenb95fe9a2016-10-05 11:15:25 -070043
44 # pick a random entry
45 # TODO should we prefer local IP addresses? Probably.
46
47 service = services[randint(0, len(services) - 1)]
48 endpoint = '{}:{}'.format(service['ServiceAddress'],
49 service['ServicePort'])
50
khenb95fe9a2016-10-05 11:15:25 -070051 return endpoint
52
53
54if __name__ == '__main__':
Khen Nursimulu9b9f1ad2017-01-10 15:43:32 -050055 print get_endpoint_from_consul('10.100.198.220:8500', 'kafka')