diff --git a/Classes/Network/PonyDebugger/FLEXNetworkObserver.m b/Classes/Network/PonyDebugger/FLEXNetworkObserver.m index 5245e8201d..221f1dde6a 100644 --- a/Classes/Network/PonyDebugger/FLEXNetworkObserver.m +++ b/Classes/Network/PonyDebugger/FLEXNetworkObserver.m @@ -1591,15 +1591,19 @@ typedef void (^SendMessageBlock)( [FLEXNetworkObserver.sharedObserver websocketTask:slf sendMessagage:message ]; - completion = ^(NSError *error) { + + id completionHook = ^(NSError *error) { [FLEXNetworkObserver.sharedObserver websocketTaskMessageSendCompletion:message error:error ]; + if (completion) { + completion(error); + } }; ((void(*)(id, SEL, id, id))objc_msgSend)( - slf, swizzledSelector, message, completion + slf, swizzledSelector, message, completionHook ); }; diff --git a/Example/FLEXample/MiscNetworkRequests.m b/Example/FLEXample/MiscNetworkRequests.m index 2930d97586..1e7fc14aa9 100644 --- a/Example/FLEXample/MiscNetworkRequests.m +++ b/Example/FLEXample/MiscNetworkRequests.m @@ -153,7 +153,7 @@ - (void)sendExampleNetworkRequests:(NSURLSession *)sessionWithDelegate { } - (void)sendExampleWebsocketTraffic:(NSURLSession *)sessionWithDelegate { - NSString *APIKey = @"oCdCMcMPQpbvNjUIzqtvF1d2X2okWpDQj4AwARJuAgtjhzKxVEjQU6IdCjwm"; + NSString *APIKey = @"VCXCEuvhGcBDP7XhiJJUDvR1e1D3eiVjgZ9VRiaV"; NSString *wsurl = [NSString stringWithFormat:@"wss://demo.piesocket.com/v3/channel_1?api_key=%@¬ify_self", APIKey]; NSURLSessionWebSocketTask *task = [sessionWithDelegate webSocketTaskWithURL:[NSURL URLWithString:wsurl]]; [task resume]; @@ -167,6 +167,8 @@ - (void)URLSession:(NSURLSession *)session webSocketTask:(NSURLSessionWebSocketT [task sendMessage:message completionHandler:^(NSError *error) { if (error) { NSLog(@"Error sending WS message: %@", error.localizedDescription); + } else { + NSLog(@"WS message sent."); } }]; }