diff options
Diffstat (limited to 'benchmarks')
| -rw-r--r-- | benchmarks/tpcc/run.py | 14 |
1 files changed, 10 insertions, 4 deletions
diff --git a/benchmarks/tpcc/run.py b/benchmarks/tpcc/run.py index 4a388bc..377d555 100644 --- a/benchmarks/tpcc/run.py +++ b/benchmarks/tpcc/run.py @@ -242,10 +242,16 @@ def register(pure, aup, work, name, view): assert status == 200, (name, status, body[:200]) -def asked(pure, name): - status, body = pure.call('GET', f'/queries/{name}') - assert status == 200, (name, status, body[:200]) - return [tuple(row.values()) for row in json.loads(body)['items']] +def asked(pure, name, page=200000): + """A query's rows, a page at a time: a whole table of W > 1 answered at once outgrows the server's budget.""" + rows = [] + while True: + status, body = pure.call('GET', f'/queries/{name}?from={len(rows)}&count={page}') + assert status == 200, (name, status, body[:200]) + answer = json.loads(body) + rows += [tuple(row.values()) for row in answer['items']] + if len(rows) >= answer['rows'] or not answer['items']: + return rows def same(a, b): |