57b84f2da3
Summary: In order to add kafka benchmark, `memgraph_bolt.cpp` has been split. Now we have `memgraph_init.cpp/hpp` files with common memgraph startup code. Kafka benchmark implements a new `main` function that doesn't start a bolt server, it just creates and starts a stream. Then it waits for the stream to start consuming and measures the time it took to import the given number of entries. This benchmark is in a new folder, `feature_benchmark`, and so should any new bechmark that measures performance of memgraphs features. Reviewers: mferencevic, teon.banek, ipaljak, vkasljevic Reviewed By: mferencevic, teon.banek Subscribers: pullbot Differential Revision: https://phabricator.memgraph.io/D1552
16 lines
552 B
Python
16 lines
552 B
Python
index_done = False
|
|
|
|
def stream(batch):
|
|
global index_done
|
|
ret = []
|
|
if not index_done:
|
|
ret.append(("CREATE INDEX ON :node(num)", {}))
|
|
index_done = True
|
|
for item in batch:
|
|
message = item.decode("utf-8").strip().split()
|
|
if len(message) == 1:
|
|
ret.append(("CREATE (:node {num: $num})", {"num": message[0]}))
|
|
elif len(message) == 2:
|
|
ret.append(("MATCH (n:node {num: $num1}), (m:node {num: $num2}) CREATE (n)-[:et]->(m)", {"num1": message[0], "num2": message[1]}))
|
|
return ret
|