9f460914ed
Summary: GraphDbAccessor is now constructed only through GraphDb. This allows the concrete GraphDb to instantiate a concrete GraphDbAccessor. This allows us to use virtual calls, so that the implementation may be kept separate. The major downside of doing things this way is heap allocation of GraphDbAccessor. In case it turns out to be a real performance issues, another solution with pointer to static implementation may be used. InsertVertexIntoRemote is now a non-member function, which reduces coupling. It made no sense for it to be member function because it used only the public parts of GraphDbAccessor. Reviewers: msantl, mtomic, mferencevic Reviewed By: msantl Subscribers: pullbot Differential Revision: https://phabricator.memgraph.io/D1504
81 lines
2.5 KiB
C++
81 lines
2.5 KiB
C++
#include <benchmark/benchmark.h>
|
|
#include <benchmark/benchmark_api.h>
|
|
#include <glog/logging.h>
|
|
|
|
#include "communication/result_stream_faker.hpp"
|
|
#include "database/graph_db.hpp"
|
|
#include "database/graph_db_accessor.hpp"
|
|
#include "query/interpreter.hpp"
|
|
#include "query/typed_value.hpp"
|
|
|
|
class ExpansionBenchFixture : public benchmark::Fixture {
|
|
protected:
|
|
// GraphDb shouldn't be global constructed/destructed. See
|
|
// documentation in database/graph_db.hpp for details.
|
|
std::experimental::optional<database::SingleNode> db_;
|
|
std::experimental::optional<query::Interpreter> interpreter_;
|
|
|
|
void SetUp(const benchmark::State &state) override {
|
|
db_.emplace();
|
|
interpreter_.emplace(db_.value());
|
|
auto dba = db_->Access();
|
|
for (int i = 0; i < state.range(0); i++) dba->InsertVertex();
|
|
|
|
// the fixed part is one vertex expanding to 1000 others
|
|
auto start = dba->InsertVertex();
|
|
start.add_label(dba->Label("Starting"));
|
|
auto edge_type = dba->EdgeType("edge_type");
|
|
for (int i = 0; i < 1000; i++) {
|
|
auto dest = dba->InsertVertex();
|
|
dba->InsertEdge(start, dest, edge_type);
|
|
}
|
|
dba->Commit();
|
|
}
|
|
|
|
void TearDown(const benchmark::State &) override {
|
|
auto dba = db_->Access();
|
|
for (auto vertex : dba->Vertices(false)) dba->DetachRemoveVertex(vertex);
|
|
dba->Commit();
|
|
interpreter_ = std::experimental::nullopt;
|
|
db_ = std::experimental::nullopt;
|
|
}
|
|
|
|
auto &interpreter() { return *interpreter_; }
|
|
};
|
|
|
|
BENCHMARK_DEFINE_F(ExpansionBenchFixture, Match)(benchmark::State &state) {
|
|
auto query = "MATCH (s:Starting) return s";
|
|
auto dba = db_->Access();
|
|
while (state.KeepRunning()) {
|
|
ResultStreamFaker<query::TypedValue> results;
|
|
interpreter()(query, *dba, {}, false).PullAll(results);
|
|
}
|
|
}
|
|
|
|
BENCHMARK_REGISTER_F(ExpansionBenchFixture, Match)
|
|
->RangeMultiplier(1024)
|
|
->Range(1, 1 << 20)
|
|
->Unit(benchmark::kMillisecond);
|
|
|
|
BENCHMARK_DEFINE_F(ExpansionBenchFixture, Expand)(benchmark::State &state) {
|
|
auto query = "MATCH (s:Starting) WITH s MATCH (s)--(d) RETURN count(d)";
|
|
auto dba = db_->Access();
|
|
while (state.KeepRunning()) {
|
|
ResultStreamFaker<query::TypedValue> results;
|
|
interpreter()(query, *dba, {}, false).PullAll(results);
|
|
}
|
|
}
|
|
|
|
BENCHMARK_REGISTER_F(ExpansionBenchFixture, Expand)
|
|
->RangeMultiplier(1024)
|
|
->Range(1, 1 << 20)
|
|
->Unit(benchmark::kMillisecond);
|
|
|
|
int main(int argc, char **argv) {
|
|
google::InitGoogleLogging(argv[0]);
|
|
|
|
::benchmark::Initialize(&argc, argv);
|
|
::benchmark::RunSpecifiedBenchmarks();
|
|
return 0;
|
|
}
|