From edbe154a3043d6116b04af9016711f31311b409e Mon Sep 17 00:00:00 2001
From: Lulu Wu <luluwu@fb.com>
Date: Thu, 1 Oct 2020 06:56:57 -0700
Subject: [PATCH] Passing JS callback as a parameter in registerSegment

Summary: Changelog: [Internal]

Reviewed By: makovkastar

Differential Revision: D24000338

fbshipit-source-id: 65d901f6d2bf51a150bce6871517d0a30ab9f821
---
 .../src/main/java/com/facebook/react/bridge/ReactContext.java  | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/ReactAndroid/src/main/java/com/facebook/react/bridge/ReactContext.java b/ReactAndroid/src/main/java/com/facebook/react/bridge/ReactContext.java
index 1bbd069bce3..859d227243f 100644
--- a/ReactAndroid/src/main/java/com/facebook/react/bridge/ReactContext.java
+++ b/ReactAndroid/src/main/java/com/facebook/react/bridge/ReactContext.java
@@ -470,7 +470,8 @@ public class ReactContext extends ContextWrapper {
    * @param segmentId
    * @param path
    */
-  public void registerSegment(int segmentId, String path) {
+  public void registerSegment(int segmentId, String path, Callback callback) {
     Assertions.assertNotNull(mCatalystInstance).registerSegment(segmentId, path);
+    Assertions.assertNotNull(callback).invoke();
   }
 }