def cluster_wait_for_service(args, cfg):
# Uses 1 second timeout to just test once to see if Framework up
if util.wait_for_framework(cfg, args['debug_flag'], 1):
service_url = cfg.api_url()
r = requests.get(service_url + 'clusters/' + args['cluster'] +
'/nodes')
util.debug_request(args['debug_flag'], r)
js = json.loads(r.text)
# Timeout must be at least 1 second
num_nodes = len(js['nodes'])
if num_nodes > 0:
node_timeout = max(args['timeout'] / num_nodes, 1)
for k in js['nodes']:
util.wait_for_node(cfg, args['cluster'], args['debug_flag'],
k, node_timeout)
if num_nodes >= args['num_nodes']:
# Okay, need to divide up the timeout properly
util.wait_for_node_status_valid(cfg,
args['cluster'],
args['debug_flag'],
js['nodes'][0],
args['num_nodes'],
args['timeout'])
return
print('Riak Mesos Framework did not respond.')
return
def wait_for_service(ctx, nodes, **kwargs):
"""Iterates over all nodes in cluster and executes node wait-for-service.
Optionally waits until the number of nodes (specified by --nodes) at
minimum are joined to the cluster"""
ctx.init_args(**kwargs)
r = ctx.api_request('get', 'clusters/' + ctx.cluster + '/nodes')
js = json.loads(r.text)
ctx.vlog(nodes)
num_nodes = len(js['nodes'])
total_timeout = ctx.timeout
if num_nodes > 0:
ctx.timeout = max(total_timeout / num_nodes, 1)
for k in js['nodes']:
wait_for_node(ctx, k)
if num_nodes >= nodes:
# Okay, need to divide up the timeout properly
ctx.timeout = total_timeout
wait_for_node_status_valid(ctx, js['nodes'][0], nodes)
else:
click.echo("No nodes have been added to cluster " + ctx.cluster)
def wait_for_service(ctx, nodes, **kwargs):
"""Iterates over all nodes in cluster and executes node wait-for-service.
Optionally waits until the number of nodes (specified by --nodes) at
minimum are joined to the cluster"""
ctx.init_args(**kwargs)
r = ctx.api_request('get', 'clusters/' + ctx.cluster + '/nodes')
js = json.loads(r.text)
ctx.vlog(nodes)
num_nodes = len(js['nodes'])
total_timeout = ctx.timeout
if num_nodes > 0:
ctx.timeout = max(total_timeout / num_nodes, 1)
for k in js['nodes']:
wait_for_node(ctx, k)
if num_nodes >= nodes:
# Okay, need to divide up the timeout properly
ctx.timeout = total_timeout
wait_for_node_status_valid(ctx, js['nodes'][0], nodes)
else:
click.echo("No nodes have been added to cluster " + ctx.cluster)
def cluster_wait_for_service(args, cfg):
# Uses 1 second timeout to just test once to see if Framework up
if util.wait_for_framework(cfg, args['debug_flag'], 1):
service_url = cfg.api_url()
r = requests.get(service_url + 'clusters/' + args['cluster'] +
'/nodes')
util.debug_request(args['debug_flag'], r)
js = json.loads(r.text)
# Timeout must be at least 1 second
num_nodes = len(js['nodes'])
if num_nodes > 0:
node_timeout = max(args['timeout'] / num_nodes, 1)
for k in js['nodes']:
util.wait_for_node(cfg, args['cluster'], args['debug_flag'], k,
node_timeout)
if num_nodes >= args['num_nodes']:
# Okay, need to divide up the timeout properly
util.wait_for_node_status_valid(cfg, args['cluster'],
args['debug_flag'], js['nodes'][0],
args['num_nodes'], args['timeout'])
return
print('Riak Mesos Framework did not respond.')
return
def wait_for_service(ctx, **kwargs):
"""Waits timeout seconds (default is 60) or until node is running.
Specify timeout with --timeout"""
ctx.init_args(**kwargs)
wait_for_node(ctx, ctx.node)
def node_wait_for_service(args, cfg):
if args['node'] == '':
raise CliError('Node name must be specified')
util.wait_for_node(cfg, args['cluster'], args['debug_flag'],
args['node'], args['timeout'])
return
def node_wait_for_service(args, cfg):
if args['node'] == '':
raise CliError('Node name must be specified')
util.wait_for_node(cfg, args['cluster'], args['debug_flag'], args['node'],
args['timeout'])
return
def wait_for_service(ctx, **kwargs):
"""Waits timeout seconds (default is 60) or until node is running.
Specify timeout with --timeout"""
ctx.init_args(**kwargs)
wait_for_node(ctx, ctx.node)
