def internal_kv_keys(self, prefix: bytes) -> List[bytes]:
req = gcs_service_pb2.InternalKVKeysRequest(prefix=prefix)
reply = self._kv_stub.InternalKVKeys(req)
if reply.status.code == GcsCode.OK:
return list(reply.results)
else:
raise RuntimeError(f"Failed to list prefix {prefix} "
f"due to error {reply.status.message}")
def internal_kv_keys(self, prefix: bytes,
namespace: Optional[bytes]) -> List[bytes]:
logger.debug(f"internal_kv_keys {prefix} {namespace}")
req = gcs_service_pb2.InternalKVKeysRequest(namespace=namespace,
prefix=prefix)
reply = self._kv_stub.InternalKVKeys(req)
if reply.status.code == GcsCode.OK:
return reply.results
else:
raise RuntimeError(f"Failed to list prefix {prefix} "
f"due to error {reply.status.message}")