X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=src%2Femuvim%2Fdcemulator%2Fnode.py;h=94d047bb8a08d83046197426e7e907cd42218a7a;hb=1f176c65521fc56f140aede047081b27341941d6;hp=9ccdb7f4de6b3f515769d947e25ec2b188a30a74;hpb=467b90856881c9422aa327958df060e515c736dc;p=osm%2Fvim-emu.git diff --git a/src/emuvim/dcemulator/node.py b/src/emuvim/dcemulator/node.py index 9ccdb7f..94d047b 100755 --- a/src/emuvim/dcemulator/node.py +++ b/src/emuvim/dcemulator/node.py @@ -5,6 +5,8 @@ Distributed Cloud Emulator (dcemulator) from mininet.node import Docker from mininet.link import Link import logging +import time +import json LOG = logging.getLogger("dcemulator") LOG.setLevel(logging.DEBUG) @@ -70,7 +72,7 @@ class Datacenter(object): DC_COUNTER = 1 - def __init__(self, label, metadata={}): + def __init__(self, label, metadata={}, resource_log_path=None): self.net = None # DCNetwork to which we belong # each node (DC) has a short internal name used by Mininet # this is caused by Mininets naming limitations for swtiches etc. @@ -80,6 +82,8 @@ class Datacenter(object): self.label = label # dict to store arbitrary metadata (e.g. latitude and longitude) self.metadata = metadata + # path to which resource information should be logged (e.g. for experiments). None = no logging + self.resource_log_path = resource_log_path # first prototype assumes one "bigswitch" per DC self.switch = None # keep track of running containers @@ -136,24 +140,46 @@ class Datacenter(object): if len(network) < 1: network.append({}) - # allocate in resource resource model and compute resource limits for new container - if self._resource_model is not None: - # TODO pass resource limits to new container (cf. Dockernet API) Issue #47 - (cpu_limit, mem_limit, disk_limit) = alloc = self._resource_model.allocate(name, flavor_name) - LOG.debug("Allocation result: %r" % str(alloc)) # create the container d = self.net.addDocker( "%s" % (name), dimage=image, dcmd=command, datacenter=self, - flavor_name=flavor_name) + flavor_name=flavor_name + ) + + # apply resource limits to container if a resource model is defined + if self._resource_model is not None: + self._resource_model.allocate(d) + # connect all given networks + # if no --net option is given, network = [{}], so 1 empty dict in the list + # this results in 1 default interface with a default ip address for nw in network: # TODO we cannot use TCLink here (see: https://github.com/mpeuster/dockernet/issues/3) self.net.addLink(d, self.switch, params1=nw, cls=Link) # do bookkeeping self.containers[name] = d + + # TODO re-enable logging + """ + # write resource log if a path is given + if self.resource_log_path is not None: + l = dict() + l["t"] = time.time() + l["name"] = name + l["compute"] = d.getStatus() + l["flavor_name"] = flavor_name + l["action"] = "allocate" + l["cpu_limit"] = cpu_limit + l["mem_limit"] = mem_limit + l["disk_limit"] = disk_limit + l["rm_state"] = None if self._resource_model is None else self._resource_model.get_state_dict() + # append to logfile + with open(self.resource_log_path, "a") as f: + f.write("%s\n" % json.dumps(l)) + """ return d # we might use UUIDs for naming later on def stopCompute(self, name): @@ -163,13 +189,37 @@ class Datacenter(object): assert name is not None if name not in self.containers: raise Exception("Container with name %s not found." % name) + LOG.debug("Stopping compute instance %r in data center %r" % (name, str(self))) + + # call resource model and free resources + if self._resource_model is not None: + self._resource_model.free(self.containers[name]) + + # remove links self.net.removeLink( link=None, node1=self.containers[name], node2=self.switch) + + # remove container self.net.removeDocker("%s" % (name)) del self.containers[name] - # call resource model and free resources - if self._resource_model is not None: - self._resource_model.free(name) + + # TODO re-enable logging + """ + # write resource log if a path is given + if self.resource_log_path is not None: + l = dict() + l["t"] = time.time() + l["name"] = name + l["flavor_name"] = None + l["action"] = "free" + l["cpu_limit"] = -1 + l["mem_limit"] = -1 + l["disk_limit"] = -1 + l["rm_state"] = None if self._resource_model is None else self._resource_model.get_state_dict() + # append to logfile + with open(self.resource_log_path, "a") as f: + f.write("%s\n" % json.dumps(l)) + """ return True def listCompute(self): @@ -192,6 +242,11 @@ class Datacenter(object): } def assignResourceModel(self, rm): + """ + Assign a resource model to this DC. + :param rm: a BaseResourceModel object + :return: + """ if self._resource_model is not None: raise Exception("There is already an resource model assigned to this DC.") self._resource_model = rm