kvstore/perftest.py
2025-10-11 14:48:24 -04:00

30 lines
724 B
Python
Executable File

#!/usr/bin/env python
import plyvel
import read_tree
from tqdm import tqdm
with plyvel.DB('../node/nodedir/data/application.db') as db:
height = read_tree.max_height(db)
total = read_tree.count(db, 's/k:emissions/', height, 'Qss', key = [62])
progress = tqdm(total = total)
it = read_tree.iterate(db, 's/k:emissions/', height, 'Qss', key = [62])
for k, v in it:
progress.update(1)
progress.close()
keys = it.inner.lookups
print(f'Number of items: {total}')
print(f'Lookups needed: {len(keys)}')
with plyvel.DB('../node/nodedir/data/application.db') as db:
progress = tqdm(total = len(keys))
for k in keys:
db.get(k)
progress.update(1)
progress.close()