Expose didCloseWithCode in RCTReconnectingWebSocket

Summary: Expose didCloseWithCode in RCTReconnectingWebSocket in order to get notified when the reconnecting websocket is closed by the end-point. Reviewed By: javache Differential Revision: D5573394 fbshipit-source-id: 78bffeb98c6bf32b059194fc07ffc1f0a7bf4aae
facebook · Aug 7, 2017 · 72dae51 · 72dae51
1 parent 220034c
commit 72dae51
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 0 deletions.
diff --git a/Libraries/WebSocket/RCTReconnectingWebSocket.h b/Libraries/WebSocket/RCTReconnectingWebSocket.h
@@ -17,6 +17,8 @@
 
 - (void)webSocket:(RCTSRWebSocket *)webSocket didReceiveMessage:(id)message;
 
+- (void)webSocket:(RCTSRWebSocket *)webSocket didCloseWithCode:(NSInteger)code reason:(NSString *)reason wasClean:(BOOL)wasClean;
+
 @end
 
 @interface RCTReconnectingWebSocket : NSObject

diff --git a/Libraries/WebSocket/RCTReconnectingWebSocket.m b/Libraries/WebSocket/RCTReconnectingWebSocket.m
@@ -80,6 +80,7 @@ - (void)webSocket:(RCTSRWebSocket *)webSocket didFailWithError:(NSError *)error
 
 - (void)webSocket:(RCTSRWebSocket *)webSocket didCloseWithCode:(NSInteger)code reason:(NSString *)reason wasClean:(BOOL)wasClean
 {
+  [self.delegate webSocket:webSocket didCloseWithCode:code reason:reason wasClean:wasClean];
   [self reconnect];
 }
 

diff --git a/Libraries/WebSocket/RCTWebSocketObserver.m b/Libraries/WebSocket/RCTWebSocketObserver.m
@@ -61,6 +61,10 @@ - (void)webSocket:(RCTSRWebSocket *)webSocket didReceiveMessage:(id)message
   }
 }
 
+- (void)webSocket:(RCTSRWebSocket *)webSocket didCloseWithCode:(NSInteger)code reason:(NSString *)reason wasClean:(BOOL)wasClean
+{
+}
+
 @end
 
 #endif
diff --git a/React/DevSupport/RCTPackagerConnection.m b/React/DevSupport/RCTPackagerConnection.m
@@ -133,6 +133,10 @@ - (void)webSocket:(RCTSRWebSocket *)webSocket didReceiveMessage:(id)message
   }
 }
 
+- (void)webSocket:(RCTSRWebSocket *)webSocket didCloseWithCode:(NSInteger)code reason:(NSString *)reason wasClean:(BOOL)wasClean
+{
+}
+
 @end
 
 #endif