Add action that resets recomposition counts

Summary: As per title - this diff adds a custom action which allows for resetting recomposition counts. Reviewed By: luluwu2032 Differential Revision: D57371575 fbshipit-source-id: 81608bd6e6430c6dd28421c385746b98dee9e6d8
facebook · May 15, 2024 · d12cdf3 · d12cdf3
1 parent e2c826d
commit d12cdf3
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/...ose/src/main/java/com/facebook/flipper/plugins/jetpackcompose/UIDebuggerComposeSupport.kt b/...ose/src/main/java/com/facebook/flipper/plugins/jetpackcompose/UIDebuggerComposeSupport.kt
@@ -60,6 +60,9 @@ object UIDebuggerComposeSupport {
               AbstractComposeViewDescriptor.layoutInspector.hideSystemNodes = newValue
               newValue
             }
+        unitAction("Reset Recomposition Counts", ActionIcon.Antd("CloseSquareOutlined")) {
+          AbstractComposeViewDescriptor.resetRecompositionCounts()
+        }
       }
     }
   }

diff --git a/.../com/facebook/flipper/plugins/jetpackcompose/descriptors/AbstractComposeViewDescriptor.kt b/.../com/facebook/flipper/plugins/jetpackcompose/descriptors/AbstractComposeViewDescriptor.kt
@@ -34,6 +34,10 @@ object AbstractComposeViewDescriptor : ChainedDescriptor<AbstractComposeView>()
         }
       }
 
+  fun resetRecompositionCounts() {
+    recompositionHandler.changeCollectionMode(startCollecting = true, keepCounts = false)
+  }
+
   override fun onGetName(node: AbstractComposeView): String = node.javaClass.simpleName
 
   private fun transform(