iOS实现OAuth2.0中刷新access token并从新请求数据操做

1、简要概述

　　OAuth2.0是OAuth协议的下一版本，时经常使用于移动客户端的开发，是一种比较安全的机制。在OAuth 2.0中，server将发行一个短有效期的access token和长生命期的refresh token。这将容许客户端无需用户再次操做而获取一个新的access token，而且也限制了access token的有效期。即当sever发送的access token过时以后，客户端会调用方法，将access token和refresh token发送给服务端，服务端将会返回新的access token和refresh token。

2、应用场景

　　用户登陆后，服务端会发行一个有效时间的access token，同时也会发行一个长生命期的refresh token。用户在进行其余网络请求时，会把access token加入请求体中（并不须要加入refresh token）。若是在请求过程当中，access token过时，返回相应的状态码。这时就回调用一个回调方法，在回调方法体中将原来的access token和refresh token发送给服务端，获取新的access token和refresh token。而后把新的access token加入刚才的请求体中，从新加载网络请求。

　　限定access token的有效时间，只是为了提升安全性。access token过时后从新获取并从新加载请求这一操做，用户是察觉不到的。这种机制在微信、QQ、微博等客户端中尤其常见。

3、实例代码

　　（1）此类继承了AFNetworking中的AFHTTPSessionManager类，并重写了里面的方法。

　　NetWorkCallBack.h：

　　

1 #import <Foundation/Foundation.h>
2 #import <AFNetworking.h>
3 @interface NetWorkCallBack : AFHTTPSessionManager 4 
5 @end

　　NetWorkCallBack.m：

　　

 1 #import "NetWorkCallBack.h"
 2 #import <SSKeychain.h>
 3 @implementation NetWorkCallBack  4 
 5 - (NSURLSessionDataTask *)dataTaskWithRequest:(NSMutableURLRequest *)urlRequest completionHandler:(void (^)(NSURLResponse *response, id responseObject, NSError *error))originalCompletionHandler{  6     
 7     //create a completion block that wraps the original
 8     void (^authFailBlock)(NSURLResponse *response, id responseObject, NSError *error) = ^(NSURLResponse *response, id responseObject, NSError *error)  9  { 10         NSHTTPURLResponse* httpResponse = (NSHTTPURLResponse*)response; 11         if([httpResponse statusCode] == 401){ 12             
13             //若是access token过时，返回错误，调用此block
14             dispatch_async(dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_LOW, 0), ^{ 15                 
16                 //调用refreshAccesstoken方法，刷新access token。
17                 [self refreshAccessToken:^(AFHTTPRequestOperation *operation) { 18                     //存取新的access token，此处我使用了KeyChain存取
19                     NSDictionary *headerInfo = operation.response.allHeaderFields; 20                     NSString *newAccessToken = [headerInfo objectForKey:@"access-token"]; 21                     NSString *newRefreshToken = [headerInfo objectForKey:@"refresh-token"]; 22                     [SSKeychain deletePasswordForService:@"<key>" account:@"access-token"]; 23                     [SSKeychain deletePasswordForService:@"<key>" account:@"refresh-token"]; 24                     [SSKeychain setPassword:newAccessToken forService:@"<key>" account:@"access-token"]; 25                     [SSKeychain setPassword:newRefreshToken forService:@"<key>" account:@"refresh-token"]; 26                     
27                     //将新的access token加入到原来的请求体中，从新发送请求。
28                     [urlRequest setValue:newAccessToken forHTTPHeaderField:@"access-token"]; 29                     
30                     NSURLSessionDataTask *originalTask = [super dataTaskWithRequest:urlRequest completionHandler:originalCompletionHandler]; 31  [originalTask resume]; 32  }]; 33  }); 34         }else{ 35             NSLog(@"no auth error"); 36  originalCompletionHandler(response, responseObject, error); 37  } 38  }; 39 
40     NSURLSessionDataTask *stask = [super dataTaskWithRequest:urlRequest completionHandler:authFailBlock]; 41     
42     return stask; 43     
44 }; 45 
46 
47 /*
48  *获取新的token的方法。如何获取能够自定义，我这里用了AFNetWorking的AFHTTPRequestOperation类 49  */
50 -(void)refreshAccessToken:(void(^)(AFHTTPRequestOperation *responseObject))refresh{ 51     NSMutableURLRequest *request = [[NSMutableURLRequest alloc] initWithURL:@"<yourURL>"]; 52     
53     [request setValue:@"application/json" forHTTPHeaderField:@"Content-Type"]; 54     
55     //将原来的access token和refresh token发送给服务器，以获取新的token
56     NSString *accessToken = [SSKeychain passwordForService:@"<key>" account:@"access-token"]; 57     NSString *refreshToken = [SSKeychain passwordForService:@"<key>" account:@"refresh-token"]; 58     
59     [request setValue:accessToken forHTTPHeaderField:@"access-token"]; 60     [request setValue:refreshToken forHTTPHeaderField:@"refresh-token"]; 61     
62     //执行网络方法
63     AFHTTPRequestOperation *httpRequestOperation = [[AFHTTPRequestOperation alloc] initWithRequest:request]; 64     [httpRequestOperation setCompletionBlockWithSuccess:^(AFHTTPRequestOperation * operation, id responseObject) { 65  refresh(operation); 66     } failure:^(AFHTTPRequestOperation * operation, NSError * error) { 67  refresh(operation); 68  }]; 69  [httpRequestOperation start]; 70 } 71 @end