40 lines
1.4 KiB
C++
40 lines
1.4 KiB
C++
|
#pragma once
|
||
|
|
|
||
|
|
#include "query_engine/code_generator/handlers/includes.hpp"
|
||
|
|
|
||
|
|
auto create_query_action =
|
||
|
|
[](CypherStateData &cypher_data,
|
||
|
|
const QueryActionData &action_data) -> std::string {
|
||
|
|
|
||
|
|
std::string code = "";
|
||
|
|
|
||
|
|
for (auto const &kv : action_data.actions) {
|
||
|
|
|
||
|
|
if (kv.second == ClauseAction::CreateNode) {
|
||
|
|
// 1. create node 2. update labels 3. update properties
|
||
|
|
auto &name = kv.first;
|
||
|
|
code += LINE(fmt::format(code::create_vertex, name));
|
||
|
|
auto entity_data = action_data.get_entity_property(name);
|
||
|
|
for (auto &property : entity_data.properties) {
|
||
|
|
auto index = action_data.parameter_index.at(
|
||
|
|
ParameterIndexKey(name, property));
|
||
|
|
code += LINE(fmt::format(code::set_vertex_property, name,
|
||
|
|
property, index));
|
||
|
|
}
|
||
|
|
for (auto &label : entity_data.tags) {
|
||
|
|
code += LINE(fmt::format(code::create_label, label));
|
||
|
|
code += LINE(fmt::format(code::add_label, name, label));
|
||
|
|
}
|
||
|
|
cypher_data.node_created(name);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (kv.second == ClauseAction::CreateRelationship) {
|
||
|
|
auto name = kv.first;
|
||
|
|
code += LINE(fmt::format(code::create_edge, name));
|
||
|
|
cypher_data.relationship_created(name);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return code;
|
||
|
|
};
|