https://github.com/potato512/SYDemo_SYSocket
通常使用网络进行文件的断点续传时，都是文件的断点下载，很少有文件的断点上传的例子。现在通过socket方式做了一个文件的断点上传。socket使用了AsyncSocket，或是GCDAsyncSocket。
AsyncSocket：http://code.google.com/p/cocoaasyncsocket/
GCDAsyncSocket：https://github.com/robbiehanson/CocoaAsyncSocket
在使用时，先将socket进行了封装。然后再使用封装类。
封装类文件
// .h文件 // // SocketManager.h // zhangshaoyu // // Created by zhangshaoyu on 15/9/22. // Copyright (c) 2015年 zhangshaoyu. All rights reserved. // socket断点上传文件 #import <Foundation/Foundation.h> /// 掉线类型（服务端掉线，或用户主动退出） typedef NS_ENUM(NSInteger, SocketDisconnectType) { /// 服务器掉线，默认为0 SocketOfflineByServer = 0, /// 用户主动cut SocketOfflineByUser = 1 }; @interface SocketManager : NSObject /// 掉线类型 @property (nonatomic, assign) SocketDisconnectType disconnecType; /// 文件路径 @property (nonatomic, copy) NSString *filePath; /// 单例 + (SocketManager *)sharedSocket; /// socket连接 - (void)socketConnectWithHost:(NSString *)host port:(UInt16)port; /// socket连接断开 - (void)socketDisconnect; /// GCDSocket连接 - (void)GCDSocketConnectWithHost:(NSString *)host port:(UInt16)port;; /// GCDSocket连接断开 - (void)GCDSocketDisconnect; @end // .m文件 // // SocketManager.m // zhangshaoyu // // Created by zhangshaoyu on 15/9/22. // Copyright (c) 2015年 zhangshaoyu. All rights reserved. // #import "SocketManager.h" #import "AsyncSocket.h" #import "GCDAsyncSocket.h" unsigned long long OffSet = 1024 * 200; static NSInteger const timeWithout = -1; static NSInteger const tagWriteData = 1; static NSInteger const tagReadData = 2; static NSString *const keyOffset = @"keyOffset"; static NSString *const keySouceId = @"keySouceId"; @interface SocketManager () <AsyncSocketDelegate, GCDAsyncSocketDelegate> @property (nonatomic, strong) GCDAsyncSocket *GCDSocket; @property (nonatomic, strong) AsyncSocket *socket; // socket @property (nonatomic, copy) NSString *socketHost; // socket的Host @property (nonatomic, assign) UInt16 socketPort; // socket的prot @property (nonatomic, strong) NSTimer *socketTimer; // 计时器 @property (nonatomic, assign) BOOL isUploadHead; // 文件上传头文件协议 @property (nonatomic, assign) unsigned long long filelength; // 文件上传大小 @property (nonatomic, strong) NSString *fileName; // 文件上传名称 @property (nonatomic, strong) NSString *fileSouceId; // 文件上传id @property (nonatomic, strong) NSFileHandle *fileHandle; // 文件操作 @property (nonatomic, assign) unsigned long long currentOffset; // 当前累计读取文件大小 @end @implementation SocketManager /// 单例 + (SocketManager *)sharedSocket { static SocketManager *sharedInstace = nil; static dispatch_once_t onceToken; dispatch_once(&onceToken, ^{ sharedInstace = [[self alloc] init]; }); return sharedInstace; } #pragma mark - setter - (void)setDisconnecType:(SocketDisconnectType)disconnecType { _disconnecType = disconnecType; self.socket.userData = _disconnecType; } #pragma mark - asyncSocket连接掉线操作 // socket连接 - (void)socketConnectWithHost:(NSString *)host port:(UInt16)port { if (!host || 0 >= host.length) { return; } self.socketHost = host; self.socketPort = port; // 在连接前先进行手动断开 self.disconnecType = SocketOfflineByUser; [self socketDisconnect]; // 确保断开后再连，如果对一个正处于连接状态的socket进行连接，会出现崩溃 self.disconnecType = SocketOfflineByServer; [self socketConnect]; } // 连接 - (void)socketConnect { /* dispatch_async(dispatch_get_global_queue(0, 0), ^{ // 处理耗时操作的代码块... self.isUploadHead = YES; // NSNumber *number = [[NSUserDefaults standardUserDefaults] objectForKey:keyOffset]; // self.currentOffset = number.longLongValue; self.currentOffset = 0; self.socket = [[AsyncSocket alloc] initWithDelegate:self]; NSError *error = nil; [self.socket connectToHost:self.socketHost onPort:self.socketPort withTimeout:3 error:&error]; // //通知主线程刷新 // dispatch_async(dispatch_get_main_queue(), ^{ // //回调或者说是通知主线程刷新， // }); }); */ self.isUploadHead = YES; NSNumber *number = [[NSUserDefaults standardUserDefaults] objectForKey:keyOffset]; self.currentOffset = number.longLongValue; // self.currentOffset = 0; self.socket = [[AsyncSocket alloc] initWithDelegate:self]; NSError *error = nil; [self.socket connectToHost:self.socketHost onPort:self.socketPort withTimeout:timeWithout error:&error]; } // socket断开连接 - (void)socketDisconnect { [self stopTimer:NO]; } #pragma mark - AsyncSocketDelegate // 连接成功回调 - (void)onSocket:(AsyncSocket *)sock didConnectToHost:(NSString *)host port:(UInt16)port { NSLog(@"socket连接成功"); // // 获取服务端返回数据 // [self.socket readDataWithTimeout:timeWithout tag:tagReadData]; // 每隔30s像服务器发送心跳包 // 在longConnectToSocket方法中进行长连接需要向服务器发送的讯息 [self startTimer:NO]; } // 重连 - (void)onSocketDidDisconnect:(AsyncSocket *)sock { NSLog(@"sorry the connect is failure %ld",sock.userData); if (SocketOfflineByServer == self.socket.userData) { // 服务器掉线，重连 [self socketConnect]; } else if (SocketOfflineByUser == self.socket.userData) { // 如果由用户断开，不进行重连 return; } } - (void)onSocket:(AsyncSocket *)sock didReadData:(NSData *)data withTag:(long)tag { NSLog(@"sock read"); // 对得到的data值进行解析与转换即可 [self fileInfoWithData:data]; // if (tagReadData == tag) // { // // 对得到的data值进行解析与转换即可 // NSString *receiveStr = [[NSString alloc] initWithData:data encoding:NSUTF8StringEncoding]; // NSLog(@"receiveStr %@", receiveStr); // NSArray *receiveArray = [receiveStr componentsSeparatedByString:@";"]; // NSString *sourceid = [receiveArray firstObject]; // NSRange rangeSource = [sourceid rangeOfString:@"sourceid="]; // self.fileSouceId = [sourceid substringFromIndex:(rangeSource.location + rangeSource.length)]; // // NSString *position = [receiveArray lastObject]; // NSRange rangePosition = [position rangeOfString:@"position="]; // self.currentOffset = [position substringFromIndex:(rangePosition.location + rangePosition.length)].integerValue; // // [self startTimer:NO]; // } } - (void)onSocket:(AsyncSocket *)sock didWriteDataWithTag:(long)tag { NSLog(@"sock wirte"); if (tagWriteData == tag) { [self startTimer:NO]; } } #pragma mark - timer - (void)startTimer:(BOOL)isGCD { [self uploadFileData:isGCD]; } - (void)stopTimer:(BOOL)isGCD { if (self.fileHandle) { [self.fileHandle closeFile]; self.fileHandle = nil; } if (isGCD) { if (self.GCDSocket) { if ([self.GCDSocket isConnected]) { [self.GCDSocket disconnect]; } self.GCDSocket.delegate = nil; self.GCDSocket = nil; } } else { // 声明是由用户主动切断 self.socket.userData = SocketOfflineByUser; if (self.socket) { if ([self.socket isConnected]) { [self.socket disconnect]; } self.socket.delegate = nil; self.socket = nil; } } self.isUploadHead = NO; NSNumber *number = [NSNumber numberWithLongLong:self.currentOffset]; [[NSUserDefaults standardUserDefaults] setObject:number forKey:keyOffset]; [[NSUserDefaults standardUserDefaults] synchronize]; } #pragma mark - 信息处理 // 文件长度，即大小 - (unsigned long long)filelength { if (!_filelength) { NSData *fileData = [[NSData alloc] initWithContentsOfFile:self.filePath]; _filelength = fileData.length; NSLog(@"fileData length %llu", _filelength); return _filelength; } return _filelength; } // 文件名称 - (NSString *)fileName { if (!_fileName) { NSRange fileRange = [self.filePath rangeOfString:@"/" options:NSBackwardsSearch]; if (fileRange.location != NSNotFound) { _fileName = [self.filePath substringFromIndex:(fileRange.location + fileRange.length)]; NSLog(@"filename %@", _fileName); } return _fileName; } return _fileName; } // 对得到的data值进行解析与转换即可 - (void)fileInfoWithData:(NSData *)data { // 对得到的data值进行解析与转换即可 NSString *receiveStr = [[NSString alloc] initWithData:data encoding:NSUTF8StringEncoding]; NSLog(@"receiveStr %@", receiveStr); NSArray *receiveArray = [receiveStr componentsSeparatedByString:@";"]; NSString *sourceid = [receiveArray firstObject]; NSRange rangeSource = [sourceid rangeOfString:@"sourceid="]; self.fileSouceId = [sourceid substringFromIndex:(rangeSource.location + rangeSource.length)]; [[NSUserDefaults standardUserDefaults] setObject:self.fileSouceId forKey:keySouceId]; [[NSUserDefaults standardUserDefaults] synchronize]; NSString *position = [receiveArray lastObject]; NSRange rangePosition = [position rangeOfString:@"position="]; self.currentOffset = [position substringFromIndex:(rangePosition.location + rangePosition.length)].integerValue; } // 上传文件 - (void)uploadFileData:(BOOL)isGCD { if (self.isUploadHead) { // 判断文件是否已有上传记录 self.fileSouceId = [[NSUserDefaults standardUserDefaults] objectForKey:keySouceId]; // 构造拼接协议 NSString *headStr = [[NSString alloc] initWithFormat:@"Content-Length=%llu;filename=%@;sourceid=%@\r\n", self.filelength, self.fileName, ((self.fileSouceId && 0 < self.fileSouceId.length) ? self.fileSouceId : @"")]; NSData *headData = [headStr dataUsingEncoding:NSUTF8StringEncoding]; if (isGCD) { [self.GCDSocket writeData:headData withTimeout:timeWithout tag:tagWriteData]; // 获取服务端返回数据 [self.GCDSocket readDataWithTimeout:timeWithout tag:tagReadData]; } else { [self.socket writeData:headData withTimeout:timeWithout tag:tagWriteData]; // 获取服务端返回数据 [self.socket readDataWithTimeout:timeWithout tag:tagReadData]; } headData = nil; self.isUploadHead = NO; } else { if (!self.fileHandle) { self.fileHandle = [NSFileHandle fileHandleForReadingAtPath:self.filePath]; } if (self.fileHandle) { // 文件当前上传记录位置，及是否继续上传 if (self.filelength > self.currentOffset) { NSLog(@"currentOffset %llu", self.currentOffset); [self.fileHandle seekToFileOffset:self.currentOffset]; self.currentOffset += OffSet; NSData *bodyData = [self.fileHandle readDataOfLength:self.currentOffset]; if (isGCD) { [self.GCDSocket writeData:bodyData withTimeout:timeWithout tag:tagWriteData]; } else { [self.socket writeData:bodyData withTimeout:timeWithout tag:tagWriteData]; } bodyData = nil; } else { // 停止上传，即上传完成 [self stopTimer:isGCD]; } } } } #pragma mark - gcdasyncSocket连接掉线操作 - (void)GCDSocketConnectWithHost:(NSString *)host port:(UInt16)port; { if (!host || 0 >= host.length) { return; } self.socketHost = host; self.socketPort = port; self.isUploadHead = YES; NSNumber *number = [[NSUserDefaults standardUserDefaults] objectForKey:keyOffset]; self.currentOffset = number.longLongValue; // self.currentOffset = 0; // Create our GCDAsyncSocket instance. // dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_DEFAULT, 0) dispatch_get_main_queue() self.GCDSocket = [[GCDAsyncSocket alloc] initWithDelegate:self delegateQueue:dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_DEFAULT, 0)]; // Now we tell the ASYNCHRONOUS socket to connect. NSError *error = nil; if (![self.GCDSocket connectToHost:self.socketHost onPort:self.socketPort error:&error]) { NSLog(@"Unable to connect to due to invalid configuration: %@", error); } else { NSLog(@"Connecting to \"%@\" on port %hu...", self.socketHost, self.socketPort); } } - (void)GCDSocketDisconnect { [self stopTimer:YES]; } #pragma mark GCDAsyncSocketDelegate - (void)socketDidDisconnect:(GCDAsyncSocket *)sock withError:(NSError *)err { // Since we requested HTTP/1.0, we expect the server to close the connection as soon as it has sent the response. NSLog(@"6 socketDidDisconnect：%@", err); [self GCDSocketDisconnect]; } - (void)socket:(GCDAsyncSocket *)sock didConnectToHost:(NSString *)host port:(UInt16)port { NSLog(@"1 didConnectToHost:%@ port:%hu", host, port); [self startTimer:YES]; } - (void)socket:(GCDAsyncSocket *)sock didReceiveTrust:(SecTrustRef)trust completionHandler:(void (^)(BOOL shouldTrustPeer))completionHandler { NSLog(@"2 didReceiveTrust:"); } - (void)socketDidSecure:(GCDAsyncSocket *)sock { // This method will be called if USE_SECURE_CONNECTION is set NSLog(@"3 socketDidSecure:"); } - (void)socket:(GCDAsyncSocket *)sock didWriteDataWithTag:(long)tag { NSLog(@"4 didWriteDataWithTag:"); if (tagWriteData == tag) { [self startTimer:YES]; } } - (void)socket:(GCDAsyncSocket *)sock didReadData:(NSData *)data withTag:(long)tag { NSLog(@"5 didReadData:"); if (tagReadData == tag) { // 对得到的data值进行解析与转换即可 [self fileInfoWithData:data]; } } @end
使用
// 1 导入头文件 #import "SocketManager.h" // 2 启动socket连接 - (void)socketStart:(UIButton *)button { // 设置IP地址 NSString *host = @"192.168.54.160"; // @"192.186.100.21"; // kTestHost // 设置端口 UInt16 port = 7876; // 10045; // 23 // 设置文件路径 NSString *filePaht = [[NSBundle mainBundle] pathForResource:@"text" ofType:@"txt"]; // NSString *filePaht = [[NSBundle mainBundle] pathForResource:@"dawan" ofType:@"mp4"]; // NSString *filePaht = [[NSBundle mainBundle] pathForResource:@"hongri" ofType:@"mp3"]; // 设置socket要上传的文件路径 [SocketManager sharedSocket].filePath = filePaht; // 启动socket连接 // [[SocketManager sharedSocket] socketConnectWithHost:host port:port]; [[SocketManager sharedSocket] GCDSocketConnectWithHost:host port:port]; } // 3 关闭socket连接 - (void)socketStop:(UIButton *)button { // 关闭socket连接 // [[SocketManager sharedSocket] socketDisconnect]; [[SocketManager sharedSocket] GCDSocketDisconnect]; }
