+ self.deactivate_cpu_limit = deactivate_cpu_limit
+ self.deactivate_mem_limit = deactivate_mem_limit
+ self.single_cu = 0
+ self.single_mu = 0
+ self.cpu_op_factor = 1.0 # over provisioning factor
+ self.mem_op_factor = 1.0
+ self.raise_no_cpu_resources_left = True
+ self.raise_no_mem_resources_left = True
+ super(UpbSimpleCloudDcRM, self).__init__()
+
+ def allocate(self, d):
+ """
+ Allocate resources for the given container.
+ Defined by d.flavor_name
+ :param d: container
+ :return:
+ """
+ self._allocated_compute_instances[d.name] = d
+ if not self.deactivate_cpu_limit:
+ self._allocate_cpu(d)
+ if not self.deactivate_mem_limit:
+ self._allocate_mem(d)
+ self._apply_limits()
+
+ def _allocate_cpu(self, d):
+ """
+ Actually allocate (bookkeeping)
+ :param d: container
+ :return:
+ """
+ fl_cu = self._get_flavor(d).get("compute")
+ # check for over provisioning
+ if self.dc_alloc_cu + fl_cu > self.dc_max_cu and self.raise_no_cpu_resources_left:
+ raise NotEnoughResourcesAvailable("Not enough compute resources left.")
+ self.dc_alloc_cu += fl_cu
+
+ def _allocate_mem(self, d):
+ """
+ Actually allocate (bookkeeping)
+ :param d: container
+ :return:
+ """
+ fl_mu = self._get_flavor(d).get("memory")
+ # check for over provisioning
+ if self.dc_alloc_mu + fl_mu > self.dc_max_mu and self.raise_no_mem_resources_left:
+ raise NotEnoughResourcesAvailable("Not enough memory resources left.")
+ self.dc_alloc_mu += fl_mu
+
+ def free(self, d):
+ """
+ Free resources allocated to the given container.
+ :param d: container
+ :return:
+ """
+ del self._allocated_compute_instances[d.name]
+ if not self.deactivate_cpu_limit:
+ self._free_cpu(d)
+ if not self.deactivate_mem_limit:
+ self._free_mem(d)
+ self._apply_limits()
+
+ def _free_cpu(self, d):
+ """
+ Free resources.
+ :param d: container
+ :return:
+ """
+ self.dc_alloc_cu -= self._get_flavor(d).get("compute")
+
+ def _free_mem(self, d):
+ """
+ Free resources.
+ :param d: container
+ :return:
+ """
+ self.dc_alloc_mu -= self._get_flavor(d).get("memory")
+
+ def _apply_limits(self):
+ """
+ Recalculate real resource limits for all allocated containers and apply them
+ to their cgroups.
+ We have to recalculate for all containers to allow e.g. over provisioning models.
+ :return:
+ """
+ for d in self._allocated_compute_instances.itervalues():
+ if not self.deactivate_cpu_limit:
+ self._apply_cpu_limits(d)
+ if not self.deactivate_mem_limit:
+ self._apply_mem_limits(d)
+
+ def _apply_cpu_limits(self, d):
+ """
+ Calculate real CPU limit (CFS bandwidth) and apply.
+ :param d: container
+ :return:
+ """
+ number_cu = self._get_flavor(d).get("compute")
+ # calculate cpu time fraction of a single compute unit
+ self.single_cu = self._compute_single_cu()
+ # calculate cpu time fraction for container with given flavor
+ cpu_time_percentage = self.single_cu * number_cu
+ # calculate input values for CFS scheduler bandwidth limitation
+ cpu_period, cpu_quota = self._calculate_cpu_cfs_values(cpu_time_percentage)
+ # apply limits to container if changed
+ if d.cpu_period != cpu_period or d.cpu_quota != cpu_quota:
+ LOG.debug("Setting CPU limit for %r: cpu_quota = cpu_period * limit = %f * %f = %f (op_factor=%f)" % (
+ d.name, cpu_period, cpu_time_percentage, cpu_quota, self.cpu_op_factor))
+ d.updateCpuLimit(cpu_period=int(cpu_period), cpu_quota=int(cpu_quota))