| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171 |
- // AFNetworkActivityIndicatorManager.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 "AFNetworkActivityIndicatorManager.h"
- #if defined(__IPHONE_OS_VERSION_MAX_ALLOWED)
- #import "AFHTTPRequestOperation.h"
- #if __IPHONE_OS_VERSION_MAX_ALLOWED >= 70000
- #import "AFURLSessionManager.h"
- #endif
- static NSTimeInterval const kAFNetworkActivityIndicatorInvisibilityDelay = 0.17;
- static NSURLRequest * AFNetworkRequestFromNotification(NSNotification *notification) {
- if ([[notification object] isKindOfClass:[AFURLConnectionOperation class]]) {
- return [(AFURLConnectionOperation *)[notification object] request];
- }
- #if __IPHONE_OS_VERSION_MAX_ALLOWED >= 70000
- if ([[notification object] respondsToSelector:@selector(originalRequest)]) {
- return [(NSURLSessionTask *)[notification object] originalRequest];
- }
- #endif
- return nil;
- }
- @interface AFNetworkActivityIndicatorManager ()
- @property (readwrite, nonatomic, assign) NSInteger activityCount;
- @property (readwrite, nonatomic, strong) NSTimer *activityIndicatorVisibilityTimer;
- @property (readonly, nonatomic, getter = isNetworkActivityIndicatorVisible) BOOL networkActivityIndicatorVisible;
- - (void)updateNetworkActivityIndicatorVisibility;
- - (void)updateNetworkActivityIndicatorVisibilityDelayed;
- @end
- @implementation AFNetworkActivityIndicatorManager
- @dynamic networkActivityIndicatorVisible;
- + (instancetype)sharedManager {
- static AFNetworkActivityIndicatorManager *_sharedManager = nil;
- static dispatch_once_t oncePredicate;
- dispatch_once(&oncePredicate, ^{
- _sharedManager = [[self alloc] init];
- });
- return _sharedManager;
- }
- + (NSSet *)keyPathsForValuesAffectingIsNetworkActivityIndicatorVisible {
- return [NSSet setWithObject:@"activityCount"];
- }
- - (id)init {
- self = [super init];
- if (!self) {
- return nil;
- }
- [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(networkRequestDidStart:) name:AFNetworkingOperationDidStartNotification object:nil];
- [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(networkRequestDidFinish:) name:AFNetworkingOperationDidFinishNotification object:nil];
- #if __IPHONE_OS_VERSION_MAX_ALLOWED >= 70000
- [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(networkRequestDidStart:) name:AFNetworkingTaskDidResumeNotification object:nil];
- [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(networkRequestDidFinish:) name:AFNetworkingTaskDidSuspendNotification object:nil];
- [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(networkRequestDidFinish:) name:AFNetworkingTaskDidCompleteNotification object:nil];
- #endif
- return self;
- }
- - (void)dealloc {
- [[NSNotificationCenter defaultCenter] removeObserver:self];
- [_activityIndicatorVisibilityTimer invalidate];
- }
- - (void)updateNetworkActivityIndicatorVisibilityDelayed {
- if (self.enabled) {
- // Delay hiding of activity indicator for a short interval, to avoid flickering
- if (![self isNetworkActivityIndicatorVisible]) {
- [self.activityIndicatorVisibilityTimer invalidate];
- self.activityIndicatorVisibilityTimer = [NSTimer timerWithTimeInterval:kAFNetworkActivityIndicatorInvisibilityDelay target:self selector:@selector(updateNetworkActivityIndicatorVisibility) userInfo:nil repeats:NO];
- [[NSRunLoop mainRunLoop] addTimer:self.activityIndicatorVisibilityTimer forMode:NSRunLoopCommonModes];
- } else {
- [self performSelectorOnMainThread:@selector(updateNetworkActivityIndicatorVisibility) withObject:nil waitUntilDone:NO modes:@[NSRunLoopCommonModes]];
- }
- }
- }
- - (BOOL)isNetworkActivityIndicatorVisible {
- return self.activityCount > 0;
- }
- - (void)updateNetworkActivityIndicatorVisibility {
- [[UIApplication sharedApplication] setNetworkActivityIndicatorVisible:[self isNetworkActivityIndicatorVisible]];
- }
- - (void)setActivityCount:(NSInteger)activityCount {
- @synchronized(self) {
- _activityCount = activityCount;
- }
- dispatch_async(dispatch_get_main_queue(), ^{
- [self updateNetworkActivityIndicatorVisibilityDelayed];
- });
- }
- - (void)incrementActivityCount {
- [self willChangeValueForKey:@"activityCount"];
- @synchronized(self) {
- _activityCount++;
- }
- [self didChangeValueForKey:@"activityCount"];
- dispatch_async(dispatch_get_main_queue(), ^{
- [self updateNetworkActivityIndicatorVisibilityDelayed];
- });
- }
- - (void)decrementActivityCount {
- [self willChangeValueForKey:@"activityCount"];
- @synchronized(self) {
- #pragma clang diagnostic push
- #pragma clang diagnostic ignored "-Wgnu"
- _activityCount = MAX(_activityCount - 1, 0);
- #pragma clang diagnostic pop
- }
- [self didChangeValueForKey:@"activityCount"];
- dispatch_async(dispatch_get_main_queue(), ^{
- [self updateNetworkActivityIndicatorVisibilityDelayed];
- });
- }
- - (void)networkRequestDidStart:(NSNotification *)notification {
- if ([AFNetworkRequestFromNotification(notification) URL]) {
- [self incrementActivityCount];
- }
- }
- - (void)networkRequestDidFinish:(NSNotification *)notification {
- if ([AFNetworkRequestFromNotification(notification) URL]) {
- [self decrementActivityCount];
- }
- }
- @end
- #endif
|