iOS socket编程

Posted 2020-06-17

//
//  ViewController.m
//  socket
//
//  Created by emerys on 16/3/2.
//  Copyright © 2016年 Emerys. All rights reserved.
//

#import "ViewController.h"

#warning 0000前面四位为消息长度，后接消息,最后以#结束

@interface ViewController ()<NSStreamDelegate,UITableViewDataSource>
{
    // 输入流
    NSInputStream *inputStream;
    //  输出流
    NSOutputStream *outputStream;
}
// 用以存储聊天记录，键值为 client_date/service_date
@property (nonatomic,strong) NSMutableArray *history;

@property (nonatomic,copy) NSString *msg;

@property (weak, nonatomic) IBOutlet UITextField *inputTextField;
@property (weak, nonatomic) IBOutlet UIButton *sendMsgButton;
@property (weak, nonatomic) IBOutlet UITableView *historyTableView;

@end

@implementation ViewController

- (void)viewDidLoad {
    [super viewDidLoad];
    // Do any additional setup after loading the view, typically from a nib.
    [self loadHistory];
    
    self.historyTableView.dataSource = self;
    
    [self connect];
}

- (void)didReceiveMemoryWarning {
    [super didReceiveMemoryWarning];
    // Dispose of any resources that can be recreated.
}

/**
 * @brief 懒加载字典
 */
-(NSMutableArray *)history{
    if (!_history) {
        _history = [NSMutableArray array];
    }
    return _history;
}
/**
 * @brief 开启链接
 */
-(void)connect{
    // 1.建立连接
    NSString *host = @"119.29.65.70";
    int port = 9868;

    // 定义C语言输入输出流
    CFReadStreamRef readStream;
    CFWriteStreamRef writeStream;
    CFStreamCreatePairWithSocketToHost(NULL, (__bridge CFStringRef)host, port, &readStream, &writeStream);
    
    // 把C语言的输入输出流转化成OC对象
    inputStream = (__bridge NSInputStream *)(readStream);
    outputStream = (__bridge NSOutputStream *)(writeStream);
    // 设置代理
    inputStream.delegate = self;
    outputStream.delegate = self;
    
    // 把输入输入流添加到主运行循环
    // 不添加主运行循环 代理有可能不工作
    [inputStream scheduleInRunLoop:[NSRunLoop mainRunLoop] forMode:NSDefaultRunLoopMode];
    [outputStream scheduleInRunLoop:[NSRunLoop mainRunLoop] forMode:NSDefaultRunLoopMode];
    
     // 打开输入输出流
    [inputStream open];
    [outputStream open];
}
/**
 * @brief 开启输入／出流监听
 */
-(void)stream:(NSStream *)aStream handleEvent:(NSStreamEvent)eventCode{

    switch (eventCode) {
        case NSStreamEventNone:{
            
        }
            break;
        case NSStreamEventOpenCompleted:{
            // 打开
//            [self showAlertMessage:@"连接服务器成功"];
            NSLog(@"ok");
        }
            break;
        case NSStreamEventHasBytesAvailable:{
            // 有数据可读
            [self readData];
        }
            break;
            

        case NSStreamEventHasSpaceAvailable:{
            // 可发
            if (self.msg) {
                [self writeData];
            }
            
        }
            break;
            

        case NSStreamEventErrorOccurred:{
            // 链接错误,那么重新链接
            [self connect];
        }
            break;
            

        case NSStreamEventEndEncountered:{
            // 结束链接,关闭流，从主运行循环移开
            [inputStream close];
            [outputStream close];
            
            [inputStream removeFromRunLoop:[NSRunLoop mainRunLoop] forMode:NSDefaultRunLoopMode];
            [outputStream removeFromRunLoop:[NSRunLoop mainRunLoop] forMode:NSDefaultRunLoopMode];
        }
            break;
            
        default:
            break;
    }
    
}
/**
 * @brief 读取数据
 */
-(void)readData{
    // 设置1024字节的缓冲区
    uint8_t buf[1024];
    
    NSInteger length = [inputStream read:buf maxLength:1024];
    
    if (length > 0) {
        NSData *data = [NSData dataWithBytes:buf length:length];
#warning 在此编码方式不同可能会引起str为空，添加进数组时程序崩溃
        // gb2312
        NSStringEncoding encoding = CFStringConvertEncodingToNSStringEncoding (kCFStringEncodingGB_18030_2000);
        
        NSString *str = [[NSString alloc] initWithData:data encoding:encoding];
        
//        NSLog(@"%@",str);
        [self.history addObject:str];
        [self.historyTableView reloadData];
        [self saveData];
    }
    
}
/**
 * @brief 发送数据
 */
