def _init_function_servers(self):
budget_dir = os.path.join(self.working_dir, 'budget')
budget_cls = BudgetApplyServer if not self.is_multi_process \
else self.manager.budget_server
self.budget_server = budget_cls(budget_dir, self.settings, None,
self.job_name)
if self.rpc_server:
BudgetApplyServer.register_rpc(self.budget_server,
self.rpc_server,
app_name=self.job_name)
self.shutdown_callbacks.append(self.budget_server.shutdown)
counter_dir = os.path.join(self.working_dir, 'counter')
counter_cls = CounterServer if not self.is_multi_process \
else self.manager.counter_server
self.counter_server = counter_cls(counter_dir, self.settings, None,
self.job_name)
if self.rpc_server:
CounterServer.register_rpc(self.counter_server,
self.rpc_server,
app_name=self.job_name)
self.shutdown_callbacks.append(self.counter_server.shutdown)
speed_dir = os.path.join(self.working_dir, 'speed')
speed_cls = SpeedControlServer if not self.is_multi_process \
else self.manager.speed_server
self.speed_server = speed_cls(speed_dir, self.settings, None,
self.job_name, self.counter_server,
self.ctx.ips)
if self.rpc_server:
SpeedControlServer.register_rpc(self.speed_server,
self.rpc_server,
app_name=self.job_name)
self.shutdown_callbacks.append(self.speed_server.shutdown)
def _init_function_servers(self):
budget_dir = os.path.join(self.working_dir, 'budget')
budget_cls = BudgetApplyServer if not self.is_multi_process \
else self.manager.budget_server
self.budget_server = budget_cls(budget_dir, self.settings,
None, self.job_name)
if self.rpc_server:
BudgetApplyServer.register_rpc(self.budget_server, self.rpc_server,
app_name=self.job_name)
self.shutdown_callbacks.append(self.budget_server.shutdown)
counter_dir = os.path.join(self.working_dir, 'counter')
counter_cls = CounterServer if not self.is_multi_process \
else self.manager.counter_server
self.counter_server = counter_cls(counter_dir, self.settings,
None, self.job_name)
if self.rpc_server:
CounterServer.register_rpc(self.counter_server, self.rpc_server,
app_name=self.job_name)
self.shutdown_callbacks.append(self.counter_server.shutdown)
speed_dir = os.path.join(self.working_dir, 'speed')
speed_cls = SpeedControlServer if not self.is_multi_process \
else self.manager.speed_server
self.speed_server = speed_cls(speed_dir, self.settings,
None, self.job_name,
self.counter_server, self.ctx.ips)
if self.rpc_server:
SpeedControlServer.register_rpc(self.speed_server, self.rpc_server,
app_name=self.job_name)
self.shutdown_callbacks.append(self.speed_server.shutdown)
