class ZeroRpcClient(object):
def __init__(self):
- self.c = zerorpc.Client()
+ self.c = zerorpc.Client(heartbeat=None, timeout=120) #heartbeat=None, timeout=120
self.c.connect("tcp://127.0.0.1:4242") # TODO hard coded for now. we'll change this later
self.cmds = {}
def start(self, args):
nw_list = list()
if args.get("network") is not None:
- networks = args.get("network").split(",")
- for nw in networks:
- nw_list.append({"ip": nw})
+ nw_list = self._parse_network(args.get("network"))
+
r = self.c.compute_action_start(
args.get("datacenter"),
args.get("name"),
args.get("image"),
- args.get("docker_command"),
- nw_list)
+ nw_list,
+ args.get("docker_command")
+ )
pp.pprint(r)
def stop(self, args):
args.get("datacenter"), args.get("name"))
pp.pprint(r)
+ def profile(self, args):
+ nw_list = list()
+ if args.get("network") is not None:
+ nw_list = self._parse_network(args.get("network"))
+
+ params = self._create_dict(
+ network=nw_list,
+ command=args.get("docker_command"),
+ input=args.get("input"),
+ output=args.get("output"))
+
+ for output in self.c.compute_profile(
+ args.get("datacenter"),
+ args.get("name"),
+ args.get("image"),
+ params
+ ):
+ print(output + '\n')
+
+ #pp.pprint(r)
+ #print(r)
+
+ def _create_dict(self, **kwargs):
+ return kwargs
+
+ def _parse_network(self, network_str):
+ '''
+ parse the options for all network interfaces of the vnf
+ :param network_str: (id=x,ip=x.x.x.x/x), ...
+ :return: list of dicts [{"id":x,"ip":"x.x.x.x/x"}, ...]
+ '''
+ nw_list = list()
+ networks = network_str[1:-1].split('),(')
+ for nw in networks:
+ nw_dict = dict(tuple(e.split('=')) for e in nw.split(','))
+ nw_list.append(nw_dict)
+
+ return nw_list
+
+
parser = argparse.ArgumentParser(description='son-emu compute')
parser.add_argument(
"command",
- choices=['start', 'stop', 'list', 'status'],
+ choices=['start', 'stop', 'list', 'status', 'profile'],
help="Action to be executed.")
parser.add_argument(
"--datacenter", "-d", dest="datacenter",
help="Name of compute instance e.g. 'vnf1'")
parser.add_argument(
"--image","-i", dest="image",
- help="Name of container image to be used e.g. 'ubuntu'")
+ help="Name of container image to be used e.g. 'ubuntu:trusty'")
parser.add_argument(
"--dcmd", "-c", dest="docker_command",
help="Startup command of the container e.g. './start.sh'")
"--net", dest="network",
help="Network properties of compute instance e.g. \
'10.0.0.123/8' or '10.0.0.123/8,11.0.0.123/24' for multiple interfaces.")
+parser.add_argument(
+ "--input", "-in", dest="input",
+ help="input interface of the vnf to profile")
+parser.add_argument(
+ "--output", "-out", dest="output",
+ help="output interface of the vnf to profile")
def main(argv):