Socket连接、心跳、重连、解包（粘包、断包）

上篇已经准备好了基本的条件，接下来就是如何与服务器之间建立一条长连接，以及如何封包、解包；

新建LXSocketManager类，用于对CocoaAsyncSocket进行封装，这样以后如果更换另外的socket库，只需要修改该文件即可。pod下来我们发现CocoaAsyncSocket有两个文件GCDAsyncSocket.h、GCDAsyncUdpSocket.h，前者基于TCP而后者基于UDP，这里选用前者。

// LXSocketManager.h
typedef NS_ENUM(NSInteger, LXSocketStatus) {
    LXSocketStatusUnknown = -1,
    LXSocketStatusUnconnect,
    LXSocketStatusConnect,
};

@class LXSocketManager;
@protocol LXSocketManagerDelegate <NSObject>

@optional
- (void)socketWillSendHeartBeat;
- (void)socket:(LXSocketManager *)socket didConnect:(NSString *)server;
- (void)socket:(LXSocketManager *)socket didReceive:(Message *)message;
@end
@interface LXSocketManager : NSObject

@property (nonatomic, assign, readonly) LXSocketStatus connectStatus;
@property (nonatomic, weak) id<LXSocketManagerDelegate> delegate;

// 连接
- (void)connectTo:(NSString *)host onPort:(uint16_t)port;
// 断开连接
- (void)disconnect;
// 重连
- (void)forceReconnect;
// 发送数据
- (void)sendData:(NSData *)data;
// 开始发送心跳
- (void)startHeartBeat;
@end

点开GCDAsyncSocket.h文件，可以看到以下方法

// 初始化
- (instancetype)init;
- (instancetype)initWithSocketQueue:(nullable dispatch_queue_t)sq;
- (instancetype)initWithDelegate:(nullable id<GCDAsyncSocketDelegate>)aDelegate delegateQueue:(nullable dispatch_queue_t)dq;
- (instancetype)initWithDelegate:(nullable id<GCDAsyncSocketDelegate>)aDelegate delegateQueue:(nullable dispatch_queue_t)dq socketQueue:(nullable dispatch_queue_t)sq;

// 连接
- (BOOL)connectToHost:(NSString *)host onPort:(uint16_t)port error:(NSError **)errPtr;
- (BOOL)connectToHost:(NSString *)host onPort:(uint16_t)port withTimeout:(NSTimeInterval)timeout error:(NSError **)errPtr;

// 断开连接
- (void)disconnect;

首先创建客户端socket

socket = [[GCDAsyncSocket alloc] initWithDelegate:self delegateQueue:dispatch_get_main_queue()];

接下来连接服务端的socket

[socket connectToHost:host onPort:port error:&error];

怎么知道是否连接成功，如何接收数据，socket中断等消息，查看GCDAsyncSocketDelegate代理，会看到以下方法

// 已连接
- (void)socket:(GCDAsyncSocket *)sock didConnectToHost:(NSString *)host port:(uint16_t)port
// 断开连接
- (void)socketDidDisconnect:(GCDAsyncSocket *)sock withError:(NSError *)err
// 接收数据
- (void)socket:(GCDAsyncSocket *)sock didReadData:(NSData *)data withTag:(long)tag

// LXSocketManager.m
#pragma mark - GCDAsyncSocketDelegate
- (void)socket:(GCDAsyncSocket *)sock didConnectToHost:(NSString *)host port:(uint16_t)port {
    LXLog(@"==============socket did connect host: %@, port: %hu==============", host, port);
    //
    [self pullMesasge];
    //
    _connectStatus = LXSocketStatusConnect;
    if ([self.delegate respondsToSelector:@selector(socket:didConnect:)]) {
        NSString *server = [NSString stringWithFormat:@"%@:%d", host, port];
        [self.delegate socket:self didConnect:server];   // 开启心跳等操作
    }
}

- (void)socketDidDisconnect:(GCDAsyncSocket *)sock withError:(NSError *)err {
    LXLog(@"==============socket did disconnect==============");
    // 停止心跳
    [self stopHeartBeat];
    _connectStatus = LXSocketStatusUnconnect;
    // 重连
    [self reconnectIfNeed];
}

- (void)socket:(GCDAsyncSocket *)sock didReadData:(NSData *)data withTag:(long)tag {

    [receiveData appendData:data];
    // 读取包内容长度
    int32_t headLength = 0;
    int32_t contentLength = [self getContentLength:receiveData withHeadLength:&headLength];
    if (contentLength <= 0) {
        [self pullMesasge];
        return;
    }
    // 还未接收到一个完整的数据
    if (headLength + contentLength > [receiveData length]) {
        // 继续接收下一条消息
        [self pullMesasge];
        return;
    }
    // 解析
    [self parseContentDataWithHeadLength:headLength withContentLength:contentLength];
    [self pullMesasge];
}

