SocketRocket委托模式：实现自定义消息处理逻辑

SocketRocket委托模式：实现自定义消息处理逻辑

【免费下载链接】SocketRocket A conforming Objective-C WebSocket client library. 项目地址: https://gitcode.com/gh_mirrors/so/SocketRocket

你是否在开发WebSocket应用时遇到消息处理混乱、线程安全问题或难以维护的事件回调？SocketRocket的委托模式（Delegate Pattern）通过清晰的接口设计和线程管理机制，让你轻松实现可靠的自定义消息处理逻辑。本文将带你深入理解委托控制器的工作原理，掌握从协议定义到高级应用的完整实现路径。

委托模式核心组件

SocketRocket的委托系统围绕SRDelegateController构建，该组件位于SocketRocket/Internal/Delegate/SRDelegateController.h，负责管理WebSocket事件的分发与线程安全。其核心功能包括：

• 委托方法可用性检测
• 线程安全的委托调用
• 自定义队列/线程管理

委托方法定义

SRDelegateController通过SRDelegateAvailableMethods结构体定义了WebSocket通信的全生命周期事件：

struct SRDelegateAvailableMethods {
    BOOL didReceiveMessage : 1;
    BOOL didReceiveMessageWithString : 1;
    BOOL didReceiveMessageWithData : 1;
    BOOL didOpen : 1;
    BOOL didFailWithError : 1;
    BOOL didCloseWithCode : 1;
    BOOL didReceivePing : 1;
    BOOL didReceivePong : 1;
    BOOL shouldConvertTextFrameToString : 1;
};

这些方法对应WebSocket连接的关键事件点，从连接建立到消息接收、错误处理等完整流程。

实现自定义委托的步骤

1. 遵循SRWebSocketDelegate协议

首先创建遵循SRWebSocketDelegate协议的类，按需实现所需的委托方法：

#import <SocketRocket/SRWebSocket.h>

@interface CustomWebSocketDelegate : NSObject <SRWebSocketDelegate>
@end

@implementation CustomWebSocketDelegate

// 连接成功回调
- (void)webSocketDidOpen:(SRWebSocket *)webSocket {
    NSLog(@"WebSocket连接已建立");
    // 连接建立后的初始化操作
}

// 接收文本消息
- (void)webSocket:(SRWebSocket *)webSocket didReceiveMessageWithString:(NSString *)string {
    NSLog(@"收到文本消息: %@", string);
    // 自定义文本消息处理逻辑
}

// 连接关闭回调
- (void)webSocket:(SRWebSocket *)webSocket didCloseWithCode:(NSInteger)code reason:(NSString *)reason wasClean:(BOOL)wasClean {
    NSLog(@"连接关闭: %@ (代码: %ld)", reason, (long)code);
    // 连接关闭后的清理工作
}

@end

2. 配置委托控制器

在创建WebSocket实例时，通过SRDelegateController关联自定义委托对象，并可指定事件处理的调度队列：

// 创建WebSocket实例
NSURLRequest *request = [NSURLRequest requestWithURL:[NSURL URLWithString:@"wss://example.com/ws"]];
SRWebSocket *webSocket = [[SRWebSocket alloc] initWithURLRequest:request];

// 配置自定义委托
CustomWebSocketDelegate *customDelegate = [[CustomWebSocketDelegate alloc] init];
webSocket.delegate = customDelegate;

// 可选：指定调度队列（默认主队列）
dispatch_queue_t wsQueue = dispatch_queue_create("com.example.websocket", DISPATCH_QUEUE_SERIAL);
webSocket.delegateController.dispatchQueue = wsQueue;

// 开始连接
[webSocket open];

委托方法调用流程

SRDelegateController通过performDelegateBlock:方法实现线程安全的事件分发，其内部逻辑位于SocketRocket/Internal/Delegate/SRDelegateController.m：

- (void)performDelegateBlock:(SRDelegateBlock)block {
    __block __strong id<SRWebSocketDelegate> delegate = nil;
    __block SRDelegateAvailableMethods availableMethods;
    dispatch_sync(self.accessQueue, ^{
        delegate = self->_delegate;
        availableMethods = self.availableDelegateMethods;
    });
    [self performDelegateQueueBlock:^{
        block(delegate, availableMethods);
    }];
}

该实现通过GCD并发队列确保委托对象的线程安全访问，并根据配置的调度策略（dispatchQueue或operationQueue）分发事件回调。

高级应用技巧

线程安全处理

为避免UI阻塞或数据竞争，建议将耗时操作分配到后台队列处理：

// 在委托控制器中配置后台队列
webSocket.delegateController.dispatchQueue = dispatch_queue_create("com.example.ws.processing", DISPATCH_QUEUE_CONCURRENT);

// 在委托方法中处理耗时操作
- (void)webSocket:(SRWebSocket *)webSocket didReceiveMessageWithData:(NSData *)data {
    dispatch_async(dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_DEFAULT, 0), ^{
        // 处理二进制数据（如解析协议缓冲区、图片解码等）
        id processedData = [self processReceivedData:data];
        
        // 切换回主线程更新UI
        dispatch_async(dispatch_get_main_queue(), ^{
            [self updateUIWithData:processedData];
        });
    });
}

方法可用性检测

在实现通用委托处理器时，可通过availableDelegateMethods检查方法是否实现，避免不必要的调用：

// 伪代码示例：框架内部消息分发逻辑
SRDelegateAvailableMethods methods = delegateController.availableDelegateMethods;
if (methods.didReceiveMessageWithString) {
    [delegate webSocket:webSocket didReceiveMessageWithString:stringData];
} else if (methods.didReceiveMessageWithData) {
    [delegate webSocket:webSocket didReceiveMessageWithData:data];
}

常见问题解决方案

委托对象释放问题

确保在WebSocket连接关闭后正确释放委托对象，避免悬垂指针：

// 在视图控制器生命周期中管理
- (void)dealloc {
    self.webSocket.delegate = nil;
    [self.webSocket close];
}

多委托场景处理

对于需要多个对象接收事件的场景，可实现委托转发器：

@interface DelegateForwarder : NSObject <SRWebSocketDelegate>
@property (nonatomic, copy) NSArray *delegates;
@end

@implementation DelegateForwarder

- (void)webSocketDidOpen:(SRWebSocket *)webSocket {
    for (id<SRWebSocketDelegate> delegate in self.delegates) {
        if ([delegate respondsToSelector:@selector(webSocketDidOpen:)]) {
            [delegate webSocketDidOpen:webSocket];
        }
    }
}
// 实现其他需要转发的委托方法...
@end

总结与最佳实践

SocketRocket的委托模式为WebSocket通信提供了灵活且安全的事件处理机制。通过本文介绍的方法，你可以：

1. 基于SRDelegateController构建可靠的事件处理流程
2. 利用队列配置实现高效的线程管理
3. 通过方法可用性检测优化委托调用
4. 采用转发器模式处理复杂的事件分发需求

建议结合官方示例TestChat/中的实现，该目录包含完整的委托应用实例，可作为实际项目开发的参考模板。

掌握这些技巧后，你将能够构建出响应迅速、易于维护且线程安全的WebSocket应用，从容应对实时通信场景中的各种挑战。

【免费下载链接】SocketRocket A conforming Objective-C WebSocket client library. 项目地址: https://gitcode.com/gh_mirrors/so/SocketRocket