Add API to remove nodes from CXXGraph by id

Issue: ZigRazor#419
Saj18 · Apr 29, 2024 · 53b9961 · 53b9961
1 parent 11e80f2
commit 53b9961
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/include/CXXGraph/Node/Node_decl.h b/include/CXXGraph/Node/Node_decl.h
@@ -49,6 +49,7 @@ class Node {
   const std::string &getUserId() const;
   const T &getData() const;
   void setData(T &&new_data);
+
   // operator
   bool operator==(const Node<T> &b) const;
   bool operator<(const Node<T> &b) const;

diff --git a/include/CXXGraph/Node/Node_impl.hpp b/include/CXXGraph/Node/Node_impl.hpp
@@ -70,6 +70,18 @@ void Node<T>::setData(T &&new_data) {
   this->data = std::move(new_data);
 }
 
+ bool deleteNodeById(const std::string &id) {
+    auto it = std::find_if(nodes.begin(), nodes.end(), [&](const auto &pair) {
+      return pair.second.getUserId() == id;
+    });
+
+    if (it != nodes.end()) {
+      nodes.erase(it);
+      return true; // Node deleted successfully
+    }
+    return false; // Node not found
+  }
+
 // The data type T must have an overload of the equality operator
 template <typename T>
 bool Node<T>::operator==(const Node<T> &b) const {