Update Android websocket module to expose onOpen callback.

discord · Jan 17, 2023 · 7caed29 · 7caed29
1 parent ade0849
commit 7caed29
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/ReactAndroid/src/main/java/com/facebook/react/modules/websocket/WebSocketModule.java b/ReactAndroid/src/main/java/com/facebook/react/modules/websocket/WebSocketModule.java
@@ -48,6 +48,13 @@ public interface ContentHandler {
     void onMessage(ByteString byteString, WritableMap params);
   }
 
+  public interface OnOpenHandler {
+
+    void onOpen(final WebSocket webSocket, final int socketId);
+  }
+
+  public @Nullable OnOpenHandler mOnOpenHandler = null;
+
   private final Map<Integer, WebSocket> mWebSocketConnections = new ConcurrentHashMap<>();
   private final Map<Integer, ContentHandler> mContentHandlers = new ConcurrentHashMap<>();
 
@@ -162,6 +169,10 @@ public void onOpen(WebSocket webSocket, Response response) {
             params.putInt("id", id);
             params.putString("protocol", response.header("Sec-WebSocket-Protocol", ""));
             sendEvent("websocketOpen", params);
+
+            if (mOnOpenHandler != null) {
+              mOnOpenHandler.onOpen(webSocket, id);
+            }
           }
 
           @Override