1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768 |
- /*
- * This file is part of the SDWebImage package.
- * (c) Olivier Poitrey <rs@dailymotion.com>
- *
- * For the full copyright and license information, please view the LICENSE
- * file that was distributed with this source code.
- */
- #import "UIView+WebCacheOperation.h"
- #if SD_UIKIT || SD_MAC
- #import "objc/runtime.h"
- static char loadOperationKey;
- typedef NSMutableDictionary<NSString *, id> SDOperationsDictionary;
- @implementation UIView (WebCacheOperation)
- - (SDOperationsDictionary *)operationDictionary {
- SDOperationsDictionary *operations = objc_getAssociatedObject(self, &loadOperationKey);
- if (operations) {
- return operations;
- }
- operations = [NSMutableDictionary dictionary];
- objc_setAssociatedObject(self, &loadOperationKey, operations, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
- return operations;
- }
- - (void)sd_setImageLoadOperation:(nullable id)operation forKey:(nullable NSString *)key {
- if (key) {
- [self sd_cancelImageLoadOperationWithKey:key];
- if (operation) {
- SDOperationsDictionary *operationDictionary = [self operationDictionary];
- operationDictionary[key] = operation;
- }
- }
- }
- - (void)sd_cancelImageLoadOperationWithKey:(nullable NSString *)key {
- // Cancel in progress downloader from queue
- SDOperationsDictionary *operationDictionary = [self operationDictionary];
- id operations = operationDictionary[key];
- if (operations) {
- if ([operations isKindOfClass:[NSArray class]]) {
- for (id <SDWebImageOperation> operation in operations) {
- if (operation) {
- [operation cancel];
- }
- }
- } else if ([operations conformsToProtocol:@protocol(SDWebImageOperation)]){
- [(id<SDWebImageOperation>) operations cancel];
- }
- [operationDictionary removeObjectForKey:key];
- }
- }
- - (void)sd_removeImageLoadOperationWithKey:(nullable NSString *)key {
- if (key) {
- SDOperationsDictionary *operationDictionary = [self operationDictionary];
- [operationDictionary removeObjectForKey:key];
- }
- }
- @end
- #endif
|