AFNetworkActivityIndicatorManager.m 6.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171
  1. // AFNetworkActivityIndicatorManager.m
  2. // Copyright (c) 2011–2015 Alamofire Software Foundation (http://alamofire.org/)
  3. //
  4. // Permission is hereby granted, free of charge, to any person obtaining a copy
  5. // of this software and associated documentation files (the "Software"), to deal
  6. // in the Software without restriction, including without limitation the rights
  7. // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  8. // copies of the Software, and to permit persons to whom the Software is
  9. // furnished to do so, subject to the following conditions:
  10. //
  11. // The above copyright notice and this permission notice shall be included in
  12. // all copies or substantial portions of the Software.
  13. //
  14. // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  15. // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  16. // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  17. // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  18. // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  19. // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  20. // THE SOFTWARE.
  21. #import "AFNetworkActivityIndicatorManager.h"
  22. #if defined(__IPHONE_OS_VERSION_MAX_ALLOWED)
  23. #import "AFHTTPRequestOperation.h"
  24. #if __IPHONE_OS_VERSION_MAX_ALLOWED >= 70000
  25. #import "AFURLSessionManager.h"
  26. #endif
  27. static NSTimeInterval const kAFNetworkActivityIndicatorInvisibilityDelay = 0.17;
  28. static NSURLRequest * AFNetworkRequestFromNotification(NSNotification *notification) {
  29. if ([[notification object] isKindOfClass:[AFURLConnectionOperation class]]) {
  30. return [(AFURLConnectionOperation *)[notification object] request];
  31. }
  32. #if __IPHONE_OS_VERSION_MAX_ALLOWED >= 70000
  33. if ([[notification object] respondsToSelector:@selector(originalRequest)]) {
  34. return [(NSURLSessionTask *)[notification object] originalRequest];
  35. }
  36. #endif
  37. return nil;
  38. }
  39. @interface AFNetworkActivityIndicatorManager ()
  40. @property (readwrite, nonatomic, assign) NSInteger activityCount;
  41. @property (readwrite, nonatomic, strong) NSTimer *activityIndicatorVisibilityTimer;
  42. @property (readonly, nonatomic, getter = isNetworkActivityIndicatorVisible) BOOL networkActivityIndicatorVisible;
  43. - (void)updateNetworkActivityIndicatorVisibility;
  44. - (void)updateNetworkActivityIndicatorVisibilityDelayed;
  45. @end
  46. @implementation AFNetworkActivityIndicatorManager
  47. @dynamic networkActivityIndicatorVisible;
  48. + (instancetype)sharedManager {
  49. static AFNetworkActivityIndicatorManager *_sharedManager = nil;
  50. static dispatch_once_t oncePredicate;
  51. dispatch_once(&oncePredicate, ^{
  52. _sharedManager = [[self alloc] init];
  53. });
  54. return _sharedManager;
  55. }
  56. + (NSSet *)keyPathsForValuesAffectingIsNetworkActivityIndicatorVisible {
  57. return [NSSet setWithObject:@"activityCount"];
  58. }
  59. - (id)init {
  60. self = [super init];
  61. if (!self) {
  62. return nil;
  63. }
  64. [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(networkRequestDidStart:) name:AFNetworkingOperationDidStartNotification object:nil];
  65. [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(networkRequestDidFinish:) name:AFNetworkingOperationDidFinishNotification object:nil];
  66. #if __IPHONE_OS_VERSION_MAX_ALLOWED >= 70000
  67. [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(networkRequestDidStart:) name:AFNetworkingTaskDidResumeNotification object:nil];
  68. [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(networkRequestDidFinish:) name:AFNetworkingTaskDidSuspendNotification object:nil];
  69. [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(networkRequestDidFinish:) name:AFNetworkingTaskDidCompleteNotification object:nil];
  70. #endif
  71. return self;
  72. }
  73. - (void)dealloc {
  74. [[NSNotificationCenter defaultCenter] removeObserver:self];
  75. [_activityIndicatorVisibilityTimer invalidate];
  76. }
  77. - (void)updateNetworkActivityIndicatorVisibilityDelayed {
  78. if (self.enabled) {
  79. // Delay hiding of activity indicator for a short interval, to avoid flickering
  80. if (![self isNetworkActivityIndicatorVisible]) {
  81. [self.activityIndicatorVisibilityTimer invalidate];
  82. self.activityIndicatorVisibilityTimer = [NSTimer timerWithTimeInterval:kAFNetworkActivityIndicatorInvisibilityDelay target:self selector:@selector(updateNetworkActivityIndicatorVisibility) userInfo:nil repeats:NO];
  83. [[NSRunLoop mainRunLoop] addTimer:self.activityIndicatorVisibilityTimer forMode:NSRunLoopCommonModes];
  84. } else {
  85. [self performSelectorOnMainThread:@selector(updateNetworkActivityIndicatorVisibility) withObject:nil waitUntilDone:NO modes:@[NSRunLoopCommonModes]];
  86. }
  87. }
  88. }
  89. - (BOOL)isNetworkActivityIndicatorVisible {
  90. return self.activityCount > 0;
  91. }
  92. - (void)updateNetworkActivityIndicatorVisibility {
  93. [[UIApplication sharedApplication] setNetworkActivityIndicatorVisible:[self isNetworkActivityIndicatorVisible]];
  94. }
  95. - (void)setActivityCount:(NSInteger)activityCount {
  96. @synchronized(self) {
  97. _activityCount = activityCount;
  98. }
  99. dispatch_async(dispatch_get_main_queue(), ^{
  100. [self updateNetworkActivityIndicatorVisibilityDelayed];
  101. });
  102. }
  103. - (void)incrementActivityCount {
  104. [self willChangeValueForKey:@"activityCount"];
  105. @synchronized(self) {
  106. _activityCount++;
  107. }
  108. [self didChangeValueForKey:@"activityCount"];
  109. dispatch_async(dispatch_get_main_queue(), ^{
  110. [self updateNetworkActivityIndicatorVisibilityDelayed];
  111. });
  112. }
  113. - (void)decrementActivityCount {
  114. [self willChangeValueForKey:@"activityCount"];
  115. @synchronized(self) {
  116. #pragma clang diagnostic push
  117. #pragma clang diagnostic ignored "-Wgnu"
  118. _activityCount = MAX(_activityCount - 1, 0);
  119. #pragma clang diagnostic pop
  120. }
  121. [self didChangeValueForKey:@"activityCount"];
  122. dispatch_async(dispatch_get_main_queue(), ^{
  123. [self updateNetworkActivityIndicatorVisibilityDelayed];
  124. });
  125. }
  126. - (void)networkRequestDidStart:(NSNotification *)notification {
  127. if ([AFNetworkRequestFromNotification(notification) URL]) {
  128. [self incrementActivityCount];
  129. }
  130. }
  131. - (void)networkRequestDidFinish:(NSNotification *)notification {
  132. if ([AFNetworkRequestFromNotification(notification) URL]) {
  133. [self decrementActivityCount];
  134. }
  135. }
  136. @end
  137. #endif