no message (00cc8243) | Commits | iOS Team / CNLiveWebImage

Browse Code »

Commit 00cc824399ac6138289af1afc5f399f26d755c75

Authored by 王帅 2016-07-29 13:45:54 +0800

0 parents

no message

Inline Side-by-side

Showing 16 changed files with 984 additions and 0 deletions

CNLiveImageCache.h 0 → 100644

View file @00cc824

		1	+++ a/CNLiveImageCache.h
		1	+//
		2	+// CNLiveImageCache.h
		3	+// WSWebImageDemo
		4	+//
		5	+// Created by 王帅 on 16/7/19.
		6	+// Copyright © 2016年王帅. All rights reserved.
		7	+//
		8	+
		9	+#import <Foundation/Foundation.h>
		10	+#import <UIKit/UIKit.h>
		11	+#import "CNLiveWebImageBlock.h"
		12	+
		13	+//typedef void(^CNLiveWebImageQueryCompletedBlock)(UIImage *image, CNLiveImageCacheType type);
		14	+typedef void(^CNLiveCompletedBlock)();
		15	+
		16	+@interface CNLiveImageCache : NSObject
		17	+
		18	++ (CNLiveImageCache *)sharedImageCache;
		19	+
		20	+- (void)storeImage:(UIImage )image forKey:(NSString )key toDisk:(BOOL)toDisk;
		21	+
		22	+- (UIImage )imageFromMemoryCacheForKey:(NSString )key;
		23	+
		24	+- (UIImage )imageFromDiskCacheForKey:(NSString )key;
		25	+
		26	+- (BOOL)queryDiskCacheForKey:(NSString *)key done:(CNLiveWebImageQueryCompletedBlock)doneBlock;
		27	+
		28	+- (void)removeCacheForKey:(NSString *)key done:(CNLiveCompletedBlock)completed;
		29	+
		30	+- (long long)diskCacheSize;
		31	+
		32	+- (void)clearMemory;
		33	+
		34	+- (void)clearDisk;
		35	+
		36	+@end

CNLiveImageCache.m 0 → 100644

