This simple class takes a SRWebsocket, closes it and retains it until it has been closed. It should prevent https://github.com/square/SocketRocket/pull/169 and other issues.

SRWebSocketCloser.h

@import Foundation;
@class SRWebSocket;

/**
 This simple class takes a SRWebsocket, closes it and retains it until it has been closed.
 */
@interface SRWebSocketCloser : NSObject

+ (instancetype)sharedInstance;
- (void)addWebSocket:(SRWebSocket *)socket;

@end

SRWebSocketCloser.m

#import "SRWebSocketCloser.h"
#import "SRWebSocket.h"

@interface SRWebSocketCloser () <SRWebSocketCloser>
@property (nonatomic, strong) NSHashTable *websockets;
@end

@implementation SRWebSocketCloser

+ (instancetype)sharedInstance {
    static SRWebSocketCloser *sharedInstance = nil;
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        sharedInstance = [SRWebSocketCloser new];
    });
    return sharedInstance;
}

- (instancetype)init {
    self = [super init];
    if (self) {
        _websockets = [NSHashTable hashTableWithOptions:NSHashTableStrongMemory|NSPointerFunctionsObjectPointerPersonality];
    }
    return self;
}

- (void)addWebSocket:(SRWebSocket *)socket {
    if (!socket) {
        return;
    }
    @synchronized(self) {
        [self.websockets addObject:socket];
    }
    
    socket.delegate = self;
    [socket close];
}

- (void)removeWebSocket:(SRWebSocket *)socket {
    if (!socket) {
        return;
    }
    @synchronized(self) {
        [self.websockets removeObject:socket];
    }
}

#pragma mark - SRWebSocketDelegate

- (void)webSocket:(SRWebSocket *)webSocket didReceiveMessage:(id)message {
}

- (void)webSocket:(SRWebSocket *)webSocket didFailWithError:(NSError *)error {
    [self removeWebSocket:webSocket];
}

- (void)webSocket:(SRWebSocket *)webSocket didCloseWithCode:(NSInteger)code reason:(NSString *)reason wasClean:(BOOL)wasClean {
    [self removeWebSocket:webSocket];
}

@end