forked from TuGraph-family/tugraph-db
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
3 changed files
with
139 additions
and
131 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,137 @@ | ||
| #include "tools/lgraph_log.h" | ||
| #include "core/data_type.h" | ||
| namespace lgraph { | ||
|
|
||
| lgraph_api::FieldData JsonToFieldData(const nlohmann::json& j_object) { | ||
| if (j_object.is_number_integer()) { | ||
| return lgraph_api::FieldData(j_object.get<int64_t>()); | ||
| } else if (j_object.is_string()) { | ||
| return lgraph_api::FieldData(j_object.get<std::string>()); | ||
| } else if (j_object.is_number_float()) { | ||
| return lgraph_api::FieldData(j_object.get<float>()); | ||
| } else if (j_object.is_boolean()) { | ||
| return lgraph_api::FieldData(j_object.get<bool>()); | ||
| } | ||
| throw std::runtime_error("unexpected json type, obj: " + j_object.dump()); | ||
| }; | ||
|
|
||
| struct VertexLine { | ||
| size_t vertexLabel; | ||
| size_t primaryFieldId; | ||
| lgraph_api::FieldData primaryFieldValue; | ||
| std::vector<size_t> field_ids; | ||
| std::vector<lgraph_api::FieldData> field_values; | ||
| }; | ||
|
|
||
| bool update_vertex(lgraph_api::GraphDB &db, const std::string &request, std::string &response) { | ||
| using namespace std; | ||
| std::string vertex_type; | ||
| try { | ||
| int add_nodes = 0; | ||
| int update_nodes = 0; | ||
| int filter_nodes = 0; | ||
| nlohmann::json input = nlohmann::json::parse(request); | ||
| vertex_type = input["type"].get<string>(); | ||
| string primary_field = input["primary"].get<string>(); | ||
| auto vertexes_count = input["vertexes"].size(); | ||
| if (vertexes_count == 0) { | ||
| nlohmann::json output; | ||
| output["ok"] = true; | ||
| output["response"]= "No vertexes to update."; | ||
| response = output.dump(); | ||
| return true; | ||
| } | ||
| std::vector<VertexLine> lines; | ||
| { | ||
| auto txn = db.CreateReadTxn(); | ||
| auto vertexLabel = txn.GetVertexLabelId(vertex_type); | ||
| auto vertexPrimary = txn.GetVertexFieldId(vertexLabel, primary_field); | ||
| for (auto& vertex : input["vertexes"]) { | ||
| try { | ||
| auto id = JsonToFieldData(vertex[primary_field]); | ||
| vector<string> field_names; | ||
| vector<lgraph_api::FieldData> field_values; | ||
| for (auto& el : vertex.items()) { | ||
| if (el.key() == primary_field) { | ||
| continue; | ||
| } | ||
| field_names.push_back(el.key()); | ||
| field_values.push_back(JsonToFieldData(el.value())); | ||
| } | ||
| auto field_ids = txn.GetVertexFieldIds(vertexLabel, field_names); | ||
| VertexLine line; | ||
| line.vertexLabel = vertexLabel; | ||
| line.primaryFieldId = vertexPrimary; | ||
| line.primaryFieldValue = id; | ||
| line.field_ids = std::move(field_ids); | ||
| line.field_values = std::move(field_values); | ||
| lines.emplace_back(std::move(line)); | ||
| } catch (const exception &e) { | ||
| continue; | ||
| } | ||
| } | ||
| txn.Abort(); | ||
| } | ||
|
|
||
| if (!lines.empty()) { | ||
| auto txn = db.CreateWriteTxn(); | ||
| for (auto& line : lines) { | ||
| try { | ||
| // try to find the vertex and update the property | ||
| auto vit = txn.GetVertexByUniqueIndex(line.vertexLabel, line.primaryFieldId, line.primaryFieldValue); | ||
| if (!line.field_ids.empty()) { | ||
| bool need_update = false; | ||
| auto fields = vit.GetFields(line.field_ids); | ||
| for (size_t i = 0; i < fields.size(); i++) { | ||
| if (fields[i] != line.field_values[i]) { | ||
| need_update = true; | ||
| break; | ||
| } | ||
| } | ||
| if (need_update) { | ||
| vit.SetFields(line.field_ids, line.field_values); | ||
| update_nodes++; | ||
| } else { | ||
| filter_nodes++; | ||
| } | ||
| } else { | ||
| filter_nodes++; | ||
| } | ||
| } catch (const runtime_error& re) { | ||
| // not found, insert | ||
| line.field_ids.push_back(line.primaryFieldId); | ||
| line.field_values.push_back(line.primaryFieldValue); | ||
| try { | ||
| txn.AddVertex(line.vertexLabel, line.field_ids, line.field_values); | ||
| } catch (const exception& e) { | ||
| LOG_WARN() << "AddVertex, type: " << vertex_type << ", e: " << e.what() << ", re: " << re.what(); | ||
| throw e; | ||
| } | ||
| add_nodes++; | ||
| } catch (const std::exception& e) { | ||
| LOG_WARN() << "GetVertexByUniqueIndex, type: " << vertex_type << ", e: " << e.what(); | ||
| } | ||
| } | ||
| txn.Commit(); | ||
| } | ||
|
|
||
| nlohmann::json output; | ||
| output["ok"] = true; | ||
| output["response"]["type"] = vertex_type; | ||
| output["response"]["add_nodes"] = add_nodes; | ||
| output["response"]["update_nodes"] = update_nodes; | ||
| output["response"]["filter_nodes"] = filter_nodes; | ||
| response = output.dump(); | ||
| return true; | ||
| } catch (const exception &e) { | ||
| auto err = e.what(); | ||
| LOG_WARN() << vertex_type << " update_nodes, exception: " << err; | ||
| nlohmann::json output; | ||
| output["ok"] = false; | ||
| output["response"] = std::string("Error on vertex ") + vertex_type + (" processing: ") + err; | ||
| response = output.dump(); | ||
| return false; | ||
| } | ||
| } | ||
|
|
||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters