mirrors/scylladb
1
0
Fork 0
mirror of https://github.com/scylladb/scylladb.git synced 2026-06-06 06:53:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

pylib/rest_client.py: add ownership API to ScyllaRESTAPIClient

Browse Source 
This change adds a member function that can be used
to access 'storage_service/ownership' API.

It will be used by tests that need to access this API.

Refs: scylladb#17342
Signed-off-by: Patryk Wrobel <patryk.wrobel@scylladb.com>
This commit is contained in:
Patryk Wrobel
2024-02-23 12:44:04 +01:00
parent dea76c4763
commit a39a5b671e
1 changed files with 11 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
test/pylib/rest_client.py
View File

@@ -162,6 +162,17 @@ class ScyllaRESTAPIClient():
assert(type(data) == list)
return data
async def get_ownership(self, dst_server_ip: IPAddress, keyspace: str = None, table: str = None) -> list:
"""Retrieve the ownership"""
if keyspace is None and table is None:
api_path = f"/storage_service/ownership/"
elif table is None:
api_path = f"/storage_service/ownership/{keyspace}"
else:
api_path = f"/storage_service/ownership/{keyspace}?cf={table}"
data = await self.client.get_json(api_path, dst_server_ip)
return data
async def get_down_endpoints(self, node_ip: IPAddress) -> list[IPAddress]:
"""Retrieve down endpoints from gossiper's point of view """
data = await self.client.get_json("/gossiper/endpoint/down/", node_ip)