- (void)pullMesasge {
    [socket readDataWithTimeout:-1 tag:110];
}
心跳

客户端每隔一段时间发送一个数据包给服务端告知服务端我还活着，这就是心跳。心跳的数据需要与服务端约定；当服务端在一定时间内没有收到心跳包，就会断开连接，客户端会收到断开连接的回调，然后进入重连机制。

重连机制

当断开连接后，每过一段时间T重连。在这里时间采用的是指数增长的，并且最大次数是4次。

封包

先将Message.proto文件编译成objc文件，然后直接调用对象delimitedData方法，接着就可以用socket发送我们的数据包了

NSData *data = [message delimitedData];
解包

当我们读取数据的时候，正常的情况是收到一个个完整的数据包，然后再反序列化成我们的ProtoBuf对象，但有时候会出现粘包、断包的情况。如何处理？一个数据包由包头和包体组成，包头有这个数据包的长度信息，因此先获取该数据包的长度，然后根据长度去截取即可。具体代码如下。

粘包、断包

/** 关键代码：获取data数据的内容长度和头部长度: index --> 头部占用长度 (头部占用长度1-4个字节) */
- (int32_t)getContentLength:(NSData *)data withHeadLength:(int32_t *)index {
    
    int8_t tmp = [self readRawByte:data headIndex:index];
    
    if (tmp >= 0) return tmp;
    
    int32_t result = tmp & 0x7f;
    if ((tmp = [self readRawByte:data headIndex:index]) >= 0) {
        result |= tmp << 7;
    } else {
        result |= (tmp & 0x7f) << 7;
        if ((tmp = [self readRawByte:data headIndex:index]) >= 0) {
            result |= tmp << 14;
        } else {
            result |= (tmp & 0x7f) << 14;
            if ((tmp = [self readRawByte:data headIndex:index]) >= 0) {
                result |= tmp << 21;
            } else {
                result |= (tmp & 0x7f) << 21;
                result |= (tmp = [self readRawByte:data headIndex:index]) << 28;
                if (tmp < 0) {
                    for (int i = 0; i < 5; i++) {
                        if ([self readRawByte:data headIndex:index] >= 0) {
                            return result;
                        }
                    }
                    
                    result = -1;
                }
            }
        }
    }
    return result;
}

/** 读取字节 */
- (int8_t)readRawByte:(NSData *)data headIndex:(int32_t *)index{
    
    if (*index >= data.length) return -1;
    
    *index = *index + 1;
    return ((int8_t *)data.bytes)[*index - 1];
}

/** 解析二进制数据：NSData --> 自定义模型对象 */
- (void)parseContentDataWithHeadLength:(int32_t)headL withContentLength:(int32_t)contentL{
    
    NSRange range = NSMakeRange(0, headL + contentL);   //本次解析data的范围
    NSData *data = [receiveData subdataWithRange:range]; //本次解析的data
    
    GPBCodedInputStream *inputStream = [GPBCodedInputStream streamWithData:data];
    
    NSError *error;
    Message *obj = [Message parseDelimitedFromCodedInputStream:inputStream extensionRegistry:nil error:&error];
    
    if (!error){
        if (obj) {
            //保存解析正确的模型对象
            if ([self.delegate respondsToSelector:@selector(socket:didReceive:)]) {
                [self.delegate socket:self didReceive:obj];
            }
        }
        [receiveData replaceBytesInRange:range withBytes:NULL length:0];  //移除已经解析过的data
    }
    
    if (receiveData.length < 1) return;
    
    //对于粘包情况下被合并的多条消息，循环递归直至解析完所有消息
    headL = 0;
    contentL = [self getContentLength:receiveData withHeadLength:&headL];
    if (headL + contentL > receiveData.length) return; //实际包不足解析，继续接收下一个包
    
    [self parseContentDataWithHeadLength:headL withContentLength:contentL]; //继续解析下一条
}
监控网络状态

因为是移动设备，网络状态的改变是非常频繁的；所以需要监控网络状态来做出相应的操作。这里选择的是RealReachability第三方库

