zuul/nodepool
Code Issues Proposed changes

Merge "Adapt load-test to new zk structure"

Browse Source
This commit is contained in:
Zuul 2023-01-16 07:20:36 +00:00 committed by Gerrit Code Review
commit 13792b3b3b
1 changed files with 11 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
tools/load-test.py
View File

@ -2,8 +2,10 @@
import logging import logging
import time import time
from nodepool import launcher, zk from nodepool import launcher
from nodepool.cmd import NodepoolApp from nodepool.cmd import NodepoolApp
from nodepool.zk import ZooKeeperClient
import nodepool.zk.zookeeper as zk
class LoadTest(NodepoolApp): class LoadTest(NodepoolApp):
@ -47,12 +49,14 @@ class LoadTest(NodepoolApp):
self.pool = launcher.NodePool(self.args.secure, self.args.config) self.pool = launcher.NodePool(self.args.secure, self.args.config)
config = self.pool.loadConfig() config = self.pool.loadConfig()
self.zk = zk.ZooKeeper(enable_cache=False) self.zk_client = ZooKeeperClient(
self.zk.connect( config.zookeeper_servers,
list(config.zookeeper_servers.values()),
tls_cert=config.zookeeper_tls_cert, tls_cert=config.zookeeper_tls_cert,
tls_key=config.zookeeper_tls_key, tls_key=config.zookeeper_tls_key,
tls_ca=config.zookeeper_tls_ca) tls_ca=config.zookeeper_tls_ca
)
self.zk_client.connect()
self.zk = zk.ZooKeeper(self.zk_client, enable_cache=False)
label = self.args.label label = self.args.label
max_queue = int(self.args.max_queue) max_queue = int(self.args.max_queue)
@ -100,13 +104,12 @@ class LoadTest(NodepoolApp):
for request in failed_requests: for request in failed_requests:
self.zk.deleteNodeRequest(request) self.zk.deleteNodeRequest(request)
self.log.info('Handling %s fulfilled requests', len(fulfilled_requests)) self.log.info(
'Handling %s fulfilled requests', len(fulfilled_requests))
for request in fulfilled_requests: for request in fulfilled_requests:
# TODO: handle nodes # TODO: handle nodes
self.zk.deleteNodeRequest(request) self.zk.deleteNodeRequest(request)
if __name__ == "__main__": if __name__ == "__main__":
LoadTest.main() LoadTest.main()