| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216 |
- // UIImageView+AFNetworking.m
- // Copyright (c) 2011–2015 Alamofire Software Foundation (http://alamofire.org/)
- //
- // Permission is hereby granted, free of charge, to any person obtaining a copy
- // of this software and associated documentation files (the "Software"), to deal
- // in the Software without restriction, including without limitation the rights
- // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
- // copies of the Software, and to permit persons to whom the Software is
- // furnished to do so, subject to the following conditions:
- //
- // The above copyright notice and this permission notice shall be included in
- // all copies or substantial portions of the Software.
- //
- // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
- // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
- // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
- // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
- // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
- // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
- // THE SOFTWARE.
- #import "UIImageView+AFNetworking.h"
- #import <objc/runtime.h>
- #if defined(__IPHONE_OS_VERSION_MIN_REQUIRED)
- #import "AFHTTPRequestOperation.h"
- @interface AFImageCache : NSCache <AFImageCache>
- @end
- #pragma mark -
- @interface UIImageView (_AFNetworking)
- @property (readwrite, nonatomic, strong, setter = af_setImageRequestOperation:) AFHTTPRequestOperation *af_imageRequestOperation;
- @end
- @implementation UIImageView (_AFNetworking)
- + (NSOperationQueue *)af_sharedImageRequestOperationQueue {
- static NSOperationQueue *_af_sharedImageRequestOperationQueue = nil;
- static dispatch_once_t onceToken;
- dispatch_once(&onceToken, ^{
- _af_sharedImageRequestOperationQueue = [[NSOperationQueue alloc] init];
- _af_sharedImageRequestOperationQueue.maxConcurrentOperationCount = NSOperationQueueDefaultMaxConcurrentOperationCount;
- });
- return _af_sharedImageRequestOperationQueue;
- }
- - (AFHTTPRequestOperation *)af_imageRequestOperation {
- return (AFHTTPRequestOperation *)objc_getAssociatedObject(self, @selector(af_imageRequestOperation));
- }
- - (void)af_setImageRequestOperation:(AFHTTPRequestOperation *)imageRequestOperation {
- objc_setAssociatedObject(self, @selector(af_imageRequestOperation), imageRequestOperation, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
- }
- @end
- #pragma mark -
- @implementation UIImageView (AFNetworking)
- @dynamic imageResponseSerializer;
- + (id <AFImageCache>)sharedImageCache {
- static AFImageCache *_af_defaultImageCache = nil;
- static dispatch_once_t oncePredicate;
- dispatch_once(&oncePredicate, ^{
- _af_defaultImageCache = [[AFImageCache alloc] init];
- [[NSNotificationCenter defaultCenter] addObserverForName:UIApplicationDidReceiveMemoryWarningNotification object:nil queue:[NSOperationQueue mainQueue] usingBlock:^(NSNotification * __unused notification) {
- [_af_defaultImageCache removeAllObjects];
- }];
- });
- #pragma clang diagnostic push
- #pragma clang diagnostic ignored "-Wgnu"
- return objc_getAssociatedObject(self, @selector(sharedImageCache)) ?: _af_defaultImageCache;
- #pragma clang diagnostic pop
- }
- + (void)setSharedImageCache:(__nullable id <AFImageCache>)imageCache {
- objc_setAssociatedObject(self, @selector(sharedImageCache), imageCache, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
- }
- #pragma mark -
- - (id <AFURLResponseSerialization>)imageResponseSerializer {
- static id <AFURLResponseSerialization> _af_defaultImageResponseSerializer = nil;
- static dispatch_once_t onceToken;
- dispatch_once(&onceToken, ^{
- _af_defaultImageResponseSerializer = [AFImageResponseSerializer serializer];
- });
- #pragma clang diagnostic push
- #pragma clang diagnostic ignored "-Wgnu"
- return objc_getAssociatedObject(self, @selector(imageResponseSerializer)) ?: _af_defaultImageResponseSerializer;
- #pragma clang diagnostic pop
- }
- - (void)setImageResponseSerializer:(id <AFURLResponseSerialization>)serializer {
- objc_setAssociatedObject(self, @selector(imageResponseSerializer), serializer, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
- }
- #pragma mark -
- - (void)setImageWithURL:(NSURL *)url {
- [self setImageWithURL:url placeholderImage:nil];
- }
- - (void)setImageWithURL:(NSURL *)url
- placeholderImage:(UIImage *)placeholderImage
- {
- NSMutableURLRequest *request = [NSMutableURLRequest requestWithURL:url];
- [request addValue:@"image/*" forHTTPHeaderField:@"Accept"];
- [self setImageWithURLRequest:request placeholderImage:placeholderImage success:nil failure:nil];
- }
- - (void)setImageWithURLRequest:(NSURLRequest *)urlRequest
- placeholderImage:(UIImage *)placeholderImage
- success:(void (^)(NSURLRequest *request, NSHTTPURLResponse * __nullable response, UIImage *image))success
- failure:(void (^)(NSURLRequest *request, NSHTTPURLResponse * __nullable response, NSError *error))failure
- {
- [self cancelImageRequestOperation];
- UIImage *cachedImage = [[[self class] sharedImageCache] cachedImageForRequest:urlRequest];
- if (cachedImage) {
- if (success) {
- success(urlRequest, nil, cachedImage);
- } else {
- self.image = cachedImage;
- }
- self.af_imageRequestOperation = nil;
- } else {
- if (placeholderImage) {
- self.image = placeholderImage;
- }
- __weak __typeof(self)weakSelf = self;
- self.af_imageRequestOperation = [[AFHTTPRequestOperation alloc] initWithRequest:urlRequest];
- self.af_imageRequestOperation.responseSerializer = self.imageResponseSerializer;
- [self.af_imageRequestOperation setCompletionBlockWithSuccess:^(AFHTTPRequestOperation *operation, id responseObject) {
- __strong __typeof(weakSelf)strongSelf = weakSelf;
- if ([[urlRequest URL] isEqual:[strongSelf.af_imageRequestOperation.request URL]]) {
- if (success) {
- success(urlRequest, operation.response, responseObject);
- } else if (responseObject) {
- strongSelf.image = responseObject;
- }
- if (operation == strongSelf.af_imageRequestOperation){
- strongSelf.af_imageRequestOperation = nil;
- }
- }
- [[[strongSelf class] sharedImageCache] cacheImage:responseObject forRequest:urlRequest];
- } failure:^(AFHTTPRequestOperation *operation, NSError *error) {
- __strong __typeof(weakSelf)strongSelf = weakSelf;
- if ([[urlRequest URL] isEqual:[strongSelf.af_imageRequestOperation.request URL]]) {
- if (failure) {
- failure(urlRequest, operation.response, error);
- }
- if (operation == strongSelf.af_imageRequestOperation){
- strongSelf.af_imageRequestOperation = nil;
- }
- }
- }];
- [[[self class] af_sharedImageRequestOperationQueue] addOperation:self.af_imageRequestOperation];
- }
- }
- - (void)cancelImageRequestOperation {
- [self.af_imageRequestOperation cancel];
- self.af_imageRequestOperation = nil;
- }
- @end
- #pragma mark -
- static inline NSString * AFImageCacheKeyFromURLRequest(NSURLRequest *request) {
- return [[request URL] absoluteString];
- }
- @implementation AFImageCache
- - (UIImage *)cachedImageForRequest:(NSURLRequest *)request {
- switch ([request cachePolicy]) {
- case NSURLRequestReloadIgnoringCacheData:
- case NSURLRequestReloadIgnoringLocalAndRemoteCacheData:
- return nil;
- default:
- break;
- }
- return [self objectForKey:AFImageCacheKeyFromURLRequest(request)];
- }
- - (void)cacheImage:(UIImage *)image
- forRequest:(NSURLRequest *)request
- {
- if (image && request) {
- [self setObject:image forKey:AFImageCacheKeyFromURLRequest(request)];
- }
- }
- @end
- #endif
|