SDImageLoadersManager.m 3.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114
  1. /*
  2. * This file is part of the SDWebImage package.
  3. * (c) Olivier Poitrey <rs@dailymotion.com>
  4. *
  5. * For the full copyright and license information, please view the LICENSE
  6. * file that was distributed with this source code.
  7. */
  8. #import "SDImageLoadersManager.h"
  9. #import "SDWebImageDownloader.h"
  10. #import "SDInternalMacros.h"
  11. @interface SDImageLoadersManager ()
  12. @property (nonatomic, strong, nonnull) dispatch_semaphore_t loadersLock;
  13. @end
  14. @implementation SDImageLoadersManager
  15. {
  16. NSMutableArray<id<SDImageLoader>>* _imageLoaders;
  17. }
  18. + (SDImageLoadersManager *)sharedManager {
  19. static dispatch_once_t onceToken;
  20. static SDImageLoadersManager *manager;
  21. dispatch_once(&onceToken, ^{
  22. manager = [[SDImageLoadersManager alloc] init];
  23. });
  24. return manager;
  25. }
  26. - (instancetype)init {
  27. self = [super init];
  28. if (self) {
  29. // initialize with default image loaders
  30. _imageLoaders = [NSMutableArray arrayWithObject:[SDWebImageDownloader sharedDownloader]];
  31. _loadersLock = dispatch_semaphore_create(1);
  32. }
  33. return self;
  34. }
  35. - (NSArray<id<SDImageLoader>> *)loaders {
  36. SD_LOCK(self.loadersLock);
  37. NSArray<id<SDImageLoader>>* loaders = [_imageLoaders copy];
  38. SD_UNLOCK(self.loadersLock);
  39. return loaders;
  40. }
  41. - (void)setLoaders:(NSArray<id<SDImageLoader>> *)loaders {
  42. SD_LOCK(self.loadersLock);
  43. [_imageLoaders removeAllObjects];
  44. if (loaders.count) {
  45. [_imageLoaders addObjectsFromArray:loaders];
  46. }
  47. SD_UNLOCK(self.loadersLock);
  48. }
  49. #pragma mark - Loader Property
  50. - (void)addLoader:(id<SDImageLoader>)loader {
  51. if (![loader conformsToProtocol:@protocol(SDImageLoader)]) {
  52. return;
  53. }
  54. SD_LOCK(self.loadersLock);
  55. [_imageLoaders addObject:loader];
  56. SD_UNLOCK(self.loadersLock);
  57. }
  58. - (void)removeLoader:(id<SDImageLoader>)loader {
  59. if (![loader conformsToProtocol:@protocol(SDImageLoader)]) {
  60. return;
  61. }
  62. SD_LOCK(self.loadersLock);
  63. [_imageLoaders removeObject:loader];
  64. SD_UNLOCK(self.loadersLock);
  65. }
  66. #pragma mark - SDImageLoader
  67. - (BOOL)canRequestImageForURL:(nullable NSURL *)url {
  68. NSArray<id<SDImageLoader>> *loaders = self.loaders;
  69. for (id<SDImageLoader> loader in loaders.reverseObjectEnumerator) {
  70. if ([loader canRequestImageForURL:url]) {
  71. return YES;
  72. }
  73. }
  74. return NO;
  75. }
  76. - (id<SDWebImageOperation>)requestImageWithURL:(NSURL *)url options:(SDWebImageOptions)options context:(SDWebImageContext *)context progress:(SDImageLoaderProgressBlock)progressBlock completed:(SDImageLoaderCompletedBlock)completedBlock {
  77. if (!url) {
  78. return nil;
  79. }
  80. NSArray<id<SDImageLoader>> *loaders = self.loaders;
  81. for (id<SDImageLoader> loader in loaders.reverseObjectEnumerator) {
  82. if ([loader canRequestImageForURL:url]) {
  83. return [loader requestImageWithURL:url options:options context:context progress:progressBlock completed:completedBlock];
  84. }
  85. }
  86. return nil;
  87. }
  88. - (BOOL)shouldBlockFailedURLWithURL:(NSURL *)url error:(NSError *)error {
  89. NSArray<id<SDImageLoader>> *loaders = self.loaders;
  90. for (id<SDImageLoader> loader in loaders.reverseObjectEnumerator) {
  91. if ([loader canRequestImageForURL:url]) {
  92. return [loader shouldBlockFailedURLWithURL:url error:error];
  93. }
  94. }
  95. return NO;
  96. }
  97. @end