def get_scheduler_serverset(cls, cluster, port=2181, verbose=False, **kw): if cluster.zk is None: raise ValueError('Cluster has no associated zookeeper ensemble!') if cluster.scheduler_zk_path is None: raise ValueError('Cluster has no defined scheduler path, must specify scheduler_zk_path ' 'in your cluster config!') zk = TwitterKazooClient.make(str('%s:%s' % (cluster.zk, port)), verbose=verbose) return zk, ServerSet(zk, cluster.scheduler_zk_path, **kw)
def get_scheduler_serverset(cls, cluster, port=2181, verbose=False, **kw): if cluster.zk is None: raise ValueError('Cluster has no associated zookeeper ensemble!') if cluster.scheduler_zk_path is None: raise ValueError('Cluster has no defined scheduler path, must specify scheduler_zk_path ' 'in your cluster config!') hosts = [h + ':{p}' for h in cluster.zk.split(',')] zk = TwitterKazooClient.make(str(','.join(hosts).format(p=port)), verbose=verbose) return zk, ServerSet(zk, cluster.scheduler_zk_path, **kw)
def get_scheduler_serverset(cls, cluster, port=2181, verbose=False, **kw): if cluster.zk is None: raise ValueError('Cluster has no associated zookeeper ensemble!') if cluster.scheduler_zk_path is None: raise ValueError( 'Cluster has no defined scheduler path, must specify scheduler_zk_path ' 'in your cluster config!') zk = TwitterKazooClient.make(str('%s:%s' % (cluster.zk, port)), verbose=verbose) return zk, ServerSet(zk, cluster.scheduler_zk_path, **kw)
def make_zk(cls, ensemble, **kw): if 'authentication' in kw: kw.update(auth_data = [kw.pop('authentication')]) tzk = TwitterKazooClient.make( ensemble + ('/%08d' % GroupTestBase.CHROOT_PREFIX), timeout=cls.CONNECT_TIMEOUT_SECS, max_retries=cls.CONNECT_RETRIES, **kw) started = threading.Event() def listen(state): if state == KazooState.CONNECTED: started.set() return True tzk.add_listener(listen) started.wait() tzk.ensure_path('/') return tzk
def make_zk(cls, ensemble, **kw): if 'authentication' in kw: kw.update(auth_data=[kw.pop('authentication')]) tzk = TwitterKazooClient.make(ensemble + ('/%08d' % GroupTestBase.CHROOT_PREFIX), timeout=cls.CONNECT_TIMEOUT_SECS, max_retries=cls.CONNECT_RETRIES, **kw) started = threading.Event() def listen(state): if state == KazooState.CONNECTED: started.set() return True tzk.add_listener(listen) started.wait() tzk.ensure_path('/') return tzk
def make_zk(cls, ensemble): return TwitterKazooClient.make(ensemble)
def sdzk(self): if self._sdzk is None: self._sdzk = TwitterKazooClient.make(hosts='%s:2181' % self.zookeeper_ensemble_url, verbose=True) log.debug('Connected to %s' % self._sdzk) return self._sdzk