View file @00cc824

		1	+++ a/CNLiveImageCache.m
		1	+//
		2	+// CNLiveImageCache.m
		3	+// WSWebImageDemo
		4	+//
		5	+// Created by 王帅 on 16/7/19.
		6	+// Copyright © 2016年王帅. All rights reserved.
		7	+//
		8	+
		9	+#import "CNLiveImageCache.h"
		10	+#import <CommonCrypto/CommonDigest.h>
		11	+#import "UIImage+MultiFormat.h"
		12	+
		13	+FOUNDATION_STATIC_INLINE NSUInteger CNLiveCacheCostForImage(UIImage *image) {
		14	+ return image.size.height * image.size.width * image.scale * image.scale;
		15	+}
		16	+
		17	+typedef void(^CNLiveWebImageNoParamsBlock)();
		18	+
		19	+@interface CNLiveImageCache ()
		20	+
		21	+@property (nonatomic, strong) NSCache *memoryCache;
		22	+@property (nonatomic, copy) NSString *diskCachePath;
		23	+@property (nonatomic, strong) dispatch_queue_t ioQueue;
		24	+
		25	+@end
		26	+
		27	+@implementation CNLiveImageCache
		28	+{
		29	+ NSFileManager *_fileManager;
		30	+}
		31	+
		32	++ (CNLiveImageCache *)sharedImageCache
		33	+{
		34	+ static dispatch_once_t onceToken;
		35	+ static id instance;
		36	+ dispatch_once(&onceToken, ^{
		37	+ instance = [self new];
		38	+ });
		39	+ return instance;
		40	+}
		41	+
		42	+- (id)init {
		43	+ return [self initWithNamespace:@"default"];
		44	+}
		45	+
		46	+- (id)initWithNamespace:(NSString *)ns {
		47	+ NSString *path = [self makeDiskCachePath:ns];
		48	+ return [self initWithNamespace:ns diskCacheDirectory:path];
		49	+}
		50	+
		51	+- (NSString )makeDiskCachePath:(NSString )fullNamespace {
		52	+ NSArray *paths = NSSearchPathForDirectoriesInDomains(NSCachesDirectory, NSUserDomainMask, YES);
		53	+ return [paths[0] stringByAppendingString:fullNamespace];
		54	+}
		55	+
		56	+- (id)initWithNamespace:(NSString )ns diskCacheDirectory:(NSString )directory {
		57	+ if (self = [super init]) {
		58	+ NSString *fullNameSpace = [@"com.cnlive.CNLiveImageCache." stringByAppendingString:ns];
		59	+ _ioQueue = dispatch_queue_create("com.cnlive.CNLiveImageCacheQueue", DISPATCH_QUEUE_SERIAL);
		60	+ _memoryCache = [[NSCache alloc] init];
		61	+ _memoryCache.name = fullNameSpace;
		62	+
		63	+ if (directory != nil) {
		64	+ _diskCachePath = [directory stringByAppendingPathComponent:fullNameSpace];
		65	+ }
		66	+ else {
		67	+ _diskCachePath = [[self makeDiskCachePath:ns] stringByAppendingPathComponent:fullNameSpace];
		68	+ }
		69	+
		70	+ dispatch_sync(_ioQueue, ^{
		71	+ _fileManager = [NSFileManager new];
		72	+ });
		73	+
		74	+ [[NSNotificationCenter defaultCenter] addObserver:self
		75	+ selector:@selector(clearMemory)
		76	+ name:UIApplicationDidReceiveMemoryWarningNotification
		77	+ object:nil];
		78	+
		79	+// [[NSNotificationCenter defaultCenter] addObserver:self
		80	+// selector:@selector(clearDisk)
		81	+// name:UIApplicationWillTerminateNotification
		82	+// object:nil];
		83	+ }
		84	+ return self;
		85	+}
		86	+
		87	+- (void)dealloc
		88	+{
		89	+ [[NSNotificationCenter defaultCenter] removeObserver:self];
		90	+}
		91	+
		92	+- (void)storeImage:(UIImage )image forKey:(NSString )key toDisk:(BOOL)toDisk
		93	+{
		94	+ if (!image \|\| !key) {
		95	+ return;
		96	+ }
		97	+
		98	+ NSUInteger cost = CNLiveCacheCostForImage(image);
		99	+ [_memoryCache setObject:image forKey:key cost:cost];
		100	+
		101	+ if (toDisk) {
		102	+ dispatch_async(_ioQueue, ^{
		103	+ NSData *data = UIImageJPEGRepresentation(image, 1.0);
		104	+
		105	+ if (data) {
		106	+ if (![_fileManager fileExistsAtPath:_diskCachePath]) {
		107	+ [_fileManager createDirectoryAtPath:_diskCachePath withIntermediateDirectories:YES attributes:nil error:NULL];
		108	+ }
		109	+ NSString *cachePathForKey = [self defaultCachePathForKey:key];
		110	+ [_fileManager createFileAtPath:cachePathForKey contents:data attributes:nil];
		111	+ }
		112	+ });
		113	+ }
		114	+}
		115	+
		116	+- (UIImage )imageFromMemoryCacheForKey:(NSString )key
		117	+{
		118	+ return [self.memoryCache objectForKey:key];
		119	+}
		120	+
		121	+- (UIImage )imageFromDiskCacheForKey:(NSString )key
		122	+{
		123	+ UIImage *memoryImage = [self imageFromMemoryCacheForKey:key];
		124	+ if (memoryImage) {
		125	+ return memoryImage;
		126	+ }
		127	+
		128	+ NSData *diskData = [self getDiskImageDataWithKey:key];
		129	+ if (diskData) {
		130	+ return [UIImage ws_imageWithData:diskData];
		131	+ }
		132	+
		133	+ return nil;
		134	+}
		135	+
		136	+- (BOOL)queryDiskCacheForKey:(NSString *)key done:(CNLiveWebImageQueryCompletedBlock)doneBlock
		137	+{
		138	+ if (!doneBlock) {
		139	+ return NO;
		140	+ }
		141	+ if (!key) {
		142	+ doneBlock(nil, CNLiveImageCacheTypeNone);
		143	+ return NO;
		144	+ }
		145	+
		146	+ UIImage *memoryImage = [self imageFromMemoryCacheForKey:key];
		147	+ if (memoryImage) {
		148	+ doneBlock(memoryImage, CNLiveImageCacheTypeMemory);
		149	+ return YES;
		150	+ }
		151	+
		152	+ UIImage *diskImage = [self imageFromDiskCacheForKey:key];
		153	+ if (diskImage) {
		154	+ doneBlock(diskImage, CNLiveImageCacheTypeDisk);
		155	+ return YES;
		156	+ }
		157	+
		158	+ doneBlock(nil, CNLiveImageCacheTypeNone);
		159	+ return NO;
		160	+
		161	+}
		162	+
		163	+- (void)removeCacheForKey:(NSString *)key done:(CNLiveCompletedBlock)completed
		164	+{
		165	+ if (!key \|\| !completed) {
		166	+ return;
		167	+ }
		168	+
		169	+ if ([self imageFromMemoryCacheForKey:key]) {
		170	+ [self.memoryCache removeObjectForKey:key];
		171	+ }
		172	+
		173	+ if ([self imageFromDiskCacheForKey:key]) {
		174	+ NSString *cachePathForKey = [self defaultCachePathForKey:key];
		175	+ [_fileManager removeItemAtPath:cachePathForKey error:nil];
		176	+ }
		177	+
		178	+ completed();
		179	+}
		180	+
		181	+- (long long)diskCacheSize
		182	+{
		183	+ if ([_fileManager fileExistsAtPath:self.diskCachePath]) {
		184	+ NSArray *subPathArray = [_fileManager subpathsAtPath:self.diskCachePath];
		185	+ if (subPathArray && subPathArray.count > 0) {
		186	+ long long totalCacheSize = 0;
		187	+ for (NSString *subPath in subPathArray) {
		188	+ NSString *fullPath = [self.diskCachePath stringByAppendingPathComponent:subPath];
		189	+ NSDictionary *cacheSizeDic = [_fileManager attributesOfItemAtPath:fullPath error:nil];
		190	+ totalCacheSize += [cacheSizeDic fileSize];
		191	+ }
		192	+ return totalCacheSize;
		193	+ }
		194	+ }
		195	+ return 0;
		196	+}
		197	+
		198	+- (void)clearMemory
		199	+{
		200	+ [self.memoryCache removeAllObjects];
		201	+}
		202	+
		203	+- (void)clearDisk
		204	+{
		205	+ [self clearDiskOnCompletion:nil];
		206	+}
		207	+
		208	+- (void)clearDiskOnCompletion:(CNLiveWebImageNoParamsBlock)completion
		209	+{
		210	+ dispatch_async(self.ioQueue, ^{
		211	+ [_fileManager removeItemAtPath:self.diskCachePath error:nil];
		212	+ [_fileManager createDirectoryAtPath:self.diskCachePath withIntermediateDirectories:YES attributes:nil error:NULL];
		213	+
		214	+ if (completion) {
		215	+ dispatch_async(dispatch_get_main_queue(), ^{
		216	+ completion();
		217	+ });
		218	+ }
		219	+ });
		220	+}
		221	+
		222	+#pragma mark <PrivateMethod>
		223	+- (NSData )getDiskImageDataWithKey:(NSString )key
		224	+{
		225	+ NSString *filePath = [self defaultCachePathForKey:key];
		226	+ NSData *imageData = [NSData dataWithContentsOfFile:filePath];
		227	+ return imageData;
		228	+}
		229	+
		230	+- (NSString )defaultCachePathForKey:(NSString )key
		231	+{
		232	+ return [self cachePathForKey:key inPath:self.diskCachePath];
		233	+}
		234	+
		235	+- (NSString )cachePathForKey:(NSString )key inPath:(NSString *)path
		236	+{
		237	+ NSString *fileName = [self cachedFileNameForKey:key];
		238	+ return [path stringByAppendingPathComponent:fileName];
		239	+}
		240	+
		241	+- (NSString )cachedFileNameForKey:(NSString )key {
		242	+ const char *str = [key UTF8String];
		243	+ if (str == NULL) {
		244	+ str = "";
		245	+ }
		246	+ unsigned char r[CC_MD5_DIGEST_LENGTH];
		247	+ CC_MD5(str, (CC_LONG)strlen(str), r);
		248	+ NSString *filename = [NSString stringWithFormat:@"%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x",
		249	+ r[0], r[1], r[2], r[3], r[4], r[5], r[6], r[7], r[8], r[9], r[10], r[11], r[12], r[13], r[14], r[15]];
		250	+
		251	+ return filename;
		252	+}
		253	+
		254	+@end

CNLiveWebImageBlock.h 0 → 100644

View file @00cc824

		1	+++ a/CNLiveWebImageBlock.h
		1	+//
		2	+// CNLiveWebImageBlock.h
		3	+// WSWebImageDemo
		4	+//
		5	+// Created by 王帅 on 16/7/27.
		6	+// Copyright © 2016年王帅. All rights reserved.
		7	+//
		8	+
		9	+#ifndef CNLiveWebImageBlock_h
		10	+#define CNLiveWebImageBlock_h
		11	+
		12	+typedef NS_ENUM(NSUInteger, CNLiveImageCacheType) {
		13	+ CNLiveImageCacheTypeNone,
		14	+ CNLiveImageCacheTypeMemory,
		15	+ CNLiveImageCacheTypeDisk
		16	+};
		17	+
		18	+typedef void(^CNLiveDownloadProgressBlock)(NSInteger receivedSize, NSInteger expectedSize);
		19	+typedef void(^CNLiveDownloadCompletedBlock)(UIImage image, NSData data, NSError *error, BOOL finished);
		20	+
		21	+typedef void(^CNLiveWebImageQueryCompletedBlock)(UIImage *image, CNLiveImageCacheType type);
		22	+typedef void(^CNLiveCompletedBlock)();
		23	+typedef void(^CNLiveWebImageNoParamsBlock)();
		24	+
		25	+#endif /* CNLiveWebImageBlock_h */

CNLiveWebImageDownload.h 0 → 100644

View file @00cc824

		1	+++ a/CNLiveWebImageDownload.h
		1	+//
		2	+// CNLiveWebImageDownload.h
		3	+// WSWebImageDemo
		4	+//
		5	+// Created by 王帅 on 16/7/14.
		6	+// Copyright © 2016年王帅. All rights reserved.
		7	+//
		8	+
		9	+#import <Foundation/Foundation.h>
		10	+#import <UIKit/UIKit.h>
		11	+#import "CNLiveWebImageOperation.h"
		12	+#import "CNLiveWebImageBlock.h"
		13	+
		14	+typedef NS_OPTIONS(NSUInteger, CNLiveWebImageDownloadOptions) {
		15	+ CNLiveWebImageDownloadProgressiveDownload = 1 << 0,
		16	+ CNLiveWebImageDownloadContinueInBackground = 1 << 1
		17	+};
		18	+
		19	+@interface CNLiveWebImageDownload : NSObject
		20	+
		21	++ (CNLiveWebImageDownload *)sharedDownload;
		22	+
		23	+- (id <CNLiveWebImageOperation>)downloadWebImageWithURL:(NSURL *)url options:(CNLiveWebImageDownloadOptions)options progress:(CNLiveDownloadProgressBlock)progress completed:(CNLiveDownloadCompletedBlock)completed;
		24	+
		25	+@end

CNLiveWebImageDownload.m 0 → 100644

