Created
March 5, 2021 11:16
-
-
Save ignasi35/4610d39735a69f3d803559e4638139ba to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import subprocess | |
| import argparse | |
| # Copy/paste/adapted from https://raw.githubusercontent.com/lightbend/lb-demos/master/bin/network_tool.py | |
| def get_deployment_pods(app_name, namespace): | |
| lines = subprocess.check_output(['kubectl', 'get', 'pods', '-n', namespace, '--selector=app=' + app_name]).split("\n") | |
| lines.pop(0) # pop header row | |
| pods = [] | |
| for line in lines: | |
| tokens = line.split() | |
| if len(tokens) > 0: | |
| pods.append(tokens[0]) | |
| return pods | |
| def get_pod_internal_ips(pod_name, namespace): | |
| lines = subprocess.check_output(["kubectl", '-n', namespace, "exec", pod_name, "--", "cat", "/etc/hosts"]).split("\n") | |
| tokens = lines[-2].split() | |
| return tokens[0] | |
| def generate_commands(pod_ips, partition_1, partition_2, action, namespace): | |
| route_action = 'add' if action == "split" else "delete" | |
| commands = [] | |
| for p1_node in partition_1: | |
| for p2_node in partition_2: | |
| commands.append(['kubectl', '-n', namespace, 'exec', p1_node, '--', 'route', route_action, '-host', pod_ips[p2_node], 'reject']) | |
| commands.append(['kubectl', '-n', namespace, 'exec', p2_node, '--', 'route', route_action, '-host', pod_ips[p1_node], 'reject']) | |
| return commands | |
| # parse arguments | |
| parser = argparse.ArgumentParser(description='utility to simulate network partitions in minikube') | |
| parser.add_argument('action', metavar='action', type=str, help='action to take on cluster. either split|restore') | |
| parser.add_argument('-n', dest='namespace', type=str, required=True, help='namespace') | |
| parser.add_argument('-a', dest='app_name', type=str, required=True, help='app selector for pods') | |
| parser.add_argument('-s', dest='split', type=int, required=True, help='number of pods you wish to split off') | |
| args = parser.parse_args() | |
| # retrieve internal pod ips | |
| pods = get_deployment_pods(args.app_name, args.namespace) | |
| pod_ips = {} | |
| for pod in pods: | |
| pod_ips[pod] = get_pod_internal_ips(pod, args.namespace) | |
| # execute routing rules per node | |
| partition_commands = generate_commands(pod_ips, pods[:args.split], pods[args.split:], args.action, args.namespace) | |
| for command in partition_commands: | |
| print "running command: " + str(command) | |
| subprocess.check_output(command) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment