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:\n%@", 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:\n%@", options);
            [asyncSocket startTLS:options];
        }
#else
        {
            // Use default trust evaluation, and provide basic security parameters
            
            NSDictionary *options = @{
                                      GCDAsyncSocketSSLPeerName : CERT_HOST
                                      };
            
            DDLogVerbose(@"Requesting StartTLS with options:\n%@", 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];