1
0
Fork 0
mirror of https://github.com/ansible-collections/community.general.git synced 2024-09-14 20:13:21 +02:00

Timeouts implemented for riak stats operation. Removed failback mode

for fetching riak stats.
This commit is contained in:
James Martin 2013-05-27 17:12:13 -04:00
parent 5ec35fa388
commit 633724dc8c

View file

@ -91,13 +91,6 @@ import json
import time import time
def is_number(s):
try:
float(s)
return True
except ValueError:
return False
def ring_check(): def ring_check():
rc, out, err = module.run_command('riak-admin ringready 2> /dev/null') rc, out, err = module.run_command('riak-admin ringready 2> /dev/null')
if rc == 0 and out.find('TRUE All nodes agree on the ring') != -1: if rc == 0 and out.find('TRUE All nodes agree on the ring') != -1:
@ -105,45 +98,6 @@ def ring_check():
else: else:
return False return False
def status_to_json():
# remove all unnecessary symbols and whitespace
stats = ''
rc, out, err = module.run_command("riak-admin status 2> /dev/null")
if rc == 0:
raw_stats = out
else:
module.fail_json(msg="Could not properly gather stats")
for line in raw_stats.splitlines():
stats += line.strip() + '\n'
stats = stats.replace('<<', '').replace('>>', '')
stats = stats.replace('\\n', '').replace(',\n', ',')
stats = stats.replace(": '", ': ').replace("'\n", "\n")
stats = stats.replace(': "[', ': [').replace(']"\n', "]\n")
stats = stats.replace('"', "'")
matchObj = re.compile(r"^(.*) : (.*)", re.M | re.I)
json_stats = '{'
for match in matchObj.finditer(stats):
key, value = match.groups()
if (value[0] == "'"):
value = value[1:-1]
if not is_number(value):
value = '"' + value + '"'
json_stats += '"' + key + '":' + value + ','
json_stats = json_stats[0:-1]
json_stats += '}'
return json_stats
def main(): def main():
ansible_facts = {} ansible_facts = {}
@ -153,8 +107,8 @@ def main():
config_dir=dict(default='/etc/riak'), config_dir=dict(default='/etc/riak'),
http_conn=dict(required=False, default='127.0.0.1:8098'), http_conn=dict(required=False, default='127.0.0.1:8098'),
target_node=dict(default='riak@127.0.0.1', required=False), target_node=dict(default='riak@127.0.0.1', required=False),
wait_for_handoffs=dict(default=False, type='bool'), wait_for_handoffs=dict(default=None, type='bool'),
wait_for_ring=dict(default=False, type='bool'), wait_for_ring=dict(default=None, type='bool'),
wait_for_service=dict( wait_for_service=dict(
required=False, default=None, choices=['kv']) required=False, default=None, choices=['kv'])
) )
@ -187,24 +141,33 @@ def main():
else: else:
module.fail_json(msg='Could not determine Riak version') module.fail_json(msg='Could not determine Riak version')
try: # here we attempt to get stats from the http stats interface for 120 seconds.
stats_raw = urllib2.urlopen( timeout = time.time() + 120
'http://%s/stats' % (http_conn), None, 5).read() while True:
except urllib2.HTTPError, e: if time.time () > timeout:
stats_raw = status_to_json() module.fail_json(msg='Timeout, could not fetch Riak stats.')
except urllib2.URLError, e: try:
stats_raw = status_to_json() stats_raw = urllib2.urlopen(
except Exception, e: 'http://%s/stats' % (http_conn), None, 5).read()
stats_raw = status_to_json() break
except urllib2.HTTPError, e:
time.sleep(5)
except urllib2.URLError, e:
time.sleep(5)
except Exception, e:
module.fail_json(msg='Could not fetch Riak stats: %s' % e)
stats = json.loads(stats_raw) # here we attempt to load those stats,
try:
stats = json.loads(stats_raw)
except:
module.fail_json(msg='Timeout, could not parse Riak stats.')
node_name = stats['nodename'] node_name = stats['nodename']
nodes = stats['ring_members'] nodes = stats['ring_members']
ring_size = stats['ring_creation_size'] ring_size = stats['ring_creation_size']
result = {'node_name': node_name, result = {'node_name': node_name,
'nodes': nodes, 'nodes': nodes,
'ring_size': ring_size, 'ring_size': ring_size,