evie/AFNetworking
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Replacing NSMutableData accumulator with NSOutputStream

Browse source
This commit is contained in:
Mattt Thompson 2012-03-25 11:47:51 -07:00
parent f2ae5c40ba
commit 7bd7ee609b
2 changed files with 32 additions and 46 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
AFNetworking/AFURLConnectionOperation.h
View file

@ -80,7 +80,7 @@ extern NSString * const kAFNetworkingIncompleteDownloadDirectoryName;
@warning Attempting to load a `file://` URL in iOS 4 may result in an `NSInvalidArgumentException`, caused by the connection returning `NSURLResponse` rather than `NSHTTPURLResponse`, which is the behavior as of iOS 5.
*/
@interface AFURLConnectionOperation : NSOperation <NSURLConnectionDataDelegate> {
@interface AFURLConnectionOperation : NSOperation <NSURLConnectionDataDelegate, NSStreamDelegate> {
@private
unsigned short _state;
BOOL _cancelled;
@ -95,7 +95,6 @@ extern NSString * const kAFNetworkingIncompleteDownloadDirectoryName;
NSData *_responseData;
long long _totalBytesRead;
NSMutableData *_dataAccumulator;
NSOutputStream *_outputStream;
}

71
AFNetworking/AFURLConnectionOperation.m
View file

@ -30,9 +30,6 @@ typedef enum {
typedef unsigned short AFOperationState;
static NSUInteger const kAFHTTPMinimumInitialDataCapacity = 1024;
static NSUInteger const kAFHTTPMaximumInitialDataCapacity = 1024 * 1024 * 8;
static NSString * const kAFNetworkingLockName = @"com.alamofire.networking.operation.lock";
NSString * const AFNetworkingErrorDomain = @"com.alamofire.networking.error";
@ -128,7 +125,6 @@ static NSString * AFIncompleteDownloadDirectory() {
@property (readwrite, nonatomic, copy) NSString *responseString;
@property (readwrite, nonatomic, copy) NSString *responseFilePath;
@property (readwrite, nonatomic, assign) long long totalBytesRead;
@property (readwrite, nonatomic, retain) NSMutableData *dataAccumulator;
@property (readwrite, nonatomic, copy) AFURLConnectionOperationProgressBlock uploadProgress;
@property (readwrite, nonatomic, copy) AFURLConnectionOperationProgressBlock downloadProgress;
@property (readwrite, nonatomic, copy) AFURLConnectionOperationAuthenticationAgainstProtectionSpaceBlock authenticationAgainstProtectionSpace;
@ -150,7 +146,6 @@ static NSString * AFIncompleteDownloadDirectory() {
@synthesize responseString = _responseString;
@synthesize responseFilePath = _responseFilePath;
@synthesize totalBytesRead = _totalBytesRead;
@synthesize dataAccumulator = _dataAccumulator;
@dynamic inputStream;
@synthesize outputStream = _outputStream;
@synthesize uploadProgress = _uploadProgress;
@ -202,6 +197,9 @@ static NSString * AFIncompleteDownloadDirectory() {
self.request = urlRequest;
self.outputStream = [NSOutputStream outputStreamToMemory];
self.outputStream.delegate = self;
self.state = AFHTTPOperationReadyState;
return self;
@ -218,7 +216,6 @@ static NSString * AFIncompleteDownloadDirectory() {
[_responseData release];
[_responseString release];
[_dataAccumulator release];
if (_outputStream) {
[_outputStream close];
@ -290,8 +287,6 @@ static NSString * AFIncompleteDownloadDirectory() {
}
}
NSLog(@"Request: %@", [self.request allHTTPHeaderFields]);
self.outputStream = [NSOutputStream outputStreamToFileAtPath:temporaryFilePath append:!![self.request valueForHTTPHeaderField:@"Range"]];
}
@ -516,10 +511,6 @@ didReceiveResponse:(NSURLResponse *)response
if (self.outputStream) {
[self.outputStream open];
} else {
NSUInteger maxCapacity = MAX((NSUInteger)llabs(response.expectedContentLength), kAFHTTPMinimumInitialDataCapacity);
NSUInteger capacity = MIN(maxCapacity, kAFHTTPMaximumInitialDataCapacity);
self.dataAccumulator = [NSMutableData dataWithCapacity:capacity];
}
NSHTTPURLResponse *HTTPResponse = (NSHTTPURLResponse *)response;
@ -550,13 +541,9 @@ didReceiveResponse:(NSURLResponse *)response
{
self.totalBytesRead += [data length];
if (self.outputStream) {
if ([self.outputStream hasSpaceAvailable]) {
const uint8_t *dataBuffer = (uint8_t *) [data bytes];
[self.outputStream write:&dataBuffer[0] maxLength:[data length]];
}
} else {
[self.dataAccumulator appendData:data];
if ([self.outputStream hasSpaceAvailable]) {
const uint8_t *dataBuffer = (uint8_t *) [data bytes];
[self.outputStream write:&dataBuffer[0] maxLength:[data length]];
}
if (self.downloadProgress) {
@ -565,27 +552,18 @@ didReceiveResponse:(NSURLResponse *)response
}
- (void)connectionDidFinishLoading:(NSURLConnection *)__unused connection {
if (self.outputStream) {
if (self.responseFilePath) {
NSLog(@"responseFilePath");
@synchronized(self) {
NSString *temporaryFilePath = [AFIncompleteDownloadDirectory() stringByAppendingPathComponent:[[NSNumber numberWithInteger:[self.responseFilePath hash]] stringValue]];
NSLog(@"temporaryFilePath: %@", temporaryFilePath);
NSFileManager *fileManager = [[[NSFileManager alloc] init] autorelease];
[fileManager moveItemAtPath:temporaryFilePath toPath:self.responseFilePath error:&_error];
NSLog(@"Error: %@", _error);
}
if (self.responseFilePath) {
@synchronized(self) {
NSString *temporaryFilePath = [AFIncompleteDownloadDirectory() stringByAppendingPathComponent:[[NSNumber numberWithInteger:[self.responseFilePath hash]] stringValue]];
NSFileManager *fileManager = [[[NSFileManager alloc] init] autorelease];
[fileManager moveItemAtPath:temporaryFilePath toPath:self.responseFilePath error:&_error];
}
[self.outputStream close];
} else {
self.responseData = [NSData dataWithData:self.dataAccumulator];
[_dataAccumulator release]; _dataAccumulator = nil;
}
self.responseData = [self.outputStream propertyForKey:NSStreamDataWrittenToMemoryStreamKey];
[self.outputStream close];
[self finish];
self.connection = nil;
@ -596,11 +574,7 @@ didReceiveResponse:(NSURLResponse *)response
{
self.error = error;
if (self.outputStream) {
[self.outputStream close];
} else {
[_dataAccumulator release]; _dataAccumulator = nil;
}
[self.outputStream close];
[self finish];
@ -621,4 +595,17 @@ didReceiveResponse:(NSURLResponse *)response
}
}
#pragma mark - NSStreamDelegate
- (void)stream:(NSStream *)stream handleEvent:(NSStreamEvent)eventCode {
switch (eventCode) {
case NSStreamEventErrorOccurred:
self.error = [stream streamError];
[self cancel];
break;
default:
break;
}
}
@end