def __init__(self, *args, **kwargs): signal.signal(signal.SIGINT, self._handle_sig_event) signal.signal(signal.SIGTERM, self._handle_sig_event) self._prev_connection_state = None self._zoo_keeper = KazooClient(hosts=get_zk_conn_string( kwargs.get('environment')), timeout=60.0) self._zoo_keeper.start() self._configuration = Configuration(self._zoo_keeper, **kwargs) self._bootstrap_zookeeper_paths() self._task_server = TaskServer(self._configuration, self._zoo_keeper) self._data_store = DataStore(self._configuration, self._zoo_keeper, self._task_server) self._web_server = WebServer(self._configuration, self._data_store, self._task_server, self._zoo_keeper)
def _create_datastore(self): return DataStore(self.configuration, self.zoo_keeper, self.task_server)