mirror of
https://git.yoctoproject.org/poky
synced 2026-06-01 00:59:48 +00:00
bitbake: hashserv: client: Add batch stream API
Changes the stream mode to do "batch" processing. This means that the sending and reciving of messages is done simultaneously so that messages can be sent as fast as possible without having to wait for each reply. This allows multiple messages to be in flight at once, reducing the effect of the round trip latency from the server. (Bitbake rev: e768d0f17bdb97f6ff013ec3a41f182fecd47a55) Signed-off-by: Joshua Watt <JPEWhacker@gmail.com> Signed-off-by: Richard Purdie <richard.purdie@linuxfoundation.org>
This commit is contained in:
committed by
Richard Purdie
parent
8fc967d0c7
commit
29c2cd4d54
@@ -594,6 +594,43 @@ class HashEquivalenceCommonTests(object):
|
||||
7: None,
|
||||
})
|
||||
|
||||
def test_get_unihash_batch(self):
|
||||
TEST_INPUT = (
|
||||
# taskhash outhash unihash
|
||||
('8aa96fcffb5831b3c2c0cb75f0431e3f8b20554a', 'afe240a439959ce86f5e322f8c208e1fedefea9e813f2140c81af866cc9edf7e','218e57509998197d570e2c98512d0105985dffc9'),
|
||||
# Duplicated taskhash with multiple output hashes and unihashes.
|
||||
('8aa96fcffb5831b3c2c0cb75f0431e3f8b20554a', '0904a7fe3dc712d9fd8a74a616ddca2a825a8ee97adf0bd3fc86082c7639914d', 'ae9a7d252735f0dafcdb10e2e02561ca3a47314c'),
|
||||
# Equivalent hash
|
||||
("044c2ec8aaf480685a00ff6ff49e6162e6ad34e1", '0904a7fe3dc712d9fd8a74a616ddca2a825a8ee97adf0bd3fc86082c7639914d', "def64766090d28f627e816454ed46894bb3aab36"),
|
||||
("e3da00593d6a7fb435c7e2114976c59c5fd6d561", "1cf8713e645f491eb9c959d20b5cae1c47133a292626dda9b10709857cbe688a", "3b5d3d83f07f259e9086fcb422c855286e18a57d"),
|
||||
('35788efcb8dfb0a02659d81cf2bfd695fb30faf9', '2765d4a5884be49b28601445c2760c5f21e7e5c0ee2b7e3fce98fd7e5970796f', 'f46d3fbb439bd9b921095da657a4de906510d2cd'),
|
||||
('35788efcb8dfb0a02659d81cf2bfd695fb30fafa', '2765d4a5884be49b28601445c2760c5f21e7e5c0ee2b7e3fce98fd7e5970796f', 'f46d3fbb439bd9b921095da657a4de906510d2ce'),
|
||||
('9d81d76242cc7cfaf7bf74b94b9cd2e29324ed74', '8470d56547eea6236d7c81a644ce74670ca0bbda998e13c629ef6bb3f0d60b69', '05d2a63c81e32f0a36542ca677e8ad852365c538'),
|
||||
)
|
||||
EXTRA_QUERIES = (
|
||||
"6b6be7a84ab179b4240c4302518dc3f6",
|
||||
)
|
||||
|
||||
for taskhash, outhash, unihash in TEST_INPUT:
|
||||
self.client.report_unihash(taskhash, self.METHOD, outhash, unihash)
|
||||
|
||||
|
||||
result = self.client.get_unihash_batch(
|
||||
[(self.METHOD, data[0]) for data in TEST_INPUT] +
|
||||
[(self.METHOD, e) for e in EXTRA_QUERIES]
|
||||
)
|
||||
|
||||
self.assertListEqual(result, [
|
||||
"218e57509998197d570e2c98512d0105985dffc9",
|
||||
"218e57509998197d570e2c98512d0105985dffc9",
|
||||
"218e57509998197d570e2c98512d0105985dffc9",
|
||||
"3b5d3d83f07f259e9086fcb422c855286e18a57d",
|
||||
"f46d3fbb439bd9b921095da657a4de906510d2cd",
|
||||
"f46d3fbb439bd9b921095da657a4de906510d2cd",
|
||||
"05d2a63c81e32f0a36542ca677e8ad852365c538",
|
||||
None,
|
||||
])
|
||||
|
||||
def test_client_pool_unihash_exists(self):
|
||||
TEST_INPUT = (
|
||||
# taskhash outhash unihash
|
||||
@@ -636,6 +673,44 @@ class HashEquivalenceCommonTests(object):
|
||||
result = client_pool.unihashes_exist(query)
|
||||
self.assertDictEqual(result, expected)
|
||||
|
||||
def test_unihash_exists_batch(self):
|
||||
TEST_INPUT = (
|
||||
# taskhash outhash unihash
|
||||
('8aa96fcffb5831b3c2c0cb75f0431e3f8b20554a', 'afe240a439959ce86f5e322f8c208e1fedefea9e813f2140c81af866cc9edf7e','218e57509998197d570e2c98512d0105985dffc9'),
|
||||
# Duplicated taskhash with multiple output hashes and unihashes.
|
||||
('8aa96fcffb5831b3c2c0cb75f0431e3f8b20554a', '0904a7fe3dc712d9fd8a74a616ddca2a825a8ee97adf0bd3fc86082c7639914d', 'ae9a7d252735f0dafcdb10e2e02561ca3a47314c'),
|
||||
# Equivalent hash
|
||||
("044c2ec8aaf480685a00ff6ff49e6162e6ad34e1", '0904a7fe3dc712d9fd8a74a616ddca2a825a8ee97adf0bd3fc86082c7639914d', "def64766090d28f627e816454ed46894bb3aab36"),
|
||||
("e3da00593d6a7fb435c7e2114976c59c5fd6d561", "1cf8713e645f491eb9c959d20b5cae1c47133a292626dda9b10709857cbe688a", "3b5d3d83f07f259e9086fcb422c855286e18a57d"),
|
||||
('35788efcb8dfb0a02659d81cf2bfd695fb30faf9', '2765d4a5884be49b28601445c2760c5f21e7e5c0ee2b7e3fce98fd7e5970796f', 'f46d3fbb439bd9b921095da657a4de906510d2cd'),
|
||||
('35788efcb8dfb0a02659d81cf2bfd695fb30fafa', '2765d4a5884be49b28601445c2760c5f21e7e5c0ee2b7e3fce98fd7e5970796f', 'f46d3fbb439bd9b921095da657a4de906510d2ce'),
|
||||
('9d81d76242cc7cfaf7bf74b94b9cd2e29324ed74', '8470d56547eea6236d7c81a644ce74670ca0bbda998e13c629ef6bb3f0d60b69', '05d2a63c81e32f0a36542ca677e8ad852365c538'),
|
||||
)
|
||||
EXTRA_QUERIES = (
|
||||
"6b6be7a84ab179b4240c4302518dc3f6",
|
||||
)
|
||||
|
||||
result_unihashes = set()
|
||||
|
||||
|
||||
for taskhash, outhash, unihash in TEST_INPUT:
|
||||
result = self.client.report_unihash(taskhash, self.METHOD, outhash, unihash)
|
||||
result_unihashes.add(result["unihash"])
|
||||
|
||||
query = []
|
||||
expected = []
|
||||
|
||||
for _, _, unihash in TEST_INPUT:
|
||||
query.append(unihash)
|
||||
expected.append(unihash in result_unihashes)
|
||||
|
||||
|
||||
for unihash in EXTRA_QUERIES:
|
||||
query.append(unihash)
|
||||
expected.append(False)
|
||||
|
||||
result = self.client.unihash_exists_batch(query)
|
||||
self.assertListEqual(result, expected)
|
||||
|
||||
def test_auth_read_perms(self):
|
||||
admin_client = self.start_auth_server()
|
||||
|
||||
Reference in New Issue
Block a user