help='The number of benchmark nodes in the cluster ' + '(optional)', dest='benchmark', default=0) parser.add_argument( '--conf', nargs='?', type=str, help='The configuration file to start the cluster with' + ' (optional)', dest='conf', default='../conf/kvs-base.yml') parser.add_argument('--ssh-key', nargs='?', type=str, help='The SSH key used to configure and connect to ' + 'each node (optional)', dest='sshkey', default='~/.ssh/id_rsa') cluster_name = util.check_or_get_env_arg('FLUENT_CLUSTER_NAME') kops_bucket = util.check_or_get_env_arg('KOPS_STATE_STORE') aws_key_id = util.check_or_get_env_arg('AWS_ACCESS_KEY_ID') aws_key = util.check_or_get_env_arg('AWS_SECRET_ACCESS_KEY') args = parser.parse_args() create_cluster(args.memory, args.ebs, args.function, args.scheduler, args.routing, args.benchmark, args.conf, args.sshkey, cluster_name, kops_bucket, aws_key_id, aws_key)
#!/usr/bin/env python3 import os import sys from add_nodes import add_nodes from remove_nodes import delete_nodes import util from routing_util import register, deregister import subprocess import time # AWS Info aws_key_id = util.check_or_get_env_arg('AWS_ACCESS_KEY_ID') aws_key = util.check_or_get_env_arg('AWS_SECRET_ACCESS_KEY') # Config File Info BASE_CONFIG_FILE = '../config/tasc-base.yml' CONFIG_FILE = './tasc-config.yml' POD_CONFIG_DIR = '/go/src/github.com/saurav-c/tasc/config' NODE_TYPES = ['tasc', 'keynode', 'routing', 'lb', 'worker', 'benchmark'] client, apps_client = util.init_k8s() def main(): args = sys.argv[1:] cmd = args[0] if cmd == 'send-conf': ip = args[1] conf = args[2] if len(args) > 2 else None