//
// 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 = @"127.0.0.1";
int port = ;
// 定义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[];
NSInteger length = [inputStream read:buf maxLength:];
if (length > ) {
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 >= ) {
[self showAlertMessage:@"您发送的数据长度超出限度，请重新编辑"];
}else{
if (msgLength < ) {
head = [NSString stringWithFormat:@"000%li",msgLength];
}else if(msgLength >= && msgLength < ){
head = [NSString stringWithFormat:@"00%li",msgLength];
}else if (msgLength >= && msgLength < ){
head = [NSString stringWithFormat:@"0%li",msgLength];
}else if (msgLength >= && msgLength < ){
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 >= ) {
[self showAlertMessage:@"您发送的数据长度超出限度，请重新编辑"];
}else{
self.msg = msg;
[self stream:outputStream handleEvent:NSStreamEventHasSpaceAvailable];
}
}
#pragma mark 数据源
-(NSInteger)numberOfSectionsInTableView:(UITableView *)tableView{
return ;
}
-(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
