Repository navigation
Expand file tree
/
Copy pathbenchmark.py
More file actions
120 lines (90 loc) · 3.12 KB
/
Copy pathbenchmark.py
File metadata and controls
120 lines (90 loc) · 3.12 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
import objectbox
import time
from objectbox.store import Store
from tests.model import TestEntity
from tests.common import create_test_store
class ObjectBoxPerf:
"""
Performance executable
"""
def __init__(self):
self.store = create_test_store()
self.box = self.store.box(TestEntity)
def remove_all(self):
self.box.remove_all()
def put_many(self, items):
self.box.put(items)
def read_all(self):
return self.box.get_all()
class Timer:
"""
Context manager to time a block of code.
Appends the resulting runtime in milliseconds to a list of floats.
"""
def __init__(self, out_list: [float]):
self.start = time.time_ns()
self.list = out_list
def __enter__(self):
return self
def __exit__(self, exc_type, exc_val, exc_tb):
self.list.append((time.time_ns() - self.start) / 1000 / 1000)
class PerfExecutor:
"""
Performance executable
"""
def __init__(self, executable):
self.perf = executable
def run(self, count=1000, runs=10):
inserts = self.__prepare_data(count)
# function => vector of runtimes in milliseconds
from collections import defaultdict
times = defaultdict(list)
progress_text = "Measuring performance"
for i in range(runs):
self.__progress_bar(progress_text, i, runs)
with Timer(times["insert-many"]):
self.perf.put_many(inserts)
with Timer(times["read-all"]):
items = self.perf.read_all()
with Timer(times["update-many"]):
self.perf.put_many(items)
with Timer(times["remove-all"]):
self.perf.remove_all()
self.__progress_bar(progress_text, runs, runs)
# print the results
print()
print('=' * 80)
print("runs\t%d\t\tcount\t%d\t\tunit\tms" % (runs, count))
print("Function\tMedian\tMean\tStdDev")
import statistics
for key in times.keys():
print("%s\t%d\t%d\t%d" % (
key,
statistics.median(times[key]),
statistics.mean(times[key]),
statistics.stdev(times[key])
))
@staticmethod
def __prepare_data(count: int) -> [TestEntity]:
result = []
for i in range(count):
object = TestEntity()
object.str = "Entity no. %d" % i
object.float = i * 1.1
object.int = i
result.append(object)
return result
@staticmethod
def __progress_bar(text, value, endvalue, bar_length=20):
import sys
percent = float(value) / endvalue
arrow = '-' * int(round(percent * bar_length) - 1) + '>'
spaces = ' ' * (bar_length - len(arrow))
sys.stdout.write("\r{0}: [{1}] {2}%".format(text, arrow + spaces, int(round(percent * 100))))
sys.stdout.flush()
if __name__ == "__main__":
Store.remove_db_files("testdata")
obPerf = ObjectBoxPerf()
executor = PerfExecutor(obPerf)
executor.run(count=10000, runs=20)
Store.remove_db_files("testdata")