iOS IM开发准备工作（四）CocoaAsyncSocket的使用

　　上一篇乱说了一阵socket，这篇要说说怎么干活了。毕竟用的起来才行。

　　我的项目里面使用的是CocoaAsyncSocket，这个是对CFSocket的封装。如果你觉得自己可以实现封装或者直接用原生的，我可以告诉你，很累；关键是等你弄出来，项目可能都要交了。这个库，支持TCP和UDP；有GCD和RunLoop两种选择。UDP相比TCP的话，可靠性低一点，一般用来传输视频，少个一两帧没有什么影响。这里我就说一下TCP的使用，当然为了发挥Apple设备的牛X性能，我用GCD。

　　　　

建立socket 单例

　　先说一点，要保持长连接。我们需要建一个全局的单例。标准单例模式写法⬇️

+ (instancetype)ShareBaseClient {
    static SocketClient * iSocketCilent;
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        iSocketCilent = [[SocketClient alloc]init];
    });
    return iSocketCilent;
}

#define USE_SECURE_CONNECTION    0 // 是否需要使用安全连接
#define USE_CFSTREAM_FOR_TLS     0 // Use old-school CFStream style technique
#define MANUALLY_EVALUATE_TRUST  1 // 是否需要人工验证


- (id)init {
    self = [super init];
    // Start the socket stuff 最后的那个是你的要放在哪个线程里面操作 果断全局
    asyncSocket = [[GCDAsyncSocket alloc] initWithDelegate:self delegateQueue:dispatch_get_main_queue()];
    NSError *error = nil;
    uint16_t port = DefalutPORT;// 默认的端口号
    // 连接是否成功 
    // WWW_HOST 连接的地址
    // 50秒 是我设置的 超时时间
    if (![asyncSocket connectToHost:WWW_HOST onPort:port withTimeout:50.0f error:&error]){
        
        NSLog(@"Unable to connect to due to invalid configuration: %@", error);

    }else{
        NSLog(@"Connecting to "%@" on port %hu...", WWW_HOST, port);
    }
    {
#if USE_SECURE_CONNECTION
        
#if USE_CFSTREAM_FOR_TLS
        {
            // Use old-school CFStream style technique
            
            NSDictionary *options = @{
                                      GCDAsyncSocketUseCFStreamForTLS : @(YES),
                                      GCDAsyncSocketSSLPeerName : CERT_HOST
                                      };
            
            DDLogVerbose(@"Requesting StartTLS with options:
%@", options);
            [asyncSocket startTLS:options];
        }
#elif MANUALLY_EVALUATE_TRUST
        {
            // Use socket:didReceiveTrust:completionHandler: delegate method for manual trust evaluation
            
            NSDictionary *options = @{
                                      GCDAsyncSocketManuallyEvaluateTrust : @(YES),
                                      GCDAsyncSocketSSLPeerName : CERT_HOST
                                      };
            
            DDLogVerbose(@"Requesting StartTLS with options:
%@", options);
            [asyncSocket startTLS:options];
        }
#else
        {
            // Use default trust evaluation, and provide basic security parameters
            
            NSDictionary *options = @{
                                      GCDAsyncSocketSSLPeerName : CERT_HOST
                                      };
            
            DDLogVerbose(@"Requesting StartTLS with options:
%@", options);
            [asyncSocket startTLS:options];
        }
#endif
#endif
    }
    return self;
}

　

接受报文　

　　如果你连接成功了，那你就可以开始收发报文了。我这里用的是delegate模式，所以我们要实现一下几个比较重要的代理：

　　连接上的第一步，我们要准备读数据。我用一个简单的枚举标识了每次读取数据的区域，一般报文都是先读报头，再去解析正文的。所以连接上之后，要首先接受报头。

#define READ_HEADER_LINE_BY_LINE  0

typedef NS_ENUM(long, ReadTagType){ //读取数据的类型
    ReadTagTypehead = 1,// 报头
    ReadTagTypebody = 2 // 主体
};

- (void)socket:(GCDAsyncSocket *)sock didConnectToHost:(NSString *)host port:(UInt16)port
{
    NSLog(@"socket:didConnectToHost:%@ port:%hu", host, port);

    // 是否一行一行的读数据，我这里设置的是 0
#if READ_HEADER_LINE_BY_LINE
    // Now we tell the socket to read the first line of the http response header.
    // As per the http protocol, we know each header line is terminated with a CRLF (carriage return, line feed).
    [asyncSocket readDataToData:[GCDAsyncSocket CRLFData] withTimeout:-1.0 tag:0];
    
#else
    // Now we tell the socket to read the full header for the http response.
    // As per the http protocol, we know the header is terminated with two CRLF's (carriage return, line feed).
    // sizeof(protocol_head)  一个报文头的长度
    // ReadTagTypehead 先读的是报文的头的
    // －1代表没有超时时间
    [asyncSocket readDataToLength:sizeof(protocol_head) withTimeout:-1 tag:ReadTagTypehead];
    
#endif
}

　　开始读取后，有数据的话会进入这个代理。

- (void)socket:(GCDAsyncSocket *)sock didReadData:(NSData *)data withTag:(long)tag {
    
    switch (tag) {
        case ReadTagTypehead:// 先 读取 head部分
        {
             // 拿到之后，我们处理一下这段头
             // 获得head里面的命令CMD 和 下一段正文的长度BodyLength
             // 把这些参数传到一个方法里面处理一下（万一是空头呢。）
            
            [self willReadBody:CMD size: BodyLength data:data];
        }
            break;
        case ReadTagTypebody:// 再 读取 body部分
        {
            // 加上这一段，我们取得了完整的数据了，可以给需要的地方发过去了
            [self haveReadData:data];
            // 然后我们去读下一个报文,还是先读报文头
            [asyncSocket readDataToLength:sizeof(protocol_head) withTimeout:-1 tag:ReadTagTypehead];
        }
            break;
        default:
            break;
    }
}

// 处理用的方法 下面两个是我写的私有方法 不是CocoaAsyncSocket的代理方法
// 准备读取body的数据 CMD需要记录下来的
- (void)willReadBody:(int)CMD size:(long)size data:(NSData *)data{
   gCMD = CMD;
    alldata = [[NSMutableData alloc]initWithData: data];
    if(size == 0) {// 如果是空头，就丢掉这段数据，继续读下一个头
        [self haveReadData:nil];
        [asyncSocket readDataToLength:sizeof(protocol_head) withTimeout:-1 tag:ReadTagTypehead];
    }else {// 如果不是的话，就去读下一段 报文的正文
        [asyncSocket readDataToLength:dataBufferSize withTimeout:-1 tag:ReadTagTypebody];
    }
}

// 读取完数据了 要回调代理
- (void)haveReadData:(NSData *)data {
    // 把头和正文拼接起来 构成完整的数据
    if(data && data.length>0)
        [alldata appendData:data];
    
//    NSLog(@"receiveData: %@",alldata);
    if(alldata.length>0) {
       // 这里去告诉 你需要处理数据的地方 让他去处理数据
    }
    alldata = nil;
}

　　到上面这一步的话，你的接受数据就算完成了。

发送报文

  [asyncSocket writeData:data withTimeout:-1.0 tag:0]; 

　　就一行，不用怀疑。把你的信息转为NSData后 调用这个方法就好了。成功的话，会进入这个代理。

  - (void)socket:(GCDAsyncSocket *)sock didWriteDataWithTag:(long)tag; 

　

socket 断开

　　当然，socket也有断开的时候，所以你需要处理这个代理。这里面你可以发个通知，让你的application断线重连一下。PS：移动网络和Wi-Fi切换，socket会断开哦。

- (void)socketDidDisconnect:(GCDAsyncSocket *)sock withError:(NSError *)err

附加：

　　当我们连续收到错误报文的时候，我们需要主动断开socket。

// 先去掉代理 再断开连接
self.asyncSocket.delegate = nil;
[self.asyncSocket disconnect];