使用socket（AsyncSocket、GCDAsyncSocket）进行文件的断点上传操作

最新推荐文章于 2024-02-29 22:49:36 发布

Gz_Jason_chen

最新推荐文章于 2024-02-29 22:49:36 发布

阅读量980

点赞数

分类专栏：网络通信

网络通信专栏收录该内容

13 篇文章

订阅专栏

本文介绍了一种使用AsyncSocket和GCDAsyncSocket实现的iOS应用中文件断点上传的方法。通过对socket进行封装，并利用定时器控制文件上传进度，实现了稳定且可恢复的文件上传过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

转自:http://blog.youkuaiyun.com/potato512/article/details/53432559

关于分包:http://ios.jobbole.com/92889/?from=singlemessage&isappinstalled=0

https://github.com/potato512/SYDemo_SYSocket

通常使用网络进行文件的断点续传时，都是文件的断点下载，很少有文件的断点上传的例子。现在通过socket方式做了一个文件的断点上传。socket使用了AsyncSocket，或是GCDAsyncSocket。

AsyncSocket：http://code.google.com/p/cocoaasyncsocket/

GCDAsyncSocket：https://github.com/robbiehanson/CocoaAsyncSocket

在使用时，先将socket进行了封装。然后再使用封装类。

封装类文件

[html]view plaincopy 
   
 // .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  

[html]view plaincopy 
   
 // .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  

使用

[html]view plaincopy 
   
 // 1 导入头文件  
 #import "SocketManager.h"  

[html]view plaincopy 
   
 // 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];  
 }  

[html]view plaincopy 
   
 // 3 关闭socket连接  
 - (void)socketStop:(UIButton *)button  
 {  
     // 关闭socket连接  
 //    [[SocketManager sharedSocket] socketDisconnect];  
     [[SocketManager sharedSocket] GCDSocketDisconnect];  
 }  