mirror of
https://github.com/krkn-chaos/krkn.git
synced 2026-04-15 06:57:28 +00:00
8dd18af161
Signed-off-by: arcprabh <arcprabh@in.ibm.com> Incorporated first round of review comments Signed-off-by: arcprabh <arcprabh@in.ibm.com> Resolve multiple node name issue for single ip Signed-off-by: arcprabh <arcprabh@in.ibm.com>
42 lines
1.5 KiB
Python
42 lines
1.5 KiB
Python
import time
|
|
import random
|
|
import logging
|
|
import kraken.kubernetes.client as kubecli
|
|
import kraken.invoke.command as runcommand
|
|
|
|
|
|
# Pick a random node with specified label selector
|
|
def get_node(node_name, label_selector):
|
|
if node_name in kubecli.list_killable_nodes():
|
|
return node_name
|
|
else:
|
|
logging.info("Node with provided node_name does not exist or the node might "
|
|
"be in NotReady state.")
|
|
nodes = kubecli.list_killable_nodes(label_selector)
|
|
if not nodes:
|
|
raise Exception("Ready nodes with the provided label selector do not exist")
|
|
logging.info("Ready nodes with the label selector %s: %s" % (label_selector, nodes))
|
|
number_of_nodes = len(nodes)
|
|
node = nodes[random.randint(0, number_of_nodes - 1)]
|
|
return node
|
|
|
|
|
|
# Wait till node status becomes Ready
|
|
def wait_for_ready_status(node, timeout):
|
|
runcommand.invoke("kubectl wait --for=condition=Ready "
|
|
"node/" + node + " --timeout=" + str(timeout) + "s")
|
|
|
|
|
|
# Wait till node status becomes NotReady
|
|
def wait_for_unknown_status(node, timeout):
|
|
for _ in range(timeout):
|
|
if kubecli.get_node_status(node) == "Unknown":
|
|
break
|
|
time.sleep(1)
|
|
if kubecli.get_node_status(node) != "Unknown":
|
|
raise Exception("Node condition status isn't Unknown")
|
|
|
|
# Get the ip of the cluster node
|
|
def get_node_ip(node):
|
|
return runcommand.invoke("kubectl get node %s -o jsonpath='{.status.addresses[?(@.type==\"InternalIP\")].address}'" % (node))
|