View file @00cc824

		1	+++ a/CNLiveWebImageDownload.m
		1	+//
		2	+// CNLiveWebImageDownload.m
		3	+// WSWebImageDemo
		4	+//
		5	+// Created by 王帅 on 16/7/14.
		6	+// Copyright © 2016年王帅. All rights reserved.
		7	+//
		8	+
		9	+#import "CNLiveWebImageDownload.h"
		10	+#import "CNLiveWebImageDownloadOperation.h"
		11	+
		12	+@interface CNLiveWebImageDownload ()
		13	+
		14	+@property (nonatomic, strong) NSOperationQueue *downloadQueue;
		15	+
		16	+@end
		17	+
		18	+@implementation CNLiveWebImageDownload
		19	+
		20	++ (CNLiveWebImageDownload *)sharedDownload
		21	+{
		22	+ static dispatch_once_t once;
		23	+ static id instance;
		24	+ dispatch_once(&once, ^{
		25	+ instance = [self new];
		26	+ });
		27	+ return instance;
		28	+}
		29	+
		30	+- (instancetype)init
		31	+{
		32	+ if ((self = [super init])) {
		33	+ _downloadQueue = [NSOperationQueue new];
		34	+ _downloadQueue.maxConcurrentOperationCount = 6;
		35	+ }
		36	+ return self;
		37	+}
		38	+
		39	+- (void)dealloc
		40	+{
		41	+ [self.downloadQueue cancelAllOperations];
		42	+}
		43	+
		44	+- (id<CNLiveWebImageOperation>)downloadWebImageWithURL:(NSURL *)url options:(CNLiveWebImageDownloadOptions)options progress:(CNLiveDownloadProgressBlock)progress completed:(CNLiveDownloadCompletedBlock)completed
		45	+{
		46	+ CNLiveWebImageDownloadOperation *operation;
		47	+ operation = [[CNLiveWebImageDownloadOperation alloc] initWithURL:url
		48	+ options:options
		49	+ progress:progress
		50	+ completed:completed];
		51	+ [self.downloadQueue addOperation:operation];
		52	+ return (id <CNLiveWebImageOperation>)operation;
		53	+}
		54	+@end

CNLiveWebImageDownloadOperation.h 0 → 100644

View file @00cc824

		1	+++ a/CNLiveWebImageDownloadOperation.h
		1	+//
		2	+// CNLiveWebImageDownloadOperation.h
		3	+// WSWebImageDemo
		4	+//
		5	+// Created by 王帅 on 16/7/25.
		6	+// Copyright © 2016年王帅. All rights reserved.
		7	+//
		8	+
		9	+#import <Foundation/Foundation.h>
		10	+#import "CNLiveWebImageDownload.h"
		11	+#import "CNLiveWebImageBlock.h"
		12	+
		13	+@interface CNLiveWebImageDownloadOperation : NSOperation
		14	+
		15	+- (id)initWithURL:(NSURL *)url options:(CNLiveWebImageDownloadOptions)options progress:(CNLiveDownloadProgressBlock)progress completed:(CNLiveDownloadCompletedBlock)completed;
		16	+
		17	+@end

CNLiveWebImageDownloadOperation.m 0 → 100644

