def configure(self, hosts): """Configure the leaf scheduler. :param hosts: list of child hosts :type hosts: list of str """ # Transfer children's constraints from list to set, so searching # elements are more efficient. self._hosts = [] for host in hosts: self._hosts.append(ChildInfo.from_thrift(host)) self._coalesce_resources(self._hosts) if self._health_checker: self._health_checker.stop() if self._enable_health_checker: # initialize health checker with the new set of children. agent_config = common.services.get(ServiceName.AGENT_CONFIG) children = dict((host.id, ServerAddress(host.address, host.port)) for host in self._hosts) self._health_checker = HealthChecker(self._scheduler_id, children, agent_config) self._health_checker.start() self._configured = ConfigStates.INITIALIZED
def setUp(self): self.child_1 = ChildInfo("child_1", "1.1.1.1", 8835) self.child_2 = ChildInfo("child_2", "1.1.1.2", 8835) self.child_3 = ChildInfo("child_3", "1.1.1.3", 8835) self.child_4 = ChildInfo("child_4", "1.1.1.3", 8835) self.child_5 = ChildInfo("child_5", "1.1.1.3", 8835) self.child_6 = ChildInfo("child_5", "1.1.1.3", 8835) self.child_7 = ChildInfo("child_5", "1.1.1.3", 8835) self.request = PlaceRequest()
def configure(self, schedulers): """Configure the branch scheduler. :param schedulers: list of child scheduler ids :type schedulers: list of ChildInfo """ # Transfer children's constraints from list to set, so searching # elements are more efficient. self._schedulers = [] for scheduler in schedulers: self._schedulers.append(ChildInfo.from_thrift(scheduler)) self._coalesce_resources(self._schedulers)