2215173a58
Summary: RandomGraphGenerator added Reviewers: teon.banek, mislav.bradac, buda Reviewed By: buda Subscribers: pullbot Differential Revision: https://phabricator.memgraph.io/D165
85 lines
2.5 KiB
C++
85 lines
2.5 KiB
C++
#include <iostream>
|
|
|
|
#include "dbms/dbms.hpp"
|
|
#include "query/console.hpp"
|
|
#include "query/interpreter.hpp"
|
|
#include "utils/random_graph_generator.hpp"
|
|
|
|
void random_generate(Dbms &dbms) {
|
|
auto dba = dbms.active();
|
|
utils::RandomGraphGenerator generator(*dba);
|
|
|
|
generator.AddVertices(1000000, {"Person"});
|
|
generator.AddEdges(5000000, "Friend");
|
|
generator.SetVertexProperty<int>(1000000, "age", utils::RandomIntGenerator(3, 60));
|
|
generator.SetVertexProperty<int>(1000000, "height", utils::RandomIntGenerator(120, 200));
|
|
|
|
generator.Commit();
|
|
}
|
|
|
|
void fill_db(Dbms &dbms) {
|
|
auto dba = dbms.active();
|
|
|
|
// labels
|
|
auto company = dba->label("Company");
|
|
auto person = dba->label("Person");
|
|
auto device = dba->label("Device");
|
|
|
|
// props
|
|
auto name = dba->property("name");
|
|
auto age = dba->property("age");
|
|
auto type = dba->property("type");
|
|
|
|
// vertices
|
|
auto memgraph = dba->insert_vertex();
|
|
memgraph.PropsSet(name, "Memgraph");
|
|
memgraph.add_label(company);
|
|
auto teon = dba->insert_vertex();
|
|
teon.PropsSet(name, "Teon");
|
|
teon.PropsSet(age, 26);
|
|
teon.add_label(person);
|
|
auto mislav = dba->insert_vertex();
|
|
mislav.PropsSet(name, "Mislav");
|
|
mislav.PropsSet(age, 22);
|
|
mislav.add_label(person);
|
|
auto florijan = dba->insert_vertex();
|
|
florijan.PropsSet(name, "Florijan");
|
|
florijan.PropsSet(age, 31);
|
|
florijan.add_label(person);
|
|
auto xps_15 = dba->insert_vertex();
|
|
xps_15.PropsSet(type, "PC");
|
|
xps_15.PropsSet(name, "Dell XPS 15");
|
|
xps_15.add_label(device);
|
|
|
|
// edges
|
|
dba->insert_edge(teon, memgraph, dba->edge_type("MEMBER_OF"));
|
|
dba->insert_edge(mislav, memgraph, dba->edge_type("MEMBER_OF"));
|
|
dba->insert_edge(florijan, memgraph, dba->edge_type("MEMBER_OF"));
|
|
|
|
dba->insert_edge(teon, mislav, dba->edge_type("FRIEND_OF"));
|
|
dba->insert_edge(mislav, teon, dba->edge_type("FRIEND_OF"));
|
|
dba->insert_edge(florijan, mislav, dba->edge_type("FRIEND_OF"));
|
|
dba->insert_edge(mislav, florijan, dba->edge_type("FRIEND_OF"));
|
|
dba->insert_edge(florijan, teon, dba->edge_type("FRIEND_OF"));
|
|
dba->insert_edge(teon, florijan, dba->edge_type("FRIEND_OF"));
|
|
|
|
dba->insert_edge(memgraph, xps_15, dba->edge_type("OWNS"));
|
|
|
|
dba->insert_edge(teon, xps_15, dba->edge_type("USES"));
|
|
dba->insert_edge(mislav, xps_15, dba->edge_type("USES"));
|
|
dba->insert_edge(florijan, xps_15, dba->edge_type("USES"));
|
|
|
|
dba->commit();
|
|
}
|
|
|
|
int main(int argc, char *argv[]) {
|
|
REGISTER_ARGS(argc, argv);
|
|
|
|
Dbms dbms;
|
|
std::cout << "Generating graph..." << std::endl;
|
|
// fill_db(dbms);
|
|
random_generate(dbms);
|
|
query::Repl(dbms);
|
|
return 0;
|
|
}
|