View file @00cc824

		1	+++ a/CNLiveWebImageDownloadOperation.m
		1	+//
		2	+// CNLiveWebImageDownloadOperation.m
		3	+// WSWebImageDemo
		4	+//
		5	+// Created by 王帅 on 16/7/25.
		6	+// Copyright © 2016年王帅. All rights reserved.
		7	+//
		8	+
		9	+#import "CNLiveWebImageDownloadOperation.h"
		10	+#import "CNLiveWebImageOperation.h"
		11	+#import <ImageIO/ImageIO.h>
		12	+#import "UIImage+MultiFormat.h"
		13	+
		14	+#define CNLiveWebImageBackgroundSessionIdentifier @"CNLiveWebImageBackgroundSessionIdentifier"
		15	+
		16	+@interface CNLiveWebImageDownloadOperation ()<CNLiveWebImageOperation, NSURLSessionDataDelegate, NSURLSessionTaskDelegate, NSURLSessionDelegate>
		17	+
		18	+@property (nonatomic, copy) CNLiveDownloadProgressBlock progressBlock;
		19	+@property (nonatomic, copy) CNLiveDownloadCompletedBlock completedBlock;
		20	+@property (nonatomic, assign) CNLiveWebImageDownloadOptions options;
		21	+@property (nonatomic, strong) NSURL *url;
		22	+
		23	+@property (nonatomic, strong) NSURLSessionConfiguration *config;
		24	+@property (nonatomic, strong) NSURLSession *session;
		25	+@property (nonatomic, strong) NSURLSessionDataTask *dataTask;
		26	+@property (nonatomic, assign) NSInteger expectedSize;
		27	+@property (nonatomic, strong) NSMutableData *imageData;
		28	+@property (nonatomic, strong) NSURLResponse *response;
		29	+
		30	+@end
		31	+
		32	+@implementation CNLiveWebImageDownloadOperation
		33	+
		34	+- (id)initWithURL:(NSURL *)url options:(CNLiveWebImageDownloadOptions)options progress:(CNLiveDownloadProgressBlock)progress completed:(CNLiveDownloadCompletedBlock)completed
		35	+{
		36	+ if ((self = [super init])) {
		37	+ _url = url;
		38	+ _progressBlock = progress;
		39	+ _completedBlock = completed;
		40	+ _options = options;
		41	+ }
		42	+ return self;
		43	+}
		44	+
		45	+- (void)start
		46	+{
		47	+ @synchronized (self) {
		48	+ if (self.isCancelled) {
		49	+ return;
		50	+ }
		51	+ }
		52	+ self.dataTask = [self.session dataTaskWithURL:self.url];
		53	+ [self.dataTask resume];
		54	+ [self.session finishTasksAndInvalidate];
		55	+}
		56	+
		57	+- (NSURLSessionConfiguration *)config
		58	+{
		59	+ if (!_config) {
		60	+ if (self.options & CNLiveWebImageDownloadContinueInBackground) {
		61	+ _config = [NSURLSessionConfiguration backgroundSessionConfigurationWithIdentifier:CNLiveWebImageBackgroundSessionIdentifier];
		62	+ }
		63	+ else {
		64	+ _config = [NSURLSessionConfiguration ephemeralSessionConfiguration];
		65	+ }
		66	+ }
		67	+ return _config;
		68	+}
		69	+
		70	+- (NSURLSession *)session
		71	+{
		72	+ if (!_session) {
		73	+ _session = [NSURLSession sessionWithConfiguration:self.config delegate:self delegateQueue:[NSOperationQueue mainQueue]];
		74	+ }
		75	+ return _session;
		76	+}
		77	+
		78	+#pragma mark <NSURLSessionDataDelegate>
		79	+- (void)URLSession:(NSURLSession )session dataTask:(NSURLSessionDataTask )dataTask didReceiveResponse:(NSURLResponse *)response completionHandler:(void (^)(NSURLSessionResponseDisposition))completionHandler
		80	+{
		81	+ if (![response respondsToSelector:@selector(statusCode)] \|\| ([((NSHTTPURLResponse )response) statusCode] < 400 && [((NSHTTPURLResponse )response) statusCode] != 304)) {
		82	+ NSInteger expected = response.expectedContentLength > 0 ? (NSInteger)response.expectedContentLength : 0;
		83	+ self.expectedSize = expected;
		84	+ if (self.progressBlock) {
		85	+ self.progressBlock(0, expected);
		86	+ }
		87	+
		88	+ self.imageData = [[NSMutableData alloc] initWithCapacity:expected];
		89	+ self.response = response;
		90	+ completionHandler(NSURLSessionResponseAllow);
		91	+ }
		92	+ else {
		93	+
		94	+ [self.dataTask cancel];
		95	+
		96	+ dispatch_async(dispatch_get_main_queue(), ^{
		97	+ if (self.completedBlock) {
		98	+ self.completedBlock(nil, nil, [NSError errorWithDomain:NSURLErrorDomain code:[((NSHTTPURLResponse *)response) statusCode] userInfo:nil], YES);
		99	+ }
		100	+ });
		101	+ }
		102	+}
		103	+
		104	+- (void)URLSession:(NSURLSession )session dataTask:(NSURLSessionDataTask )dataTask didReceiveData:(NSData *)data
		105	+{
		106	+ [self.imageData appendData:data];
		107	+ if (self.options & CNLiveWebImageDownloadProgressiveDownload && self.completedBlock) {
		108	+ const NSInteger totalSize = self.imageData.length;
		109	+ CGImageSourceRef imageSource = CGImageSourceCreateWithData((__bridge CFDataRef)self.imageData, NULL);
		110	+ if (totalSize < self.expectedSize) {
		111	+ CGImageRef partialImageRef = CGImageSourceCreateImageAtIndex(imageSource, 0, NULL);
		112	+ if (partialImageRef) {
		113	+ UIImage *image = [UIImage imageWithCGImage:partialImageRef];
		114	+ CGImageRelease(partialImageRef);
		115	+ dispatch_async(dispatch_get_main_queue(), ^{
		116	+ if (self.completedBlock) {
		117	+ self.completedBlock(image, self.imageData, nil, NO);
		118	+ }
		119	+ });
		120	+ }
		121	+ }
		122	+
		123	+ CFRelease(imageSource);
		124	+ }
		125	+
		126	+ dispatch_async(dispatch_get_main_queue(), ^{
		127	+ if (self.progressBlock) {
		128	+ self.progressBlock(self.imageData.length, self.expectedSize);
		129	+ }
		130	+ });
		131	+}
		132	+
		133	+- (void)URLSession:(NSURLSession )session task:(NSURLSessionTask )task didCompleteWithError:(NSError *)error
		134	+{
		135	+ if (error) {
		136	+ dispatch_async(dispatch_get_main_queue(), ^{
		137	+ if (self.completedBlock) {
		138	+ self.completedBlock(nil, nil, error, NO);
		139	+ }
		140	+ });
		141	+ }
		142	+ else {
		143	+ dispatch_async(dispatch_get_main_queue(), ^{
		144	+ if (self.completedBlock) {
		145	+ self.completedBlock([UIImage ws_imageWithData:self.imageData], self.imageData, nil, YES);
		146	+ }
		147	+ });
		148	+ }
		149	+}
		150	+
		151	+- (void)cancel
		152	+{
		153	+ if (self.session) {
		154	+ [self.session invalidateAndCancel];
		155	+ }
		156	+ self.completedBlock = nil;
		157	+ self.progressBlock = nil;
		158	+ self.session = nil;
		159	+ self.imageData = nil;
		160	+}
		161	+
		162	+
		163	+
		164	+
		165	+
		166	+
		167	+
		168	+
		169	+
		170	+
		171	+
		172	+
		173	+
		174	+
		175	+@end

CNLiveWebImageManager.h 0 → 100644

