Skip to content
Snippets Groups Projects
Commit 78fd443b authored by Erik Reid's avatar Erik Reid
Browse files

set result backend transport options properly

parent 7db67331
No related branches found
No related tags found
No related merge requests found
...@@ -9,6 +9,8 @@ assert os.path.isfile(os.environ['INVENTORY_PROVIDER_CONFIG_FILENAME']), ( ...@@ -9,6 +9,8 @@ assert os.path.isfile(os.environ['INVENTORY_PROVIDER_CONFIG_FILENAME']), (
'config file %r not found' % 'config file %r not found' %
os.environ['INVENTORY_PROVIDER_CONFIG_FILENAME']) os.environ['INVENTORY_PROVIDER_CONFIG_FILENAME'])
broker_transport_options = result_backend_transport_options = dict()
with open(os.environ['INVENTORY_PROVIDER_CONFIG_FILENAME']) as f: with open(os.environ['INVENTORY_PROVIDER_CONFIG_FILENAME']) as f:
logging.info( logging.info(
'loading config from: %r' 'loading config from: %r'
...@@ -17,26 +19,30 @@ with open(os.environ['INVENTORY_PROVIDER_CONFIG_FILENAME']) as f: ...@@ -17,26 +19,30 @@ with open(os.environ['INVENTORY_PROVIDER_CONFIG_FILENAME']) as f:
if 'sentinel' in inventory_provider_config: if 'sentinel' in inventory_provider_config:
params = inventory_provider_config['sentinel'] params = inventory_provider_config['sentinel']
sentinel = redis.sentinel.Sentinel( _broker_scheme = 'sentinel'
[(params['hostname'], int(params['port']))], broker_transport_options['master_name'] = params['name']
socket_timeout=0.5) if 'sentinel_socket_timeout' in params:
master = sentinel.discover_master(params['name']) broker_transport_options['sentinel_socket_timeout'] \
assert master = params['sentinel_socket_timeout']
_broker_hostname = master[0]
_broker_port = master[1]
else: else:
params = inventory_provider_config['redis'] params = inventory_provider_config['redis']
_broker_hostname = params['hostname'] _broker_scheme = 'redis'
_broker_port = int(params['port'])
if 'redis_socket_timeout' in params:
broker_transport_options['redis_socket_timeout'] \
= params['redis_socket_timeout']
_broker_hostname = params['hostname']
_broker_port = params['port']
_broker_db_index = 1 # TODO: this should be a config param _broker_db_index = 1 # TODO: this should be a config param
if ':' in _broker_hostname: if ':' in _broker_hostname:
# assume this means hostname is an ipv6 address # assume this means hostname is an ipv6 address
_broker_hostname = '[%s]' % _broker_hostname _broker_hostname = f'[_broker_hostname]'
broker_url = result_backend = 'redis://%s:%d/%d' % ( broker_url = result_backend = (f'{_broker_scheme}://{_broker_hostname}'
_broker_hostname, _broker_port, _broker_db_index) f':{_broker_port}/{_broker_db_index}')
logger.debug('broker_url: %r' % broker_url) logger.debug('broker_url: %r' % broker_url)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment