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!')
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 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
