feat: Add debug_getTotalAllocatedHybridObjects()

packages/react-native-nitro-modules/cpp/core/HybridObject.cpp

@@ -4,12 +4,23 @@
 
 #include "HybridObject.hpp"
 #include "CommonGlobals.hpp"
+#include "HybridNitroModulesProxy.hpp"
 #include "JSIConverter.hpp"
 #include "NitroDefines.hpp"
 
 namespace margelo::nitro {
 
-HybridObject::HybridObject(const char* NON_NULL name) : HybridObjectPrototype(), _name(name) {}
+HybridObject::HybridObject(const char* NON_NULL name) : HybridObjectPrototype(), _name(name) {
+#ifdef NITRO_DEBUG
+  HybridNitroModulesProxy::debug_notifyHybridObjectAllocated();
+#endif
+}
+
+HybridObject::~HybridObject() {
+#ifdef NITRO_DEBUG
+  HybridNitroModulesProxy::debug_notifyHybridObjectDeallocated();
+#endif
+}
 
 std::string HybridObject::toString() {
   return "[HybridObject " + std::string(_name) + "]";

packages/react-native-nitro-modules/cpp/core/HybridObject.hpp

@@ -35,7 +35,7 @@ class HybridObject : public virtual jsi::NativeState, public HybridObjectPrototy
    * Called when no more references to the given `HybridObject` exist in both C++ and JS.
    * JS might keep references for longer, as it is a garbage collected language.
    */
-  ~HybridObject() override = default;
+  ~HybridObject() override;
   /**
    * HybridObjects cannot be copied.
    */

packages/react-native-nitro-modules/cpp/entrypoint/HybridNitroModulesProxy.cpp

@@ -29,6 +29,8 @@ void HybridNitroModulesProxy::loadHybridMethods() {
 
     prototype.registerHybridGetter("buildType", &HybridNitroModulesProxy::getBuildType);
     prototype.registerHybridGetter("version", &HybridNitroModulesProxy::getVersion);
+
+    prototype.registerHybridGetter("debug_getTotalAllocatedHybridObjects", &HybridNitroModulesProxy::debug_getTotalAllocatedHybridObjects);
   });
 }
 
@@ -94,4 +96,20 @@ std::string HybridNitroModulesProxy::getVersion() {
   return NITRO_VERSION;
 }
 
+// Allocation tests
+static std::atomic_size_t _hybridObjectInstancesCount{0};
+double HybridNitroModulesProxy::debug_getTotalAllocatedHybridObjects() {
+  size_t count = _hybridObjectInstancesCount.load(std::memory_order_relaxed);
+  return static_cast<double>(count);
+}
+
+void HybridNitroModulesProxy::debug_notifyHybridObjectAllocated() {
+  _hybridObjectInstancesCount.fetch_add(1, std::memory_order_relaxed);
+}
+
+void HybridNitroModulesProxy::debug_notifyHybridObjectDeallocated() {
+  _hybridObjectInstancesCount.fetch_sub(1, std::memory_order_relaxed);
+}
+
+
 } // namespace margelo::nitro

packages/react-native-nitro-modules/cpp/entrypoint/HybridNitroModulesProxy.hpp

@@ -46,6 +46,11 @@ class HybridNitroModulesProxy final : public HybridObject {
   std::string getBuildType();
   std::string getVersion();
 
+  // Allocation tests
+  double debug_getTotalAllocatedHybridObjects();
+  static void debug_notifyHybridObjectAllocated();
+  static void debug_notifyHybridObjectDeallocated();
+
 private:
   static constexpr auto TAG = "NitroModulesProxy";
 };

packages/react-native-nitro-modules/src/NitroModulesProxy.ts

@@ -93,4 +93,10 @@ export interface NitroModulesProxy
    * owns it - not JS.
    */
   createNativeArrayBuffer(size: number): ArrayBuffer
+
+  /**
+   * Gets a total number of currently allocated {@linkcode HybridObject}s.
+   * @internal
+   */
+  debug_getTotalAllocatedHybridObjects(): number
 }