From 208ba933778c90971e038bd3699595f3dc28774e Mon Sep 17 00:00:00 2001
From: etkmao <etkmao@tecent.com>
Date: Mon, 4 Sep 2023 10:13:17 +0800
Subject: [PATCH] fix(core): fix performance for node eliminated

---
 dom/include/dom/dom_node.h                     |  6 ++++++
 .../dom/layer_optimized_render_manager.h       |  4 ----
 dom/src/dom/layer_optimized_render_manager.cc  | 18 ++----------------
 3 files changed, 8 insertions(+), 20 deletions(-)

diff --git a/dom/include/dom/dom_node.h b/dom/include/dom/dom_node.h
index e56964c9deb..8b957411a44 100644
--- a/dom/include/dom/dom_node.h
+++ b/dom/include/dom/dom_node.h
@@ -119,6 +119,8 @@ class DomNode : public std::enable_shared_from_this<DomNode> {
   inline void SetLayoutOnly(bool layout_only) { layout_only_ = layout_only; }
   inline bool IsVirtual() { return is_virtual_; }
   inline void SetIsVirtual(bool is_virtual) { is_virtual_ = is_virtual; }
+  inline bool IsEnableEliminated() { return enable_eliminated_; }
+  inline void SetEnableEliminated(bool enable_eliminated) { enable_eliminated_ = enable_eliminated; }
   inline void SetIndex(int32_t index) { index_ = index; }
   inline int32_t GetIndex() const { return index_; }
   inline void SetRootNode(std::weak_ptr<RootNode> root_node) { root_node_ = root_node; }
@@ -230,6 +232,10 @@ class DomNode : public std::enable_shared_from_this<DomNode> {
   bool is_virtual_{};
   bool layout_only_ = false;
 
+  // Node can only be eliminated for the first time,
+  // and if they cannot be eliminated for the first time, they cannot be eliminated at all times.
+  bool enable_eliminated_ = true;
+
   std::weak_ptr<DomNode> parent_;
   std::vector<std::shared_ptr<DomNode>> children_;
 
diff --git a/dom/include/dom/layer_optimized_render_manager.h b/dom/include/dom/layer_optimized_render_manager.h
index a060bad94df..73c76d3d2ba 100644
--- a/dom/include/dom/layer_optimized_render_manager.h
+++ b/dom/include/dom/layer_optimized_render_manager.h
@@ -73,10 +73,6 @@ class LayerOptimizedRenderManager : public RenderManager {
 
   void FindValidChildren(const std::shared_ptr<DomNode>& node,
                          std::vector<std::shared_ptr<DomNode>>& valid_children_nodes);
-
-  // Record nodes that cannot be eliminated. Nodes can only be eliminated for the first time,
-  // and if they cannot be eliminated for the first time, they cannot be eliminated at all times.
-  std::set<uint32_t> not_eliminated_node_ids_;
 };
 
 }  // namespace dom
diff --git a/dom/src/dom/layer_optimized_render_manager.cc b/dom/src/dom/layer_optimized_render_manager.cc
index fa70cf2686f..646b8befcef 100644
--- a/dom/src/dom/layer_optimized_render_manager.cc
+++ b/dom/src/dom/layer_optimized_render_manager.cc
@@ -128,19 +128,6 @@ void LayerOptimizedRenderManager::DeleteRenderNode(std::weak_ptr<RootNode> root_
   FOOTSTONE_DLOG(INFO) << "[Hippy Statistic] delete node size before optimize = " << nodes.size()
                        << ", delete node size after optimize  = " << nodes_to_delete.size();
   if (!nodes_to_delete.empty()) {
-    for (auto& node : nodes_to_delete) {
-      // Recursively delete all ids on the node tree.
-      std::vector<std::shared_ptr<DomNode>> node_stack;
-      node_stack.push_back(node);
-      while (!node_stack.empty()) {
-        auto back_node = node_stack.back();
-        node_stack.pop_back();
-        not_eliminated_node_ids_.erase(back_node->GetId());
-        for (auto& child : back_node->GetChildren()) {
-          node_stack.push_back(child);
-        }
-      }
-    }
     render_manager_->DeleteRenderNode(root_node, std::move(nodes_to_delete));
   }
 }
@@ -289,10 +276,9 @@ bool LayerOptimizedRenderManager::IsJustLayoutProp(const char *prop_name) const
 }
 
 bool LayerOptimizedRenderManager::CanBeEliminated(const std::shared_ptr<DomNode>& node) {
-  bool eliminated = (node->IsLayoutOnly() || node->IsVirtual()) &&
-                    (not_eliminated_node_ids_.find(node->GetId()) == not_eliminated_node_ids_.end());
+  bool eliminated = (node->IsLayoutOnly() || node->IsVirtual()) && node->IsEnableEliminated();
   if (!eliminated) {
-    not_eliminated_node_ids_.insert(node->GetId());
+    node->SetEnableEliminated(false);
   }
   return eliminated;
 }