-(void)writeData{
    NSString *msg = self.msg;
    NSString *head;
    NSUInteger msgLength = msg.length;
    // 超出最大限制,或没有数据, 则不发
    if (!msgLength && msgLength >= 10000) {
        [self showAlertMessage:@"您发送的数据长度超出限度，请重新编辑"];
    }else{
        if (msgLength < 10) {
            head = [NSString stringWithFormat:@"000%li",msgLength];
        }else if(msgLength >= 10 && msgLength < 100){
            head = [NSString stringWithFormat:@"00%li",msgLength];
        }else if (msgLength >= 100 && msgLength < 1000){
            head = [NSString stringWithFormat:@"0%li",msgLength];
        }else if (msgLength >= 1000 && msgLength < 10000){
            head = [NSString stringWithFormat:@"%li",msgLength];
        }
        NSString *str = [NSString stringWithFormat:@"%@%@#",head,msg];
        
        NSData *data = [str dataUsingEncoding:NSUTF8StringEncoding];
        
        [outputStream write:data.bytes maxLength:data.length];
        self.msg = nil;
        [self.history addObject:msg];
        [self.historyTableView reloadData];
        [self saveData];
    }

}
/**
 * @brief 显示一些提示信息
 * @param msg 表示将要显示的信息
 */
-(void)showAlertMessage:(NSString *)msg{
    __block UIAlertController *alert = [UIAlertController alertControllerWithTitle:@"温馨提示" message:msg preferredStyle:UIAlertControllerStyleAlert];
    UIAlertAction *action = [UIAlertAction actionWithTitle:@"确定" style:UIAlertActionStyleDefault handler:^(UIAlertAction * _Nonnull action) {
        [alert dismissViewControllerAnimated:YES completion:nil];
        alert = nil;
    }];
    [alert addAction:action];
    [self presentViewController:alert animated:YES completion:nil];
}
/**
 * @brief 发送数据按钮回调
 */
- (IBAction)senMessage:(id)sender {
    NSString *msg = self.inputTextField.text;
//    NSString *head;
    NSUInteger msgLength = msg.length;
    // 超出最大限制或没有数据, 则不发
    if (!msgLength && msgLength >= 10000) {
        [self showAlertMessage:@"您发送的数据长度超出限度，请重新编辑"];
    }else{
        self.msg = msg;
        [self stream:outputStream handleEvent:NSStreamEventHasSpaceAvailable];
    }

}


#pragma mark 数据源
-(NSInteger)numberOfSectionsInTableView:(UITableView *)tableView{
    return 1;
}
-(NSInteger)tableView:(UITableView *)tableView numberOfRowsInSection:(NSInteger)section{
    return [self.history count];
}
-(UITableViewCell *)tableView:(UITableView *)tableView cellForRowAtIndexPath:(NSIndexPath *)indexPath{
    UITableViewCell *cell = [tableView dequeueReusableCellWithIdentifier:@"cell"];
    if (!cell) {
        cell = [[UITableViewCell alloc] initWithStyle:UITableViewCellStyleDefault reuseIdentifier:@"cell"];
    }
    NSString *str = self.history[indexPath.row];
    cell.textLabel.text = str;
    return cell;
}
/**
 * @brief 检测文件是否存在
 */
-(BOOL)checkFile{
//    NSString *dPath = [NSSearchPathForDirectoriesInDomains(NSDocumentationDirectory, NSUserDomainMask, YES) firstObject];
    NSString *home = NSHomeDirectory();
    NSString *path = [home stringByAppendingPathComponent:@"documents/history.plist"];

    NSFileManager *manager = [NSFileManager defaultManager];
    return [manager fileExistsAtPath:path];
}

/**
 * @brief 保存数据到本地
 */
-(void)saveData{
  
    if (self.history.count) {

        NSString *home = NSHomeDirectory();
        NSString *path = [home stringByAppendingPathComponent:@"documents/history.plist"];
        if([self.history writeToFile:path atomically:YES]){
            NSLog(@"save ok");
        }else{
            NSLog(@"save error");
        }

    }
    
   
}
/**
 * @brief 读取历史数据
 */
-(void)loadHistory{
    NSString *home = NSHomeDirectory();
    NSString *path = [home stringByAppendingPathComponent:@"documents/history.plist"];
    
    NSArray *array = [NSArray arrayWithContentsOfFile:path];
    if (array.count) {
        [self.history addObjectsFromArray:array];
        [self.historyTableView reloadData];
    }

}



@end