mirrors/poky
1
0
Fork 0
mirror of https://git.yoctoproject.org/poky synced 2026-06-04 02:00:04 +00:00
Code Issues Projects Releases Wiki Activity

bitbake: asyncrpc: Add context manager API

Browse Source 
Adds context manager API for the asyncrcp client class which allow
writing code that will automatically close the connection like so:

    with hashserv.create_client(address) as client:
       ...

Rework the bitbake-hashclient tool and PR server to use this new API to
fix warnings about unclosed event loops when exiting

(Bitbake rev: 52226a7244968c8dad6f4ee9ccec57ac1979217a)

Signed-off-by: Joshua Watt <JPEWhacker@gmail.com>
Signed-off-by: Richard Purdie <richard.purdie@linuxfoundation.org>
(cherry picked from commit d01d684a0f6398270fe35ed59b7d28f3fd9b7e41)
Signed-off-by: Steve Sakoman <steve@sakoman.com>
This commit is contained in:
Joshua Watt
2023-11-03 08:26:21 -06:00
committed by Steve Sakoman
parent f768ffb891
commit afe1706a77
3 changed files with 33 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files
bitbake/bin/bitbake-hashclient
+17 -19
View File
@@ -56,25 +56,24 @@ def main():
nonlocal missed_hashes
nonlocal max_time
client = hashserv.create_client(args.address)
with hashserv.create_client(args.address) as client:
for i in range(args.requests):
taskhash = hashlib.sha256()
taskhash.update(args.taskhash_seed.encode('utf-8'))
taskhash.update(str(i).encode('utf-8'))
for i in range(args.requests):
taskhash = hashlib.sha256()
taskhash.update(args.taskhash_seed.encode('utf-8'))
taskhash.update(str(i).encode('utf-8'))
start_time = time.perf_counter()
l = client.get_unihash(METHOD, taskhash.hexdigest())
elapsed = time.perf_counter() - start_time
start_time = time.perf_counter()
l = client.get_unihash(METHOD, taskhash.hexdigest())
elapsed = time.perf_counter() - start_time
with lock:
if l:
found_hashes += 1
else:
missed_hashes += 1
with lock:
if l:
found_hashes += 1
else:
missed_hashes += 1
max_time = max(elapsed, max_time)
pbar.update()
max_time = max(elapsed, max_time)
pbar.update()
max_time = 0
found_hashes = 0
@@ -174,9 +173,8 @@ def main():
func = getattr(args, 'func', None)
if func:
client = hashserv.create_client(args.address)
return func(args, client)
with hashserv.create_client(args.address) as client:
return func(args, client)
return 0