added sample row keys

googleapis · Oct 6, 2023 · f053125 · f053125
1 parent 319aa63
commit f053125
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/test_proxy/handlers/client_handler_legacy.py b/test_proxy/handlers/client_handler_legacy.py
@@ -204,4 +204,8 @@ async def ReadModifyWriteRow(self, request, **kwargs):
 
     @client_handler.error_safe
     async def SampleRowKeys(self, request, **kwargs):
-        raise NotImplementedError()
+        table_id = request["table_name"].split("/")[-1]
+        instance = self.client.instance(self.instance_id)
+        table = instance.table(table_id)
+        response = list(table.sample_row_keys())
+        return response