zuul
/
nodepool
Code Issues Proposed changes

Merge "Update ZooKeeper class connection methods"

This commit is contained in:
Zuul 2022-06-29 20:40:40 +00:00 committed by Gerrit Code Review
parent 6386170914 7bbdfdc9fd
commit e668905f08
8 changed files with 178 additions and 184 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
nodepool/builder.py
View File

@ -599,6 +599,11 @@ class BuildWorker(BaseWorker):
interval, zk) interval, zk)
self.log = logging.getLogger("nodepool.builder.BuildWorker.%s" % name) self.log = logging.getLogger("nodepool.builder.BuildWorker.%s" % name)
self.name = 'BuildWorker.%s' % name self.name = 'BuildWorker.%s' % name
self._lost_zk_connection = False
zk.client.on_connection_lost_listeners.append(self._onConnectionLost)
def _onConnectionLost(self):
self._lost_zk_connection = True
def _getBuildLogRoot(self, name): def _getBuildLogRoot(self, name):
log_dir = self._config.build_log_dir log_dir = self._config.build_log_dir
@ -699,6 +704,7 @@ class BuildWorker(BaseWorker):
:returns: The updated ImageBuild data structure. :returns: The updated ImageBuild data structure.
''' '''
self._lost_zk_connection = False
data = zk.ImageBuild() data = zk.ImageBuild()
data.state = zk.BUILDING data.state = zk.BUILDING
data.builder_id = self._builder_id data.builder_id = self._builder_id
@ -953,9 +959,8 @@ class BuildWorker(BaseWorker):
if self._statsd: if self._statsd:
pipeline = self._statsd.pipeline() pipeline = self._statsd.pipeline()
if self._zk.didLoseConnection: if self._lost_zk_connection:
self.log.info("ZooKeeper lost while building %s" % diskimage.name) self.log.info("ZooKeeper lost while building %s" % diskimage.name)
self._zk.resetLostFlag()
build_data.state = zk.FAILED build_data.state = zk.FAILED
elif p.returncode or did_timeout: elif p.returncode or did_timeout:
self.log.info( self.log.info(

2
nodepool/launcher.py
View File

@ -381,7 +381,7 @@ class PoolWorker(threading.Thread, stats.StatsReporter):
# Make sure we're always registered with ZK # Make sure we're always registered with ZK
hostname = socket.gethostname() hostname = socket.gethostname()
self.component_info = PoolComponent( self.component_info = PoolComponent(
self.zk.zk_client, hostname, self.zk.client, hostname,
version=get_version_string()) version=get_version_string())
labels = set() labels = set()
for prov_cfg in self.nodepool.config.providers.values(): for prov_cfg in self.nodepool.config.providers.values():

8
nodepool/tests/__init__.py
View File

@ -689,21 +689,21 @@ class DBTestCase(BaseTestCase):
return a + b return a + b
return a + '/' + b return a + '/' + b
data, stat = self.zk.client.get(node) data, stat = self.zk.kazoo_client.get(node)
self.log.debug("Node: %s" % (node,)) self.log.debug("Node: %s" % (node,))
if data: if data:
self.log.debug(data) self.log.debug(data)
for child in self.zk.client.get_children(node): for child in self.zk.kazoo_client.get_children(node):
self.printZKTree(join(node, child)) self.printZKTree(join(node, child))
def getZKTree(self, path, ret=None): def getZKTree(self, path, ret=None):
"""Return the contents of a ZK tree as a dictionary""" """Return the contents of a ZK tree as a dictionary"""
if ret is None: if ret is None:
ret = {} ret = {}
for key in self.zk.client.get_children(path): for key in self.zk.kazoo_client.get_children(path):
subpath = os.path.join(path, key) subpath = os.path.join(path, key)
ret[subpath] = self.zk.client.get( ret[subpath] = self.zk.kazoo_client.get(
os.path.join(path, key))[0] os.path.join(path, key))[0]
self.getZKTree(subpath, ret) self.getZKTree(subpath, ret)
return ret return ret

2
nodepool/tests/unit/test_commands.py
View File

@ -488,7 +488,7 @@ class TestNodepoolCMD(tests.DBTestCase):
"-c", configfile, 'export-image-data', tf.name) "-c", configfile, 'export-image-data', tf.name)
nodepoolcmd.main() nodepoolcmd.main()
# Delete data from ZK # Delete data from ZK
self.zk.client.delete('/nodepool', recursive=True) self.zk.kazoo_client.delete('/nodepool', recursive=True)
self.patch_argv( self.patch_argv(
"-c", configfile, 'import-image-data', tf.name) "-c", configfile, 'import-image-data', tf.name)

10
nodepool/tests/unit/test_launcher.py
View File

