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