View file @00cc824

		1	+++ a/CNLiveWebImageManager.h
		1	+//
		2	+// CNLiveWebImageManager.h
		3	+// WSWebImageDemo
		4	+//
		5	+// Created by 王帅 on 16/7/27.
		6	+// Copyright © 2016年王帅. All rights reserved.
		7	+//
		8	+
		9	+#import <Foundation/Foundation.h>
		10	+#import <UIKit/UIKit.h>
		11	+#import "CNLiveWebImageBlock.h"
		12	+#import "CNLiveWebImageOperation.h"
		13	+
		14	+typedef NS_OPTIONS(NSUInteger, CNLiveWebImageOptions) {
		15	+ CNLiveWebImageRetryFailed = 1 << 0,
		16	+ CNLiveWebImageCacheMemoryOnly = 1 << 1,
		17	+ CNLiveWebImageProgressiveDownload = 1 << 2,
		18	+ CNLiveWebImageContinueInBackground = 1 << 3
		19	+};
		20	+
		21	+@interface CNLiveWebImageManager : NSObject
		22	+
		23	++ (nonnull id)sharedManager;
		24	+
		25	+- (nonnull id <CNLiveWebImageOperation>)downloadWebImageWithURL:(nonnull NSURL *)url options:(CNLiveWebImageOptions)options progress:(nonnull CNLiveDownloadProgressBlock)progress completed:(nonnull CNLiveDownloadCompletedBlock)completed;
		26	+
		27	+- (void)storeImage:(nonnull UIImage )image imageUrl:(nonnull NSURL )url toDisk:(BOOL)toDisk;
		28	+
		29	+- (nullable UIImage )imageFromMemoryCacheWithImageUrl:(nonnull NSURL )url;
		30	+
		31	+- (nullable UIImage )imageFromDiskCacheWithImageUrl:(nonnull NSURL )url;
		32	+
		33	+- (BOOL)queryDiskCacheWithImageUrl:(nonnull NSURL *)url done:(nonnull CNLiveWebImageQueryCompletedBlock)doneBlock;
		34	+
		35	+- (void)removeCacheWithUrl:(nonnull NSURL *)url done:(nonnull CNLiveCompletedBlock)completed;
		36	+
		37	+- (long long)diskCacheSize;
		38	+
		39	+- (void)clearMemory;
		40	+
		41	+- (void)clearDisk;
		42	+
		43	+@end

CNLiveWebImageManager.m 0 → 100644

View file @00cc824

		1	+++ a/CNLiveWebImageManager.m
		1	+//
		2	+// CNLiveWebImageManager.m
		3	+// WSWebImageDemo
		4	+//
		5	+// Created by 王帅 on 16/7/27.
		6	+// Copyright © 2016年王帅. All rights reserved.
		7	+//
		8	+
		9	+#import "CNLiveWebImageManager.h"
		10	+#import "CNLiveWebImageDownload.h"
		11	+#import "CNLiveImageCache.h"
		12	+
		13	+@interface CNLiveWebImageManager ()
		14	+
		15	+@property (nonatomic, strong) CNLiveImageCache *imageCache;
		16	+@property (nonatomic, strong) NSMutableSet *failedURLs;
		17	+
		18	+@end
		19	+
		20	+@implementation CNLiveWebImageManager
		21	+
		22	++ (id)sharedManager
		23	+{
		24	+ static id manager;
		25	+ static dispatch_once_t once;
		26	+ dispatch_once(&once, ^{
		27	+ manager = [self new];
		28	+ });
		29	+ return manager;
		30	+}
		31	+
		32	+- (instancetype)init
		33	+{
		34	+ if ((self = [super init])) {
		35	+ _imageCache = [CNLiveImageCache sharedImageCache];
		36	+ _failedURLs = [NSMutableSet new];
		37	+ }
		38	+ return self;
		39	+}
		40	+
		41	+- (id<CNLiveWebImageOperation>)downloadWebImageWithURL:(NSURL *)url options:(CNLiveWebImageOptions)options progress:(CNLiveDownloadProgressBlock)progress completed:(CNLiveDownloadCompletedBlock)completed
		42	+{
		43	+ if ([url isKindOfClass:NSString.class]) {
		44	+ url = [NSURL URLWithString:(NSString *)url];
		45	+ }
		46	+ if (![url isKindOfClass:NSURL.class]) {
		47	+ url = nil;
		48	+ }
		49	+
		50	+ BOOL isFailedUrl = NO;
		51	+ @synchronized (self.failedURLs) {
		52	+ isFailedUrl = [self.failedURLs containsObject:url];
		53	+ }
		54	+ if (!url \|\| (!(options & CNLiveWebImageRetryFailed) && isFailedUrl)) {
		55	+ NSError *error = [NSError errorWithDomain:NSURLErrorDomain code:NSURLErrorFileDoesNotExist userInfo:nil];
		56	+ completed(nil, nil, error, YES);
		57	+ }
		58	+
		59	+ BOOL hasCacheImage = [self.imageCache queryDiskCacheForKey:url.absoluteString done:^(UIImage *image, CNLiveImageCacheType type) {
		60	+ if (image) {
		61	+ completed(image,UIImageJPEGRepresentation(image, 1.0), nil, YES);
		62	+ }
		63	+ }];;
		64	+
		65	+ if (hasCacheImage) {
		66	+ return nil;
		67	+ }
		68	+
		69	+
		70	+ CNLiveWebImageDownloadOptions downloadOptions = 0;
		71	+ if (options & CNLiveWebImageProgressiveDownload) downloadOptions \|= CNLiveWebImageDownloadProgressiveDownload;
		72	+ if (options & CNLiveWebImageContinueInBackground) downloadOptions \|= CNLiveWebImageDownloadContinueInBackground;
		73	+ id <CNLiveWebImageOperation> operation = [[CNLiveWebImageDownload sharedDownload] downloadWebImageWithURL:url options:downloadOptions progress:progress completed:^(UIImage image, NSData data, NSError *error, BOOL finished) {
		74	+ if (error) {
		75	+ BOOL shouldBeFailedURLAlliOSVersion = (error.code != NSURLErrorNotConnectedToInternet && error.code != NSURLErrorCancelled && error.code != NSURLErrorTimedOut);
		76	+ BOOL shouldBeFailedURLiOS7 = (NSFoundationVersionNumber > NSFoundationVersionNumber_iOS_6_1 && error.code != NSURLErrorInternationalRoamingOff && error.code != NSURLErrorCallIsActive && error.code != NSURLErrorDataNotAllowed);
		77	+ if (shouldBeFailedURLAlliOSVersion \|\| shouldBeFailedURLiOS7) {
		78	+ @synchronized (self.failedURLs) {
		79	+ [self.failedURLs addObject:url];
		80	+ }
		81	+ }
		82	+ }
		83	+ else {
		84	+ if ((options & CNLiveWebImageRetryFailed)) {
		85	+ @synchronized (self.failedURLs) {
		86	+ [self.failedURLs removeObject:url];
		87	+ }
		88	+ }
		89	+
		90	+ BOOL cacheOnDisk = !(options & CNLiveWebImageCacheMemoryOnly);
		91	+ if (image && finished) {
		92	+ if (cacheOnDisk) {
		93	+ [self.imageCache storeImage:image forKey:[url absoluteString] toDisk:YES];
		94	+ }
		95	+ else {
		96	+ [self.imageCache storeImage:image forKey:[url absoluteString] toDisk:NO];
		97	+ }
		98	+ }
		99	+
		100	+ completed(image, data, error, finished);
		101	+ }
		102	+
		103	+ }];
		104	+ return operation;
		105	+}
		106	+
		107	+- (void)storeImage:(UIImage )image imageUrl:(NSURL )url toDisk:(BOOL)toDisk
		108	+{
		109	+ [self.imageCache storeImage:image forKey:url.absoluteString toDisk:toDisk];
		110	+}
		111	+
		112	+- (UIImage )imageFromMemoryCacheWithImageUrl:(NSURL )url
		113	+{
		114	+ return [self.imageCache imageFromMemoryCacheForKey:url.absoluteString];
		115	+}
		116	+
		117	+- (UIImage )imageFromDiskCacheWithImageUrl:(NSURL )url
		118	+{
		119	+ return [self.imageCache imageFromDiskCacheForKey:url.absoluteString];
		120	+}
		121	+
		122	+- (BOOL)queryDiskCacheWithImageUrl:(NSURL *)url done:(CNLiveWebImageQueryCompletedBlock)doneBlock
		123	+{
		124	+ return [self.imageCache queryDiskCacheForKey:url.absoluteString done:doneBlock];
		125	+}
		126	+
		127	+- (void)removeCacheWithUrl:(NSURL *)url done:(CNLiveCompletedBlock)completed
		128	+{
		129	+ [self.imageCache removeCacheForKey:url.absoluteString done:completed];
		130	+}
		131	+
		132	+- (long long)diskCacheSize
		133	+{
		134	+ return [self.imageCache diskCacheSize];
		135	+}
		136	+
		137	+- (void)clearMemory
		138	+{
		139	+ [self.imageCache clearMemory];
		140	+}
		141	+
		142	+- (void)clearDisk
		143	+{
		144	+ [self.imageCache clearDisk];
		145	+}
		146	+
		147	+@end

