fix reuse

lkzhao · Sep 15, 2021 · 3272b73 · 3272b73
1 parent 81f2063
commit 3272b73
Showing 1 changed file with 3 additions and 9 deletions.
diff --git a/Sources/UIComponent/Core/Model/ViewRenderNode/ViewRenderNode.swift b/Sources/UIComponent/Core/Model/ViewRenderNode/ViewRenderNode.swift
@@ -20,6 +20,9 @@ public extension ViewRenderNode {
     View()
   }
   func _makeView() -> Any {
+    if View.self is UIView.Type, let reuseKey = reuseKey {
+      return ReuseManager.shared.dequeue(identifier: reuseKey, makeView() as! UIView)
+    }
     return makeView()
   }
   // MARK: AnyViewRenderNode
@@ -28,12 +31,3 @@ public extension ViewRenderNode {
     return updateView(view)
   }
 }
-
-public extension ViewRenderNode where View: UIView {
-  func _makeView() -> Any {
-    if let reuseKey = reuseKey {
-      return ReuseManager.shared.dequeue(identifier: reuseKey, makeView())
-    }
-    return makeView()
-  }
-}