// 开启监听
[GLobalRealReachability startNotifier];
[[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(networkChange:) name:kRealReachabilityChangedNotification object:nil];

#pragma mark - network reachability
//
- (void)networkChange:(NSNotification *)notif {
    RealReachability *reachability = (RealReachability *)notif.object;
    ReachabilityStatus status = [reachability currentReachabilityStatus];
    switch (status) {
        case RealStatusNotReachable:
        case RealStatusUnknown: {
            LXLog(@"network unknown or no reachable");
            if (self.socket.connectStatus == LXSocketStatusConnect) {
                [self.socket disconnect];
            }
            break;
        }
        case RealStatusViaWiFi:
        case RealStatusViaWWAN: {
            LXLog(@"wifi or wwan");
            if (self.socket.connectStatus != LXSocketStatusConnect) {
                // 重连
                [self.socket forceReconnect];
            }
            break;
        }
    }
}
参考文章

1、ProtoBuf粘包、断包处理 https://www.cnblogs.com/tandaxia/archive/2017/04/16/6718695.html

最后编辑于 ：
©著作权归作者所有,转载或内容合作请联系作者
  • 人面猴
    序言：七十年代末，一起剥皮案震惊了整个滨河市，随后出现的几起案子，更是在滨河造成了极大的恐慌，老刑警刘岩，带你破解...
    沈念sama阅读 194,524评论 5赞 460
  • 死咒
    序言：滨河连续发生了三起死亡事件，死亡现场离奇诡异，居然都是意外死亡，警方通过查阅死者的电脑和手机，发现死者居然都...
    沈念sama阅读 81,869评论 2赞 371
  • 救了他两次的神仙让他今天三更去死
    文/潘晓璐 我一进店门，熙熙楼的掌柜王于贵愁眉苦脸地迎上来，“玉大人，你说我怎么就摊上这事。” “怎么了？”我有些...
    开封第一讲书人阅读 141,813评论 0赞 320
  • 道士缉凶录：失踪的卖姜人
    文/不坏的土叔 我叫张陵，是天一观的道长。 经常有香客问我，道长，这世上最难降的妖魔是什么？ 我笑而不...
    开封第一讲书人阅读 52,210评论 1赞 263
  • 港岛之恋（遗憾婚礼）
    正文 为了忘掉前任，我火速办了婚礼，结果婚礼上，老公的妹妹穿的比我还像新娘。我一直安慰自己，他们只是感情好，可当我...
    茶点故事阅读 61,085评论 4赞 355
  • 恶毒庶女顶嫁案：这布局不是一般人想出来的
    文/花漫 我一把揭开白布。 她就那样静静地躺着，像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上，一...
    开封第一讲书人阅读 46,117评论 1赞 272
  • 城市分裂传说
    那天，我揣着相机与录音，去河边找鬼。 笑死，一个胖子当着我的面吹牛，可吹牛的内容都是我干的。 我是一名探鬼主播，决...
    沈念sama阅读 36,533评论 3赞 381
  • 双鸳鸯连环套：你想象不到人心有多黑
    文/苍兰香墨 我猛地睁开眼，长吁一口气：“原来是场噩梦啊……” “哼！你这毒妇竟也来了？” 一声冷哼从身侧响起，我...
    开封第一讲书人阅读 35,219评论 0赞 253
  • 万荣杀人案实录
    序言：老挝万荣一对情侣失踪，失踪者是张志新（化名）和其女友刘颖，没想到半个月后，有当地人在树林里发现了一具尸体，经...
    沈念sama阅读 39,487评论 1赞 290
  • 护林员之死
    正文 独居荒郊野岭守林人离奇死亡，尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 34,582评论 2赞 309
  • 白月光启示录
    正文 我和宋清朗相恋三年，在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 36,362评论 1赞 326
  • 活死人
    序言：一个原本活蹦乱跳的男人离奇死亡，死状恐怖，灵堂内的尸体忽然破棺而出，到底是诈尸还是另有隐情，我是刑警宁泽，带...
    沈念sama阅读 32,218评论 3赞 312
  • 日本核电站爆炸内幕
    正文 年R本政府宣布，位于F岛的核电站，受9级特大地震影响，放射性物质发生泄漏。R本人自食恶果不足惜，却给世界环境...
    茶点故事阅读 37,589评论 3赞 299
  • 男人毒药：我在死后第九天来索命
    文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹，春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 28,899评论 0赞 17
  • 一桩弑父案，背后竟有这般阴谋
    文/苍兰香墨 我抬头看了看天上的太阳。三九已至，却和暖如春，着一层夹袄步出监牢的瞬间，已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,176评论 1赞 250
  • 情欲美人皮
    我被黑心中介骗来泰国打工， 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留，地道东北人。 一个月前我还...
    沈念sama阅读 41,503评论 2赞 341
  • 代替公主和亲
    正文 我出身青楼，却偏偏与公主长得像，于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子，可洞房花烛夜当晚...
    茶点故事阅读 40,707评论 2赞 335