CNLiveWebImageOperation.h 0 → 100644

View file @00cc824

		1	+++ a/CNLiveWebImageOperation.h
		1	+//
		2	+// CNLiveWebImageOperation.h
		3	+// WSWebImageDemo
		4	+//
		5	+// Created by 王帅 on 16/7/19.
		6	+// Copyright © 2016年王帅. All rights reserved.
		7	+//
		8	+
		9	+#import <Foundation/Foundation.h>
		10	+
		11	+@protocol CNLiveWebImageOperation <NSObject>
		12	+
		13	+- (void)cancel;
		14	+
		15	+@end

NSData+ImageContentType.h 0 → 100644

View file @00cc824

		1	+++ a/NSData+ImageContentType.h
		1	+//
		2	+// NSData+ImageContentType.h
		3	+// WSWebImageDemo
		4	+//
		5	+// Created by 王帅 on 16/7/21.
		6	+// Copyright © 2016年王帅. All rights reserved.
		7	+//
		8	+
		9	+#import <Foundation/Foundation.h>
		10	+
		11	+@interface NSData (ImageContentType)
		12	+
		13	++ (NSString )ws_imageContentTypeWithData:(NSData )data;
		14	+
		15	+@end

NSData+ImageContentType.m 0 → 100644

View file @00cc824

		1	+++ a/NSData+ImageContentType.m
		1	+//
		2	+// NSData+ImageContentType.m
		3	+// WSWebImageDemo
		4	+//
		5	+// Created by 王帅 on 16/7/21.
		6	+// Copyright © 2016年王帅. All rights reserved.
		7	+//
		8	+
		9	+#import "NSData+ImageContentType.h"
		10	+
		11	+@implementation NSData (ImageContentType)
		12	+
		13	++ (NSString )ws_imageContentTypeWithData:(NSData )data
		14	+{
		15	+ uint8_t c;
		16	+ [data getBytes:&c length:1];
		17	+ switch (c) {
		18	+ case 0xFF:
		19	+ return @"image/jpeg";
		20	+ case 0x89:
		21	+ return @"image/png";
		22	+ case 0x47:
		23	+ return @"image/gif";
		24	+ case 0x49:
		25	+ case 0x4D:
		26	+ return @"image/tiff";
		27	+ case 0x52:
		28	+
		29	+ if ([data length] < 12) {
		30	+ return nil;
		31	+ }
		32	+
		33	+ NSString *testString = [[NSString alloc] initWithData:[data subdataWithRange:NSMakeRange(0, 12)] encoding:NSASCIIStringEncoding];
		34	+ if ([testString hasPrefix:@"RIFF"] && [testString hasSuffix:@"WEBP"]) {
		35	+ return @"image/webp";
		36	+ }
		37	+
		38	+ return nil;
		39	+ }
		40	+ return nil;
		41	+}
		42	+
		43	+@end

