在我的应用程序中，在我的 API 类的 NSURLSession 中可以同时(异步)对我的服务器进行 2-4 个 API 调用。为了向我的服务器发出 API 请求，我必须在每个 NSURLRequest 的 HTTPHeaderField 中提供身份验证 token 。 token有效期为一天，如果过了一天就失效了，我需要刷新token。

我在 API 类的以下代码中执行此操作:

/*!
 * @brief sends a request as an NSHTTPURLResponse. This method is private.
 * @param request The request to send.
 * @param success A block to be called if the request is successful.
 * @param error A block to be called if the request fails.
 */
-(void)sendTask:(NSURLRequest*)request successCallback:(void (^)(NSDictionary*))success errorCallback:(void (^)(NSString*))errorCallback
{
    NSURLSessionDataTask *task = [self.session dataTaskWithRequest:request completionHandler:^(NSData *data, NSURLResponse *response, NSError *error)
    {
        [self parseResponse:response data:data fromRequest:request successCallback:success errorCallback:^(NSString *error)
        {
            //if auth token expired and getting "not authenticated" error (status 401)
            NSHTTPURLResponse *httpResp = (NSHTTPURLResponse*) response;
            if (httpResp.statusCode == 401) {
                [self refreshAuthenticationTokenWithSuccessCallback:^(NSDictionary *response) {
                    self.authToken = response[@"token"];
                    //attempt to re-try the request that failed due to token expiration
                    [self sendTask:request successCallback:success errorCallback:errorCallback];
                } errorCallback:^(NSString *error) {
                    //two weeks have passed and the token is no longer refreshable
                    NSLog(@"TOKEN NOT REFRESHABLE! HAVE TO LOG IN MANUALLY");
                }];
            }
        }];
    }];
    [task resume];
}

这个 sendTask 方法会随着我在应用程序中发出的每个 API 请求而执行，所以我才意识到这是一种糟糕的做法。如果 3 个 API 请求由于 token 无效而失败(一天过去了)，那么这 3 个 API 请求都将尝试进行 API 调用以刷新身份验证 token 。

如果其中一个 API 请求失败，我有没有办法只刷新一次身份验证 token ，然后重新尝试失败的 API 调用？

编辑

我编辑了问题的标题以表明我正在使用 NSURLSession

进展

到目前为止，为了防止多个失败的 API 请求同时尝试刷新身份验证 token ，我有一个用于所有失败请求的 NSArray 和一个 NSNumber用作锁以确保身份验证 token 仅尝试刷新一次。我在以下代码中执行此操作:

-(void)sendTask:(NSURLRequest*)request successCallback:(void (^)(NSDictionary*))success errorCallback:(void (^)(NSString*))errorCallback
{
    NSURLSessionDataTask *task = [self.session dataTaskWithRequest:request completionHandler:^(NSData *data, NSURLResponse *response, NSError *error)
    {
        MyAPIInterface *__weak weakSelf = self;
        [self parseResponse:response data:data fromRequest:request successCallback:success errorCallback:^(NSString *error)
        {
            NSHTTPURLResponse *httpResp = (NSHTTPURLResponse*) response;
            if (httpResp.statusCode == 401) {
                if ([error isEqualToString:@"invalid_credentials"]) {
                    errorCallback(@"Invalid username and/or password");
                }
                else if ([error isEqualToString:@"Unknown error"]) {
                    errorCallback(error);
                }
                else {
                    if (!weakSelf.alreadyRefreshingToken.boolValue) {

                        //lock alreadyRefreshingToken boolean
                        weakSelf.alreadyRefreshingToken = [NSNumber numberWithBool:YES];
                        NSLog(@"NOT REFRESHING TOKEN");

                        // add failed request to failedRequests array
                        NSMutableArray *mutableFailedRequests = [weakSelf.failedRequests mutableCopy];
                        [mutableFailedRequests addObject:request];
                        weakSelf.failedRequests = [mutableFailedRequests copy];

                        // refresh auth token
                        [weakSelf refreshAuthenticationTokenWithSuccessCallback:^(NSDictionary *response) {

                            //store authToken
                            weakSelf.authToken = response[@"token"];
                            NSUserDefaults* defaults = [NSUserDefaults standardUserDefaults];
                            [defaults setObject:weakSelf.authToken forKey:@"authToken"];
                            [defaults synchronize];

                            //attempt to re-try all requests that failed due to token expiration
                            for (NSURLRequest *failedRequest in weakSelf.failedRequests) {
                                [weakSelf sendTask:failedRequest successCallback:success errorCallback:errorCallback];
                            }

                            //clear failedRequests array and unlock alreadyRefreshingToken boolean
                            [weakSelf clearFailedRequests];
                            weakSelf.alreadyRefreshingToken = [NSNumber numberWithBool:NO];

                            NSLog(@"TOKEN REFRESHING SUCCESSFUL THO");

                        } errorCallback:^(NSString *error) {

                            NSLog(@"TOKEN NOT REFRESHABLE! HAVE TO LOG IN MANUALLY");

                            //clear failedRequests array
                            [weakSelf clearFailedRequests];

                            errorCallback(@"Your login session has expired");

                        }];
                    }
                    else  {
                        NSLog(@"ALREADY REFRESHING TOKEN. JUST ADD TO FAILED LIST");
                        NSMutableArray *mutableFailedRequests = [weakSelf.failedRequests mutableCopy];
                        [mutableFailedRequests addObject:request];
                        weakSelf.failedRequests = [mutableFailedRequests copy];
                    }
                }
            }
            else {
                NSLog(@"ERROR STRING THO: %@", error);
                errorCallback(error);
            }
        }];
    }];
    [task resume];
}

#pragma mark Custom Methods

-(void)clearFailedRequests {
    NSMutableArray *mutableFailedRequests = [self.failedRequests mutableCopy];
    [mutableFailedRequests removeAllObjects];
    self.failedRequests = [mutableFailedRequests copy];
}

我的做法是否正确？我偏执的一个部分是我并没有真正在某些点调用 success 或 error 回调。这会导致问题吗？

最佳答案

尝试使用 [weakSelf sendTask] 而不是 [self sendTask:]。检查以下代码:

-(void)sendTask:(NSURLRequest*)request successCallback:(void (^)(NSDictionary*))success errorCallback:(void (^)(NSString*))errorCallback
{

   __weak __typeof(self)weakSelf = self;
    NSURLSessionDataTask *task = [self.session dataTaskWithRequest:request completionHandler:^(NSData *data, NSURLResponse *response, NSError *error)
    {
        [self parseResponse:response data:data fromRequest:request successCallback:success errorCallback:^(NSString *error)
        {
            //if auth token expired and getting "not authenticated" error (status 401)
            NSHTTPURLResponse *httpResp = (NSHTTPURLResponse*) response;
            if (httpResp.statusCode == 401) {
                [self refreshAuthenticationTokenWithSuccessCallback:^(NSDictionary *response) {
                    self.authToken = response[@"token"];
                    //attempt to re-try the request that failed due to token expiration
                    [weakSelf sendTask:request successCallback:success errorCallback:errorCallback];
                } errorCallback:^(NSString *error) {
                    //two weeks have passed and the token is no longer refreshable
                    NSLog(@"TOKEN NOT REFRESHABLE! HAVE TO LOG IN MANUALLY");
                }];
            }
        }];
    }];
    [task resume];
}

关于iOS - 在 NSURLSession 中重新尝试失败的 NSURLRequests，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/34823840/