@ -91,7 +91,7 @@ class TestLauncher(tests.DBTestCase):
# Verify the cleanup thread removed the lock # Verify the cleanup thread removed the lock
self.assertIsNotNone( self.assertIsNotNone(
self.zk.client.exists(self.zk._requestLockPath(req.id)) self.zk.kazoo_client.exists(self.zk._requestLockPath(req.id))
) )
self.zk.deleteNodeRequest(req) self.zk.deleteNodeRequest(req)
self.waitForNodeRequestLockDeletion(req.id) self.waitForNodeRequestLockDeletion(req.id)
@ -766,7 +766,7 @@ class TestLauncher(tests.DBTestCase):
# than what we are going to request. # than what we are going to request.
hostname = socket.gethostname() hostname = socket.gethostname()
dummy_component = PoolComponent( dummy_component = PoolComponent(
self.zk.zk_client, hostname, self.zk.client, hostname,
version=get_version_string()) version=get_version_string())
dummy_component.content.update({ dummy_component.content.update({
'id': 'dummy', 'id': 'dummy',
@ -2426,14 +2426,14 @@ class TestLauncher(tests.DBTestCase):
# Create empty node # Create empty node
path = "%s" % self.zk._nodePath("12345") path = "%s" % self.zk._nodePath("12345")
self.log.debug("node path %s", path) self.log.debug("node path %s", path)
self.zk.client.create(path, makepath=True) self.zk.kazoo_client.create(path, makepath=True)
self.assertTrue(self.zk.client.exists(path)) self.assertTrue(self.zk.kazoo_client.exists(path))
pool = self.useNodepool(configfile, watermark_sleep=1) pool = self.useNodepool(configfile, watermark_sleep=1)
pool.cleanup_interval = .1 pool.cleanup_interval = .1
pool.start() pool.start()
while self.zk.client.exists(path): while self.zk.kazoo_client.exists(path):
time.sleep(.1) time.sleep(.1)
def test_leaked_port_cleanup(self): def test_leaked_port_cleanup(self):

108
nodepool/tests/unit/test_zk.py
View File

@ -30,7 +30,7 @@ class TestComponentRegistry(tests.DBTestCase):
def test_pool_component(self): def test_pool_component(self):
hostname = socket.gethostname() hostname = socket.gethostname()
launcher = PoolComponent( launcher = PoolComponent(
self.zk.zk_client, hostname, self.zk.client, hostname,
version=get_version_string()) version=get_version_string())
launcher.content.update({ launcher.content.update({
'id': "launcher-Poolworker.provider-main-" + uuid.uuid4().hex, 'id': "launcher-Poolworker.provider-main-" + uuid.uuid4().hex,
@ -92,7 +92,7 @@ class TestZooKeeper(tests.DBTestCase):
def test_imageBuildLock(self): def test_imageBuildLock(self):
path = self.zk._imageBuildLockPath("ubuntu-trusty") path = self.zk._imageBuildLockPath("ubuntu-trusty")
with self.zk.imageBuildLock("ubuntu-trusty", blocking=False): with self.zk.imageBuildLock("ubuntu-trusty", blocking=False):
self.assertIsNotNone(self.zk.client.exists(path)) self.assertIsNotNone(self.zk.kazoo_client.exists(path))
def test_imageBuildLock_exception_nonblocking(self): def test_imageBuildLock_exception_nonblocking(self):
image = "ubuntu-trusty" image = "ubuntu-trusty"
@ -115,7 +115,7 @@ class TestZooKeeper(tests.DBTestCase):
with self.zk.imageBuildNumberLock( with self.zk.imageBuildNumberLock(
"ubuntu-trusty", "0000", blocking=False "ubuntu-trusty", "0000", blocking=False
): ):
self.assertIsNotNone(self.zk.client.exists(path)) self.assertIsNotNone(self.zk.kazoo_client.exists(path))
def test_imageBuildNumberLock_exception_nonblocking(self): def test_imageBuildNumberLock_exception_nonblocking(self):
image = "ubuntu-trusty" image = "ubuntu-trusty"
@ -152,12 +152,12 @@ class TestZooKeeper(tests.DBTestCase):
upload.provider_name, upload.provider_name,
upload.id) upload.id)
with self.zk.imageUploadNumberLock(upload, blocking=False): with self.zk.imageUploadNumberLock(upload, blocking=False):
self.assertIsNotNone(self.zk.client.exists(path)) self.assertIsNotNone(self.zk.kazoo_client.exists(path))
self.zk.deleteUpload(upload.image_name, self.zk.deleteUpload(upload.image_name,
upload.build_id, upload.build_id,
upload.provider_name, upload.provider_name,
upload.id) upload.id)
self.assertIsNone(self.zk.client.exists(path)) self.assertIsNone(self.zk.kazoo_client.exists(path))
def test_imageUploadNumberLock_orphan(self): def test_imageUploadNumberLock_orphan(self):
upload = zk.ImageUpload() upload = zk.ImageUpload()
@ -171,7 +171,7 @@ class TestZooKeeper(tests.DBTestCase):
upload.id) upload.id)
with self.zk.imageUploadNumberLock(upload, blocking=False): with self.zk.imageUploadNumberLock(upload, blocking=False):
self.assertIsNotNone(self.zk.client.exists(path)) self.assertIsNotNone(self.zk.kazoo_client.exists(path))
self.zk.deleteUpload(upload.image_name, self.zk.deleteUpload(upload.image_name,
upload.build_id, upload.build_id,
upload.provider_name, upload.provider_name,
@ -183,7 +183,7 @@ class TestZooKeeper(tests.DBTestCase):
# We now recreated an empty image upload number node # We now recreated an empty image upload number node
pass pass
self.assertIsNotNone(self.zk.client.exists(path)) self.assertIsNotNone(self.zk.kazoo_client.exists(path))
# Should not throw an exception because of the empty upload # Should not throw an exception because of the empty upload
self.zk.getImageUpload(upload.image_name, upload.build_id, self.zk.getImageUpload(upload.image_name, upload.build_id,
@ -200,7 +200,7 @@ class TestZooKeeper(tests.DBTestCase):
upload.provider_name, upload.provider_name,
upload.id) upload.id)
with self.zk.imageUploadNumberLock(upload, blocking=False): with self.zk.imageUploadNumberLock(upload, blocking=False):
self.assertIsNotNone(self.zk.client.exists(path)) self.assertIsNotNone(self.zk.kazoo_client.exists(path))
def test_imageUploadNumberLock_exception_nonblocking(self): def test_imageUploadNumberLock_exception_nonblocking(self):
upload = zk.ImageUpload() upload = zk.ImageUpload()
@ -232,7 +232,7 @@ class TestZooKeeper(tests.DBTestCase):
path = self.zk._imageUploadLockPath("ubuntu-trusty", "0000", "prov1") path = self.zk._imageUploadLockPath("ubuntu-trusty", "0000", "prov1")
with self.zk.imageUploadLock("ubuntu-trusty", "0000", "prov1", with self.zk.imageUploadLock("ubuntu-trusty", "0000", "prov1",
blocking=False): blocking=False):
self.assertIsNotNone(self.zk.client.exists(path)) self.assertIsNotNone(self.zk.kazoo_client.exists(path))
def test_imageUploadLock_exception_nonblocking(self): def test_imageUploadLock_exception_nonblocking(self):
image = "ubuntu-trusty" image = "ubuntu-trusty"
@ -372,7 +372,7 @@ class TestZooKeeper(tests.DBTestCase):
# We now created an empty build number node # We now created an empty build number node
pass pass
self.assertIsNotNone(self.zk.client.exists(path)) self.assertIsNotNone(self.zk.kazoo_client.exists(path))
# Should not throw an exception because of the empty upload # Should not throw an exception because of the empty upload
self.assertIsNone(self.zk.getBuild(image, build_number)) self.assertIsNone(self.zk.getBuild(image, build_number))
@ -501,11 +501,15 @@ class TestZooKeeper(tests.DBTestCase):
v3.state = zk.FAILED v3.state = zk.FAILED
v4 = zk.ImageBuild() v4 = zk.ImageBuild()
v4.state = zk.DELETING v4.state = zk.DELETING
self.zk.client.create(path + "/1", value=v1.serialize(), makepath=True) self.zk.kazoo_client.create(path + "/1", value=v1.serialize(),
self.zk.client.create(path + "/2", value=v2.serialize(), makepath=True) makepath=True)
self.zk.client.create(path + "/3", value=v3.serialize(), makepath=True) self.zk.kazoo_client.create(path + "/2", value=v2.serialize(),
self.zk.client.create(path + "/4", value=v4.serialize(), makepath=True) makepath=True)
self.zk.client.create(path + "/lock", makepath=True) self.zk.kazoo_client.create(path + "/3", value=v3.serialize(),
makepath=True)
self.zk.kazoo_client.create(path + "/4", value=v4.serialize(),
makepath=True)
self.zk.kazoo_client.create(path + "/lock", makepath=True)
matches = self.zk.getBuilds(image, None) matches = self.zk.getBuilds(image, None)
self.assertEqual(4, len(matches)) self.assertEqual(4, len(matches))
@ -521,11 +525,15 @@ class TestZooKeeper(tests.DBTestCase):
v3.state = zk.FAILED v3.state = zk.FAILED
v4 = zk.ImageBuild() v4 = zk.ImageBuild()
v4.state = zk.DELETING v4.state = zk.DELETING
self.zk.client.create(path + "/1", value=v1.serialize(), makepath=True) self.zk.kazoo_client.create(path + "/1", value=v1.serialize(),
self.zk.client.create(path + "/2", value=v2.serialize(), makepath=True) makepath=True)
self.zk.client.create(path + "/3", value=v3.serialize(), makepath=True) self.zk.kazoo_client.create(path + "/2", value=v2.serialize(),
self.zk.client.create(path + "/4", value=v4.serialize(), makepath=True) makepath=True)
self.zk.client.create(path + "/lock", makepath=True) self.zk.kazoo_client.create(path + "/3", value=v3.serialize(),
makepath=True)
self.zk.kazoo_client.create(path + "/4", value=v4.serialize(),
makepath=True)
self.zk.kazoo_client.create(path + "/lock", makepath=True)
matches = self.zk.getBuilds(image, [zk.DELETING, zk.FAILED]) matches = self.zk.getBuilds(image, [zk.DELETING, zk.FAILED])
self.assertEqual(2, len(matches)) self.assertEqual(2, len(matches))
@ -540,11 +548,15 @@ class TestZooKeeper(tests.DBTestCase):
v3.state = zk.FAILED v3.state = zk.FAILED
v4 = zk.ImageUpload() v4 = zk.ImageUpload()
v4.state = zk.DELETING v4.state = zk.DELETING
self.zk.client.create(path + "/1", value=v1.serialize(), makepath=True) self.zk.kazoo_client.create(path + "/1", value=v1.serialize(),
self.zk.client.create(path + "/2", value=v2.serialize(), makepath=True) makepath=True)
self.zk.client.create(path + "/3", value=v3.serialize(), makepath=True) self.zk.kazoo_client.create(path + "/2", value=v2.serialize(),
self.zk.client.create(path + "/4", value=v4.serialize(), makepath=True) makepath=True)
self.zk.client.create(path + "/lock", makepath=True) self.zk.kazoo_client.create(path + "/3", value=v3.serialize(),
makepath=True)
self.zk.kazoo_client.create(path + "/4", value=v4.serialize(),
makepath=True)
self.zk.kazoo_client.create(path + "/lock", makepath=True)
matches = self.zk.getUploads("trusty", "000", "rax", matches = self.zk.getUploads("trusty", "000", "rax",
[zk.DELETING, zk.FAILED]) [zk.DELETING, zk.FAILED])
@ -560,11 +572,15 @@ class TestZooKeeper(tests.DBTestCase):
v3.state = zk.FAILED v3.state = zk.FAILED
v4 = zk.ImageUpload() v4 = zk.ImageUpload()
v4.state = zk.DELETING v4.state = zk.DELETING
self.zk.client.create(path + "/1", value=v1.serialize(), makepath=True) self.zk.kazoo_client.create(path + "/1", value=v1.serialize(),
self.zk.client.create(path + "/2", value=v2.serialize(), makepath=True) makepath=True)
self.zk.client.create(path + "/3", value=v3.serialize(), makepath=True) self.zk.kazoo_client.create(path + "/2", value=v2.serialize(),
self.zk.client.create(path + "/4", value=v4.serialize(), makepath=True) makepath=True)
self.zk.client.create(path + "/lock", makepath=True) self.zk.kazoo_client.create(path + "/3", value=v3.serialize(),
makepath=True)
self.zk.kazoo_client.create(path + "/4", value=v4.serialize(),
makepath=True)
self.zk.kazoo_client.create(path + "/lock", makepath=True)
matches = self.zk.getUploads("trusty", "000", "rax", None) matches = self.zk.getUploads("trusty", "000", "rax", None)
self.assertEqual(4, len(matches)) self.assertEqual(4, len(matches))
@ -592,9 +608,9 @@ class TestZooKeeper(tests.DBTestCase):
def test_deleteUpload(self): def test_deleteUpload(self):
path = self.zk._imageUploadPath("trusty", "000", "rax") + "/000001" path = self.zk._imageUploadPath("trusty", "000", "rax") + "/000001"
self.zk.client.create(path, makepath=True) self.zk.kazoo_client.create(path, makepath=True)
self.zk.deleteUpload("trusty", "000", "rax", "000001") self.zk.deleteUpload("trusty", "000", "rax", "000001")
self.assertIsNone(self.zk.client.exists(path)) self.assertIsNone(self.zk.kazoo_client.exists(path))
def test_getNodeRequests_empty(self): def test_getNodeRequests_empty(self):
self.assertEqual([], self.zk.getNodeRequests()) self.assertEqual([], self.zk.getNodeRequests())
@ -604,10 +620,10 @@ class TestZooKeeper(tests.DBTestCase):
r2 = self.zk._requestPath("100-456") r2 = self.zk._requestPath("100-456")
r3 = self.zk._requestPath("100-123") r3 = self.zk._requestPath("100-123")
r4 = self.zk._requestPath("400-123") r4 = self.zk._requestPath("400-123")
self.zk.client.create(r1, makepath=True, ephemeral=True) self.zk.kazoo_client.create(r1, makepath=True, ephemeral=True)
self.zk.client.create(r2, makepath=True, ephemeral=True) self.zk.kazoo_client.create(r2, makepath=True, ephemeral=True)
self.zk.client.create(r3, makepath=True, ephemeral=True) self.zk.kazoo_client.create(r3, makepath=True, ephemeral=True)
self.zk.client.create(r4, makepath=True, ephemeral=True) self.zk.kazoo_client.create(r4, makepath=True, ephemeral=True)
self.assertEqual( self.assertEqual(
["100-123", "100-456", "400-123", "500-123"], ["100-123", "100-456", "400-123", "500-123"],
@ -618,8 +634,8 @@ class TestZooKeeper(tests.DBTestCase):
r = zk.NodeRequest("500-123") r = zk.NodeRequest("500-123")
r.state = zk.REQUESTED r.state = zk.REQUESTED
path = self.zk._requestPath(r.id) path = self.zk._requestPath(r.id)
self.zk.client.create(path, value=r.serialize(), self.zk.kazoo_client.create(path, value=r.serialize(),
makepath=True, ephemeral=True) makepath=True, ephemeral=True)
o = self.zk.getNodeRequest(r.id) o = self.zk.getNodeRequest(r.id)
self.assertIsInstance(o, zk.NodeRequest) self.assertIsInstance(o, zk.NodeRequest)
self.assertEqual(r.id, o.id) self.assertEqual(r.id, o.id)
@ -628,8 +644,8 @@ class TestZooKeeper(tests.DBTestCase):
self.assertIsNone(self.zk.getNodeRequest("invalid")) self.assertIsNone(self.zk.getNodeRequest("invalid"))
def test_getNodes(self): def test_getNodes(self):
self.zk.client.create(self.zk._nodePath('100'), makepath=True) self.zk.kazoo_client.create(self.zk._nodePath('100'), makepath=True)
self.zk.client.create(self.zk._nodePath('200'), makepath=True) self.zk.kazoo_client.create(self.zk._nodePath('200'), makepath=True)
nodes = self.zk.getNodes() nodes = self.zk.getNodes()
self.assertIn('100', nodes) self.assertIn('100', nodes)
self.assertIn('200', nodes) self.assertIn('200', nodes)
@ -638,7 +654,7 @@ class TestZooKeeper(tests.DBTestCase):
n = zk.Node('100') n = zk.Node('100')
n.state = zk.BUILDING n.state = zk.BUILDING
path = self.zk._nodePath(n.id) path = self.zk._nodePath(n.id)
self.zk.client.create(path, value=n.serialize(), makepath=True) self.zk.kazoo_client.create(path, value=n.serialize(), makepath=True)
o = self.zk.getNode(n.id) o = self.zk.getNode(n.id)
self.assertIsInstance(o, zk.Node) self.assertIsInstance(o, zk.Node)
self.assertEqual(n.id, o.id) self.assertEqual(n.id, o.id)
@ -659,7 +675,7 @@ class TestZooKeeper(tests.DBTestCase):
self.zk.lockNode(node) self.zk.lockNode(node)
self.assertIsNotNone(node.lock) self.assertIsNotNone(node.lock)
self.assertIsNotNone( self.assertIsNotNone(
self.zk.client.exists(self.zk._nodeLockPath(node.id)) self.zk.kazoo_client.exists(self.zk._nodeLockPath(node.id))
) )
self.zk.unlockNode(node) self.zk.unlockNode(node)
self.assertIsNone(node.lock) self.assertIsNone(node.lock)
@ -678,7 +694,7 @@ class TestZooKeeper(tests.DBTestCase):
self.zk.storeNode(node) self.zk.storeNode(node)
self.assertIsNotNone(node.id) self.assertIsNotNone(node.id)
self.assertIsNotNone( self.assertIsNotNone(
self.zk.client.exists(self.zk._nodePath(node.id)) self.zk.kazoo_client.exists(self.zk._nodePath(node.id))
) )
return node return node
@ -700,7 +716,7 @@ class TestZooKeeper(tests.DBTestCase):
req.node_types.append('label1') req.node_types.append('label1')
self.zk.storeNodeRequest(req) self.zk.storeNodeRequest(req)
self.assertIsNotNone( self.assertIsNotNone(
self.zk.client.exists(self.zk._requestPath(req.id)) self.zk.kazoo_client.exists(self.zk._requestPath(req.id))
) )
return req return req
@ -722,14 +738,14 @@ class TestZooKeeper(tests.DBTestCase):
req = self._create_node_request() req = self._create_node_request()
self.zk.deleteNodeRequest(req) self.zk.deleteNodeRequest(req)
self.assertIsNone( self.assertIsNone(
self.zk.client.exists(self.zk._requestPath(req.id)) self.zk.kazoo_client.exists(self.zk._requestPath(req.id))
) )
def test_deleteNode(self): def test_deleteNode(self):
n1 = self._create_node() n1 = self._create_node()
self.zk.deleteNode(n1) self.zk.deleteNode(n1)
self.assertIsNone( self.assertIsNone(
self.zk.client.exists(self.zk._nodePath(n1.id)) self.zk.kazoo_client.exists(self.zk._nodePath(n1.id))
) )
def test_getReadyNodesOfTypes(self): def test_getReadyNodesOfTypes(self):

219
nodepool/zk/zookeeper.py
View File

@ -18,7 +18,6 @@ import logging
import time import time
import uuid import uuid
from kazoo.client import KazooState
from kazoo import exceptions as kze from kazoo import exceptions as kze
from kazoo.recipe.lock import Lock from kazoo.recipe.lock import Lock
from kazoo.recipe.cache import TreeCache, TreeEvent from kazoo.recipe.cache import TreeCache, TreeEvent
@ -27,6 +26,7 @@ from kazoo.recipe.election import Election
from nodepool import exceptions as npe from nodepool import exceptions as npe
from nodepool.logconfig import get_annotated_logger from nodepool.logconfig import get_annotated_logger
from nodepool.zk.components import COMPONENT_REGISTRY from nodepool.zk.components import COMPONENT_REGISTRY
from nodepool.zk import ZooKeeperBase
# States: # States:
# We are building this image (or node) but it is not ready for use. # We are building this image (or node) but it is not ready for use.
@ -676,7 +676,7 @@ class Node(BaseModel):
self.requestor = d.get('requestor') self.requestor = d.get('requestor')
class ZooKeeper(object): class ZooKeeper(ZooKeeperBase):
''' '''
Class implementing the ZooKeeper interface. Class implementing the ZooKeeper interface.
@ -704,13 +704,11 @@ class ZooKeeper(object):
# Log zookeeper retry every 10 seconds # Log zookeeper retry every 10 seconds
retry_log_rate = 10 retry_log_rate = 10
def __init__(self, zk_client, enable_cache=True): def __init__(self, client, enable_cache=True):
''' '''
Initialize the ZooKeeper object. Initialize the ZooKeeper object.
''' '''
self.zk_client = zk_client # nodepool.zk.ZooKeeperClient super().__init__(client)
self.client = zk_client.client # KazooClient
self._became_lost = False
self._last_retry_log = 0 self._last_retry_log = 0
self._node_cache = None self._node_cache = None
self._request_cache = None self._request_cache = None
@ -719,22 +717,35 @@ class ZooKeeper(object):
self.enable_cache = enable_cache self.enable_cache = enable_cache
self.node_stats_event = None self.node_stats_event = None
if self.enable_cache: if self.client.connected:
self._node_cache = TreeCache(self.client, self.NODE_ROOT) self._onConnect()
self._node_cache.listen_fault(self.cacheFaultListener)
self._node_cache.listen(self.nodeCacheListener)
self._node_cache.start()
self._request_cache = TreeCache(self.client, self.REQUEST_ROOT) COMPONENT_REGISTRY.create(self.client)
self._request_cache.listen_fault(self.cacheFaultListener)
self._request_cache.listen(self.requestCacheListener)
self._request_cache.start()
COMPONENT_REGISTRY.create(self.zk_client)
# ======================================================================= # =======================================================================
# Private Methods # Private Methods
# ======================================================================= # =======================================================================
def _onConnect(self):
if self.enable_cache:
self._node_cache = TreeCache(self.kazoo_client, self.NODE_ROOT)
self._node_cache.listen_fault(self.cacheFaultListener)
self._node_cache.listen(self.nodeCacheListener)
self._node_cache.start()
self._request_cache = TreeCache(self.kazoo_client,
self.REQUEST_ROOT)
self._request_cache.listen_fault(self.cacheFaultListener)
self._request_cache.listen(self.requestCacheListener)
self._request_cache.start()
def _onDisconnect(self):
if self._node_cache is not None:
self._node_cache.close()
self._node_cache = None
if self._request_cache is not None:
self._request_cache.close()
self._request_cache = None
def _electionPath(self, election): def _electionPath(self, election):
return "%s/%s" % (self.ELECTION_ROOT, election) return "%s/%s" % (self.ELECTION_ROOT, election)
@ -800,7 +811,7 @@ class ZooKeeper(object):
def _getImageBuildLock(self, image, blocking=True, timeout=None): def _getImageBuildLock(self, image, blocking=True, timeout=None):
lock_path = self._imageBuildLockPath(image) lock_path = self._imageBuildLockPath(image)
try: try:
lock = Lock(self.client, lock_path) lock = Lock(self.kazoo_client, lock_path)
have_lock = lock.acquire(blocking, timeout) have_lock = lock.acquire(blocking, timeout)
except kze.LockTimeout: except kze.LockTimeout:
raise npe.TimeoutException( raise npe.TimeoutException(
@ -820,7 +831,7 @@ class ZooKeeper(object):
blocking=True, timeout=None): blocking=True, timeout=None):
lock_path = self._imageBuildNumberLockPath(image, build_number) lock_path = self._imageBuildNumberLockPath(image, build_number)
try: try:
lock = Lock(self.client, lock_path) lock = Lock(self.kazoo_client, lock_path)
have_lock = lock.acquire(blocking, timeout) have_lock = lock.acquire(blocking, timeout)
except kze.LockTimeout: except kze.LockTimeout:
raise npe.TimeoutException( raise npe.TimeoutException(
@ -842,7 +853,7 @@ class ZooKeeper(object):
lock_path = self._imageUploadNumberLockPath(image, build_number, lock_path = self._imageUploadNumberLockPath(image, build_number,
provider, upload_number) provider, upload_number)
try: try:
lock = Lock(self.client, lock_path) lock = Lock(self.kazoo_client, lock_path)
have_lock = lock.acquire(blocking, timeout) have_lock = lock.acquire(blocking, timeout)
except kze.LockTimeout: except kze.LockTimeout:
raise npe.TimeoutException( raise npe.TimeoutException(
@ -864,7 +875,7 @@ class ZooKeeper(object):
blocking=True, timeout=None): blocking=True, timeout=None):
lock_path = self._imageUploadLockPath(image, build_number, provider) lock_path = self._imageUploadLockPath(image, build_number, provider)
try: try:
lock = Lock(self.client, lock_path) lock = Lock(self.kazoo_client, lock_path)
have_lock = lock.acquire(blocking, timeout) have_lock = lock.acquire(blocking, timeout)
except kze.LockTimeout: except kze.LockTimeout:
raise npe.TimeoutException( raise npe.TimeoutException(
@ -881,20 +892,6 @@ class ZooKeeper(object):
return lock return lock
def _connection_listener(self, state):
'''
Listener method for Kazoo connection state changes.
.. warning:: This method must not block.
'''
if state == KazooState.LOST:
self.log.debug("ZooKeeper connection: LOST")
self._became_lost = True
elif state == KazooState.SUSPENDED:
self.log.debug("ZooKeeper connection: SUSPENDED")
else:
self.log.debug("ZooKeeper connection: CONNECTED")
def logConnectionRetryEvent(self): def logConnectionRetryEvent(self):
''' '''
Kazoo retry callback Kazoo retry callback
@ -910,28 +907,15 @@ class ZooKeeper(object):
@property @property
def connected(self): def connected(self):
if self.client is None: return self.client.connected
return False
return self.client.state == KazooState.CONNECTED
@property @property
def suspended(self): def suspended(self):
if self.client is None: return self.client.suspended
return True
return self.client.state == KazooState.SUSPENDED
@property @property
def lost(self): def lost(self):
if self.client is None: return self.client.lost
return True
return self.client.state == KazooState.LOST
@property
def didLoseConnection(self):
return self._became_lost
def resetLostFlag(self):
self._became_lost = False
def disconnect(self): def disconnect(self):
''' '''
@ -941,18 +925,7 @@ class ZooKeeper(object):
cluster connection. cluster connection.
''' '''
if self._node_cache is not None: self.client.disconnect()
self._node_cache.close()
self._node_cache = None
if self._request_cache is not None:
self._request_cache.close()
self._request_cache = None
if self.client is not None and self.client.connected:
self.client.stop()
self.client.close()
self.client = None
def resetHosts(self, hosts): def resetHosts(self, hosts):
''' '''
@ -960,8 +933,7 @@ class ZooKeeper(object):
:param str host_list: A ZK host list :param str host_list: A ZK host list
''' '''
if self.client is not None: self.client.resetHosts(hosts)
self.client.set_hosts(hosts=hosts)
@contextmanager @contextmanager
def imageBuildLock(self, image, blocking=True, timeout=None): def imageBuildLock(self, image, blocking=True, timeout=None):
@ -1086,7 +1058,7 @@ class ZooKeeper(object):
path = self.IMAGE_ROOT path = self.IMAGE_ROOT
try: try:
images = self.client.get_children(path) images = self.kazoo_client.get_children(path)
except kze.NoNodeError: except kze.NoNodeError:
return [] return []
return sorted(images) return sorted(images)
@ -1100,7 +1072,7 @@ class ZooKeeper(object):
path = self._imagePausePath(image) path = self._imagePausePath(image)
try: try:
data, stat = self.client.get(path) data, stat = self.kazoo_client.get(path)
except kze.NoNodeError: except kze.NoNodeError:
return False return False
return True return True
@ -1113,12 +1085,12 @@ class ZooKeeper(object):
if paused: if paused:
try: try:
self.client.create(path) self.kazoo_client.create(path)
except kze.NodeExistsError: except kze.NodeExistsError:
pass pass
else: else:
try: try:
self.client.delete(path) self.kazoo_client.delete(path)
except kze.NoNodeError: except kze.NoNodeError:
pass pass
@ -1133,7 +1105,7 @@ class ZooKeeper(object):
path = self._imageBuildsPath(image) path = self._imageBuildsPath(image)
try: try:
builds = self.client.get_children(path) builds = self.kazoo_client.get_children(path)
except kze.NoNodeError: except kze.NoNodeError:
return [] return []
builds = [x for x in builds if x != 'lock'] builds = [x for x in builds if x != 'lock']
@ -1152,7 +1124,7 @@ class ZooKeeper(object):
path = self._imageProviderPath(image, build_number) path = self._imageProviderPath(image, build_number)
try: try:
providers = self.client.get_children(path) providers = self.kazoo_client.get_children(path)
except kze.NoNodeError: except kze.NoNodeError:
return [] return []
@ -1172,7 +1144,7 @@ class ZooKeeper(object):
path = self._imageUploadPath(image, build_number, provider) path = self._imageUploadPath(image, build_number, provider)
try: try:
uploads = self.client.get_children(path) uploads = self.kazoo_client.get_children(path)
except kze.NoNodeError: except kze.NoNodeError:
return [] return []
@ -1191,7 +1163,7 @@ class ZooKeeper(object):
path = self._imageBuildsPath(image) + "/%s" % build_number path = self._imageBuildsPath(image) + "/%s" % build_number
try: try:
data, stat = self.client.get(path) data, stat = self.kazoo_client.get(path)
except kze.NoNodeError: except kze.NoNodeError:
return None return None
@ -1218,7 +1190,7 @@ class ZooKeeper(object):
path = self._imageBuildsPath(image) path = self._imageBuildsPath(image)
try: try:
builds = self.client.get_children(path) builds = self.kazoo_client.get_children(path)
except kze.NoNodeError: except kze.NoNodeError:
return [] return []
@ -1285,7 +1257,7 @@ class ZooKeeper(object):
build_path = self._imageBuildsPath(image) + "/" build_path = self._imageBuildsPath(image) + "/"
if build_number is None: if build_number is None:
path = self.client.create( path = self.kazoo_client.create(
build_path, build_path,
value=build_data.serialize(), value=build_data.serialize(),
sequence=True, sequence=True,
@ -1293,7 +1265,7 @@ class ZooKeeper(object):
build_number = path.split("/")[-1] build_number = path.split("/")[-1]
else: else:
path = build_path + build_number path = build_path + build_number
self.client.set(path, build_data.serialize()) self.kazoo_client.set(path, build_data.serialize())
return build_number return build_number
@ -1314,7 +1286,7 @@ class ZooKeeper(object):
path = path + "/%s" % upload_number path = path + "/%s" % upload_number
try: try:
data, stat = self.client.get(path) data, stat = self.kazoo_client.get(path)
except kze.NoNodeError: except kze.NoNodeError:
return None return None
@ -1347,7 +1319,7 @@ class ZooKeeper(object):
path = self._imageUploadPath(image, build_number, provider) path = self._imageUploadPath(image, build_number, provider)
try: try:
uploads = self.client.get_children(path) uploads = self.kazoo_client.get_children(path)
except kze.NoNodeError: except kze.NoNodeError:
return [] return []
@ -1411,7 +1383,7 @@ class ZooKeeper(object):
path = self._imageUploadPath(image, build_number, provider) path = self._imageUploadPath(image, build_number, provider)
try: try:
uploads = self.client.get_children(path) uploads = self.kazoo_client.get_children(path)
except kze.NoNodeError: except kze.NoNodeError:
uploads = [] uploads = []
@ -1454,7 +1426,7 @@ class ZooKeeper(object):
''' '''
# We expect the image builds path to already exist. # We expect the image builds path to already exist.
build_path = self._imageBuildsPath(image) build_path = self._imageBuildsPath(image)
if not self.client.exists(build_path): if not self.kazoo_client.exists(build_path):
raise npe.ZKException( raise npe.ZKException(
"Cannot find build %s of image %s" % (build_number, image) "Cannot find build %s of image %s" % (build_number, image)
) )
@ -1466,7 +1438,7 @@ class ZooKeeper(object):
image, build_number, provider) + "/" image, build_number, provider) + "/"
if upload_number is None: if upload_number is None:
path = self.client.create( path = self.kazoo_client.create(
upload_path, upload_path,
value=image_data.serialize(), value=image_data.serialize(),
sequence=True, sequence=True,
@ -1474,7 +1446,7 @@ class ZooKeeper(object):
upload_number = path.split("/")[-1] upload_number = path.split("/")[-1]
else: else:
path = upload_path + upload_number path = upload_path + upload_number
self.client.set(path, image_data.serialize()) self.kazoo_client.set(path, image_data.serialize())
return upload_number return upload_number
@ -1487,7 +1459,7 @@ class ZooKeeper(object):
:returns: True if request is pending, False otherwise :returns: True if request is pending, False otherwise
''' '''
path = self._imageBuildRequestPath(image) path = self._imageBuildRequestPath(image)
if self.client.exists(path) is not None: if self.kazoo_client.exists(path) is not None:
return True return True
return False return False
@ -1498,7 +1470,7 @@ class ZooKeeper(object):
latest_build, *_ = builds latest_build, *_ = builds
builds_path = self._imageBuildNumberPath(image, latest_build) builds_path = self._imageBuildNumberPath(image, latest_build)
return self.client.exists(builds_path) return self.kazoo_client.exists(builds_path)
def getBuildRequest(self, image): def getBuildRequest(self, image):
"""Get a build request for the given image. """Get a build request for the given image.
@ -1509,13 +1481,13 @@ class ZooKeeper(object):
""" """
path = self._imageBuildRequestPath(image) path = self._imageBuildRequestPath(image)
try: try:
_, stat = self.client.get(path) _, stat = self.kazoo_client.get(path)
except kze.NoNodeError: except kze.NoNodeError:
return return
pending = True pending = True
lock_path = self._imageBuildLockPath(image) lock_path = self._imageBuildLockPath(image)
lock_stat = self.client.exists(lock_path) lock_stat = self.kazoo_client.exists(lock_path)
if lock_stat and lock_stat.children_count: if lock_stat and lock_stat.children_count:
build_stat = self._latestImageBuildStat(image) build_stat = self._latestImageBuildStat(image)
# If there is a lock, but no build we assume that the build # If there is a lock, but no build we assume that the build
@ -1534,7 +1506,7 @@ class ZooKeeper(object):
:param str image: The image name. :param str image: The image name.
''' '''
path = self._imageBuildRequestPath(image) path = self._imageBuildRequestPath(image)
self.client.ensure_path(path) self.kazoo_client.ensure_path(path)
def removeBuildRequest(self, image): def removeBuildRequest(self, image):
''' '''
@ -1544,7 +1516,7 @@ class ZooKeeper(object):
''' '''
path = self._imageBuildRequestPath(image) path = self._imageBuildRequestPath(image)
try: try:
self.client.delete(path) self.kazoo_client.delete(path)
except kze.NoNodeError: except kze.NoNodeError:
pass pass
@ -1571,7 +1543,7 @@ class ZooKeeper(object):
try: try:
# NOTE: Need to do recursively to remove lock znodes # NOTE: Need to do recursively to remove lock znodes
self.client.delete(path, recursive=True) self.kazoo_client.delete(path, recursive=True)
except kze.NoNodeError: except kze.NoNodeError:
pass pass
@ -1590,7 +1562,7 @@ class ZooKeeper(object):
path = path + "/%s" % upload_number path = path + "/%s" % upload_number
try: try:
# NOTE: Need to do recursively to remove lock znodes # NOTE: Need to do recursively to remove lock znodes
self.client.delete(path, recursive=True) self.kazoo_client.delete(path, recursive=True)
except kze.NoNodeError: except kze.NoNodeError:
pass pass
@ -1610,7 +1582,7 @@ class ZooKeeper(object):
:returns: A list of request nodes. :returns: A list of request nodes.
''' '''
try: try:
requests = self.client.get_children(self.REQUEST_ROOT) requests = self.kazoo_client.get_children(self.REQUEST_ROOT)
except kze.NoNodeError: except kze.NoNodeError:
return [] return []
@ -1621,7 +1593,7 @@ class ZooKeeper(object):
Get the current list of all node request lock ids. Get the current list of all node request lock ids.
''' '''
try: try:
lock_ids = self.client.get_children(self.REQUEST_LOCK_ROOT) lock_ids = self.kazoo_client.get_children(self.REQUEST_LOCK_ROOT)
except kze.NoNodeError: except kze.NoNodeError:
return [] return []
return lock_ids return lock_ids
@ -1640,7 +1612,7 @@ class ZooKeeper(object):
''' '''
path = self._requestLockPath(lock_id) path = self._requestLockPath(lock_id)
try: try:
data, stat = self.client.get(path) data, stat = self.kazoo_client.get(path)
except kze.NoNodeError: except kze.NoNodeError:
return None return None
d = NodeRequestLockStats(lock_id) d = NodeRequestLockStats(lock_id)
@ -1655,7 +1627,7 @@ class ZooKeeper(object):
''' '''
path = self._requestLockPath(lock_id) path = self._requestLockPath(lock_id)
try: try:
self.client.delete(path, recursive=True) self.kazoo_client.delete(path, recursive=True)
except kze.NoNodeError: except kze.NoNodeError:
pass pass
@ -1680,7 +1652,7 @@ class ZooKeeper(object):
# call. # call.
try: try:
path = self._requestPath(request) path = self._requestPath(request)
data, stat = self.client.get(path) data, stat = self.kazoo_client.get(path)
except kze.NoNodeError: except kze.NoNodeError:
return None return None
@ -1696,7 +1668,7 @@ class ZooKeeper(object):
''' '''
path = self._requestPath(request.id) path = self._requestPath(request.id)
data, stat = self.client.get(path) data, stat = self.kazoo_client.get(path)
if data: if data:
d = self._bytesToDict(data) d = self._bytesToDict(data)
@ -1717,7 +1689,7 @@ class ZooKeeper(object):
if not request.event_id: if not request.event_id:
request.event_id = uuid.uuid4().hex request.event_id = uuid.uuid4().hex
path = "%s/%s-" % (self.REQUEST_ROOT, priority) path = "%s/%s-" % (self.REQUEST_ROOT, priority)
path = self.client.create( path = self.kazoo_client.create(
path, path,
value=request.serialize(), value=request.serialize(),
ephemeral=True, ephemeral=True,
@ -1732,7 +1704,7 @@ class ZooKeeper(object):
"Attempt to update non-existing request %s" % request) "Attempt to update non-existing request %s" % request)
path = self._requestPath(request.id) path = self._requestPath(request.id)
self.client.set(path, request.serialize()) self.kazoo_client.set(path, request.serialize())
def deleteNodeRequest(self, request): def deleteNodeRequest(self, request):
''' '''
@ -1745,7 +1717,7 @@ class ZooKeeper(object):
path = self._requestPath(request.id) path = self._requestPath(request.id)
try: try:
self.client.delete(path) self.kazoo_client.delete(path)
except kze.NoNodeError: except kze.NoNodeError:
pass pass
@ -1772,7 +1744,7 @@ class ZooKeeper(object):
node_request_id=request.id) node_request_id=request.id)
path = self._requestLockPath(request.id) path = self._requestLockPath(request.id)
try: try:
lock = Lock(self.client, path) lock = Lock(self.kazoo_client, path)
have_lock = lock.acquire(blocking, timeout) have_lock = lock.acquire(blocking, timeout)
except kze.LockTimeout: except kze.LockTimeout:
raise npe.TimeoutException( raise npe.TimeoutException(
@ -1833,7 +1805,7 @@ class ZooKeeper(object):
''' '''
path = self._nodeLockPath(node.id) path = self._nodeLockPath(node.id)
try: try:
lock = Lock(self.client, path, identifier) lock = Lock(self.kazoo_client, path, identifier)
have_lock = lock.acquire(blocking, timeout, ephemeral) have_lock = lock.acquire(blocking, timeout, ephemeral)
except kze.LockTimeout: except kze.LockTimeout:
raise npe.TimeoutException( raise npe.TimeoutException(
@ -1875,7 +1847,7 @@ class ZooKeeper(object):
''' '''
path = self._nodeLockPath(node.id) path = self._nodeLockPath(node.id)
try: try:
self.client.delete(path, recursive=True) self.kazoo_client.delete(path, recursive=True)
except kze.NoNodeError: except kze.NoNodeError:
pass pass
@ -1884,7 +1856,7 @@ class ZooKeeper(object):
Return the contenders for a node lock. Return the contenders for a node lock.
''' '''
path = self._nodeLockPath(node.id) path = self._nodeLockPath(node.id)
lock = Lock(self.client, path) lock = Lock(self.kazoo_client, path)
return lock.contenders() return lock.contenders()
def getNodes(self): def getNodes(self):
@ -1894,7 +1866,7 @@ class ZooKeeper(object):
:returns: A list of nodes. :returns: A list of nodes.
''' '''
try: try:
return self.client.get_children(self.NODE_ROOT) return self.kazoo_client.get_children(self.NODE_ROOT)
except kze.NoNodeError: except kze.NoNodeError:
return [] return []
@ -1919,7 +1891,7 @@ class ZooKeeper(object):
# call. # call.
try: try:
path = self._nodePath(node) path = self._nodePath(node)
data, stat = self.client.get(path) data, stat = self.kazoo_client.get(path)
except kze.NoNodeError: except kze.NoNodeError:
return None return None
@ -1939,7 +1911,7 @@ class ZooKeeper(object):
''' '''
path = self._nodePath(node.id) path = self._nodePath(node.id)
data, stat = self.client.get(path) data, stat = self.kazoo_client.get(path)
if data: if data:
d = self._bytesToDict(data) d = self._bytesToDict(data)
@ -1971,7 +1943,7 @@ class ZooKeeper(object):
else: else:
node.created_time = time.time() node.created_time = time.time()
path = self.client.create( path = self.kazoo_client.create(
node_path, node_path,
value=node.serialize(), value=node.serialize(),
sequence=True, sequence=True,
@ -1979,7 +1951,7 @@ class ZooKeeper(object):
node.id = path.split("/")[-1] node.id = path.split("/")[-1]
else: else:
path = self._nodePath(node.id) path = self._nodePath(node.id)
self.client.set(path, node.serialize()) self.kazoo_client.set(path, node.serialize())
def watchNode(self, node, callback): def watchNode(self, node, callback):
'''Watch an existing node for changes. '''Watch an existing node for changes.
@ -2000,7 +1972,7 @@ class ZooKeeper(object):
return callback(node, deleted) return callback(node, deleted)
path = self._nodePath(node.id) path = self._nodePath(node.id)
self.client.DataWatch(path, _callback_wrapper) self.kazoo_client.DataWatch(path, _callback_wrapper)
def deleteRawNode(self, node_id): def deleteRawNode(self, node_id):
''' '''
@ -2012,7 +1984,7 @@ class ZooKeeper(object):
''' '''
path = self._nodePath(node_id) path = self._nodePath(node_id)
try: try:
self.client.delete(path, recursive=True) self.kazoo_client.delete(path, recursive=True)
except kze.NoNodeError: except kze.NoNodeError:
pass pass
@ -2031,7 +2003,7 @@ class ZooKeeper(object):
# anything so that we can detect a race condition where the # anything so that we can detect a race condition where the
# lock is removed before the node deletion occurs. # lock is removed before the node deletion occurs.
node.state = DELETED node.state = DELETED
self.client.set(path, node.serialize()) self.kazoo_client.set(path, node.serialize())
self.deleteRawNode(node.id) self.deleteRawNode(node.id)
def getReadyNodesOfTypes(self, labels, cached=True): def getReadyNodesOfTypes(self, labels, cached=True):
@ -2248,7 +2220,7 @@ class ZooKeeper(object):
path = self._imageProviderPath(image, build) path = self._imageProviderPath(image, build)
path = "%s/%s" % (path, provider_name) path = "%s/%s" % (path, provider_name)
try: try:
self.client.delete(path, recursive=True) self.kazoo_client.delete(path, recursive=True)
except kze.NoNodeError: except kze.NoNodeError:
pass pass
@ -2389,7 +2361,7 @@ class ZooKeeper(object):
def getStatsElection(self, identifier): def getStatsElection(self, identifier):
path = self._electionPath('stats') path = self._electionPath('stats')
return Election(self.client, path, identifier) return Election(self.kazoo_client, path, identifier)
def exportImageData(self): def exportImageData(self):
''' '''
@ -2404,7 +2376,7 @@ class ZooKeeper(object):
for build_no in self.getBuildNumbers(image_name): for build_no in self.getBuildNumbers(image_name):
build_path = self._imageBuildsPath(image_name) + "/" + build_no build_path = self._imageBuildsPath(image_name) + "/" + build_no
try: try:
build_data, stat = self.client.get(build_path) build_data, stat = self.kazoo_client.get(build_path)
except kze.NoNodeError: except kze.NoNodeError:
continue continue
ret[build_path] = build_data.decode('utf8') ret[build_path] = build_data.decode('utf8')
@ -2416,7 +2388,8 @@ class ZooKeeper(object):
image_name, build_no, provider_name) + "/" image_name, build_no, provider_name) + "/"
upload_path += upload_no upload_path += upload_no
try: try:
upload_data, stat = self.client.get(upload_path) upload_data, stat = self.kazoo_client.get(
upload_path)
except kze.NoNodeError: except kze.NoNodeError:
continue continue
ret[upload_path] = upload_data.decode('utf8') ret[upload_path] = upload_data.decode('utf8')
@ -2466,19 +2439,19 @@ class ZooKeeper(object):
highest_num[key] = max(highest_num.get(key, num), num) highest_num[key] = max(highest_num.get(key, num), num)
for path, num in highest_num.items(): for path, num in highest_num.items():
for x in range(num): for x in range(num):
node = self.client.create( node = self.kazoo_client.create(
path + '/', makepath=True, sequence=True) path + '/', makepath=True, sequence=True)
# If this node isn't in our import data, go ahead and # If this node isn't in our import data, go ahead and
# delete it. # delete it.
if node not in import_data: if node not in import_data:
self.client.delete(node) self.kazoo_client.delete(node)
for path, data in import_data.items(): for path, data in import_data.items():
# We may have already created a node above; in that # We may have already created a node above; in that
# case, just set the data on it. # case, just set the data on it.
if self.client.exists(path): if self.kazoo_client.exists(path):
self.client.set(path, self.kazoo_client.set(path,
value=data.encode('utf8')) value=data.encode('utf8'))
else: else:
self.client.create(path, self.kazoo_client.create(path,
value=data.encode('utf8'), value=data.encode('utf8'),
makepath=True) makepath=True)

4
tools/print-zk.py
View File

@ -47,12 +47,12 @@ def join(a, b):
return a+'/'+b return a+'/'+b
def print_tree(node): def print_tree(node):
data, stat = zk.client.get(node) data, stat = zk.kazoo_client.get(node)
print("Node: %s %s" % (node, stat)) print("Node: %s %s" % (node, stat))
if data: if data:
print(data) print(data)
for child in zk.client.get_children(node): for child in zk.kazoo_client.get_children(node):
print() print()
print_tree(join(node, child)) print_tree(join(node, child))