UIImage+MultiFormat.h 0 → 100644

View file @00cc824

		1	+++ a/UIImage+MultiFormat.h
		1	+//
		2	+// UIImage+MultiFormat.h
		3	+// WSWebImageDemo
		4	+//
		5	+// Created by 王帅 on 16/7/21.
		6	+// Copyright © 2016年王帅. All rights reserved.
		7	+//
		8	+
		9	+#import <UIKit/UIKit.h>
		10	+
		11	+@interface UIImage (MultiFormat)
		12	+
		13	++ (UIImage )ws_imageWithData:(NSData )data;
		14	+
		15	+@end

UIImage+MultiFormat.m 0 → 100644

View file @00cc824

		1	+++ a/UIImage+MultiFormat.m
		1	+//
		2	+// UIImage+MultiFormat.m
		3	+// WSWebImageDemo
		4	+//
		5	+// Created by 王帅 on 16/7/21.
		6	+// Copyright © 2016年王帅. All rights reserved.
		7	+//
		8	+
		9	+#import "UIImage+MultiFormat.h"
		10	+#import "NSData+ImageContentType.h"
		11	+
		12	+@implementation UIImage (MultiFormat)
		13	+
		14	++ (UIImage )ws_imageWithData:(NSData )data
		15	+{
		16	+ if (!data) {
		17	+ return nil;
		18	+ }
		19	+ UIImage *image;
		20	+ NSString *imageContentType = [NSData ws_imageContentTypeWithData:data];
		21	+
		22	+ if ([imageContentType isEqualToString:@"image/gif"]) {
		23	+ //TODO 处理GIF图片
		24	+ }
		25	+ else if ([imageContentType isEqualToString:@"image/webp"]) {
		26	+ //TODO 处理WEBP图片
		27	+ }
		28	+ else {
		29	+ image = [[UIImage alloc] initWithData:data];
		30	+ }
		31	+ return image;
		32	+}
		33	+
		34	+@end

UIImageView+WebCache.h 0 → 100644

View file @00cc824

		1	+++ a/UIImageView+WebCache.h
		1	+//
		2	+// UIImageView+WebCache.h
		3	+// WSWebImageDemo
		4	+//
		5	+// Created by 王帅 on 16/7/26.
		6	+// Copyright © 2016年王帅. All rights reserved.
		7	+//
		8	+
		9	+#import <UIKit/UIKit.h>
		10	+#import "CNLiveWebImageManager.h"
		11	+
		12	+@interface UIImageView (WebCache)
		13	+
		14	+- (void)ws_setImageWithUrl:(NSURL *)url;
		15	+
		16	+- (void)ws_setImageWithUrl:(NSURL )url placeholderImage:(UIImage )placeholder;
		17	+
		18	+@end

UIImageView+WebCache.m 0 → 100644

View file @00cc824

		1	+++ a/UIImageView+WebCache.m
		1	+//
		2	+// UIImageView+WebCache.m
		3	+// WSWebImageDemo
		4	+//
		5	+// Created by 王帅 on 16/7/26.
		6	+// Copyright © 2016年王帅. All rights reserved.
		7	+//
		8	+
		9	+#import "UIImageView+WebCache.h"
		10	+#import "CNLiveWebImageBlock.h"
		11	+
		12	+@implementation UIImageView (WebCache)
		13	+
		14	+- (void)ws_setImageWithUrl:(NSURL *)url
		15	+{
		16	+ [self ws_setImageWithUrl:url placeholderImage:nil options:0 progress:nil completed:nil];
		17	+}
		18	+
		19	+- (void)ws_setImageWithUrl:(NSURL )url placeholderImage:(UIImage )placeholder
		20	+{
		21	+ [self ws_setImageWithUrl:url placeholderImage:placeholder options:0 progress:nil completed:nil];
		22	+}
		23	+
		24	+- (void)ws_setImageWithUrl:(NSURL )url placeholderImage:(UIImage )placeholder options:(CNLiveWebImageOptions)options progress:(CNLiveDownloadProgressBlock)progress completed:(CNLiveDownloadCompletedBlock)completed
		25	+{
		26	+ if (placeholder) {
		27	+ self.image = placeholder;
		28	+ }
		29	+
		30	+ if (url) {
		31	+ __weak __typeof(self) weakSelf = self;
		32	+ [[CNLiveWebImageManager sharedManager] downloadWebImageWithURL:url
		33	+ options:options
		34	+ progress:progress
		35	+ completed:^(UIImage image, NSData data, NSError *error, BOOL finished) {
		36	+ if (image && !error && finished) {
		37	+ weakSelf.image = image;
		38	+ }
		39	+ }];
		40	+ }
		41	+ else {
		42	+ dispatch_async(dispatch_get_main_queue(), ^{
		43	+ NSError *error = [NSError errorWithDomain:NSURLErrorDomain code:-1 userInfo:@{NSLocalizedDescriptionKey : @"Trying to load a nil url"}];
		44	+ if (completed) {
		45	+ completed(nil, nil, error, NO);
		46	+ }
		47	+ });
		48	+ }
		49	+
		50	+}
		51	+
		52	+@end

iOS Team / CNLiveWebImage · Commits

GitLab

no message