IOS拦截重定向请求(302)的几种方式

By | 2015年7月12日
去评论本文

前言

在多数情况下，我们做的网络请求是返回200状态码的，但也有返回302的时候，比如使用基于Oauth2认证协议的API时，在认证阶段，需要提供一个回调地址，当用户授权后，服务器会返回一个302 Response，Response Header中会一个Location字段，包含了我们的回调地址，同时会有一个Code参数。我们在程序中该如何处理这个请求，并拿到这个Code参数呢。下面由我来为大家讲解下几种方式的做法，各取所需。

假设您知道并使用过Oauth2认证协议

（一）UIWebView控件

这是最常见的做法，但是UIWebView是无法拦截302请求的，只能等待整个流程完成回到回调地址时，我们在webView控件的webViewDidFinishLoad回调方法处理数据。

首先，我们需要让ViewController类继承UIWebViewDelegate协议，然后实现webViewDidFinishLoad方法： 

class WebLoginViewController: UIViewController,UIWebViewDelegate {

    @IBOutlet var webView: UIWebView!

    override func viewDidLoad() {
        super.viewDidLoad()

        webView.scalesPageToFit = true
        webView.delegate = self
    }

    override func didReceiveMemoryWarning() {
        super.didReceiveMemoryWarning()
    }

    func webViewDidFinishLoad(webView: UIWebView) {
        //处理数据
    }
}

接着在启动时给webview一个加载地址，先载入指定的登陆页面：

override func viewDidLoad() {
    super.viewDidLoad()
    webView.scalesPageToFit = true
    webView.delegate = self

    let url = "https://www.oschina.net/xxxxxx"
    //程序启动后，让webview加载 OSChina的验证登陆界面
    webView.loadRequest(NSURLRequest(URL: NSURL(string: url)!))
}

当整个请求链完成后，我们在DidFinishLoad中通过判断请求的url，来确认是否已经回到了回调地址上 

func webViewDidFinishLoad(webView: UIWebView) {

    var url = webView.request?.URL!.absoluteString

    if url!.hasPrefix("回调地址url")
    {
        //从一个url字符串中拿到Code值
        let code = url!.GetCodeL()
        println("code = \(code)")

        //拿到Code后，可以开始请求Token了
    }
}

很显然，这种方法还需要等待webView来处理回调地址的请求，而这个请求对我们的程序来说是完全没有必要的。

我们要做的是拦截 302！

（二）基于NSURLConnection来设置拦截

在很多教程中都提到了NSURLConnection，它可以发送一个请求，比如： 

let request = NSURLRequest(URL: NSURL(string: "http://devonios.com")!)
NSURLConnection.sendAsynchronousRequest(request, queue: NSOperationQueue()) { (response, data,
error) -> Void in
    //处理返回数据
}

如果要发送POST的话，需要使用可编辑的NSMutableURLRequest类（它是继承NSURLRequest类的）。

我们需要的拦截效果，其实就是要给NSURLConnection设置一个delegate，提供一个事件发生时的回调方法。

NSURLConnection类有一个构造函数： 

init?(request request: NSURLRequest, delegate delegate: AnyObject?)

第二个参数就是我们需要设置的delegate。对应的delegate是：NSURLConnectionDataDelegate。

我们在Dash中可以看到它有这些东西：

开始写代码了: 

class LoginViewController: UIViewController,NSURLConnectionDataDelegate {
    func connection(){
        //创建一个可以编辑的NSURLRequest
        var mutableRequest = NSMutableURLRequest(URL: NSURL(string: "http://devonios.com")!)
        mutableRequest.HTTPMethod = "POST"
        //设置POST请求的表单数据
        mutableRequest.HTTPBody = paramString.dataUsingEncoding(NSStringEncoding.allZeros, allowLossyConversion: true)
        //使用构造函数方法创建一个NSURLConnection的实例
        var connection:NSURLConnection = NSURLConnection(request: mutableRequest, delegate: self)!
        connection.start()
    }
    //处理重定向请求的方法
    func connection(connection: NSURLConnection, willSendRequest request: NSURLRequest, redirectResponse response: NSURLResponse?) -> NSURLRequest? {

        if let r = response{

            //当前重定向请求的url，包含了Code参数
            let requesturl = request.URLString

            //得到Code，由于Code参数设置了属性观察器，所以当Code被赋值时，会自动去获取Token
            self.code = requesturl.GetCode()

            //因为已经拿到Code了，所以拦截掉当前这个重定向请求，直接返回nil
            return nil
        }
        return request
    }
    //整个请求完成后，即拦截到302后，不再请求了就返回这里
    func connection(connection: NSURLConnection, didReceiveResponse response: NSURLResponse) {
        if (某些判断条件){
            self.navigationController?.popViewControllerAnimated(true)
        }
    }
}

（三）基于NSURLSession类来设置拦截

NSURLSession是IOS 7中开始出现的全新的网络接口类，和NSURLConnection类似，同样需要设置delegate。 

class MyRequestController:NSObject,NSURLSessionTaskDelegate {

    let session:NSURLSession?

    init(){
        let sessionConfig = NSURLSessionConfiguration.defaultSessionConfiguration()
        session = NSURLSession(configuration: sessionConfig, delegate: self, delegateQueue: nil)
    }

    deinit{
        session!.invalidateAndCancel()
    }
    //处理重定向请求，直接使用nil来取消重定向请求
    func URLSession(session: NSURLSession, task: NSURLSessionTask, willPerformHTTPRedirection response: NSHTTPURLResponse, newRequest request: NSURLRequest, completionHandler: (NSURLRequest!) -> Void) {
        completionHandler(nil)
    }

    func sendRequest() {

        var URL = NSURL(string: "http://devonios.com")
        let request = NSMutableURLRequest(URL: URL!)
        request.HTTPMethod = "POST"

        request.HTTPBody = paramString.dataUsingEncoding(NSStringEncoding.allZeros, allowLossyConversion: true)

        let task = session!.dataTaskWithRequest(request, completionHandler: { (data : NSData!, response : NSURLResponse!, error : NSError!) -> Void in
            //由于拦截了302，设置了completionHandler参数为nil，所以忽略了重定向请求，这里返回的Response就是包含302状态码的Response了。
            let resp:NSHTTPURLResponse = response as! NSHTTPURLResponse
            println("包含302状态的Response Header字段 ： \(resp.allHeaderFields)")  })
            task.resume()
    }
}

目前为止，我们通过为NSURLConnection或者NSURLSession设置一个Delegate，通过回调方法来拦截（其实就是返回个nil）。

但是在一个项目中，我们通常会使用Alamofire这种第三库来操作网络请求，我要是再自己再重新写个请求，那岂不是很麻烦？

（四）完善Alamofire库，实现拦截302请求

Alamofire啥就不多说了，分析它的代码可以发现，是使用NSURLSession来实现请求的。

既然如此，那么我们就要找到NSURLSession，为它设置delegate，然后重写willPerformHttpRedirection。

在Alamofire.swift文件中，request方法是暴露给我们调用的，Manager类的sharedInstance属性来管理自身对象。 

public func request(method: Method, URLString: URLStringConvertible, parameters: [String: AnyObject]? = nil, encoding: ParameterEncoding = .URL) -> Request {
    return Manager.sharedInstance.request(method, URLString, parameters: parameters, encoding: encoding)
}

Manager.sharedInstance属性的实现，定义了请求头信息，然后调用构造函数 

 public static let sharedInstance: Manager = {
     let configuration: NSURLSessionConfiguration = NSURLSessionConfiguration.defaultSessionConfiguration()
     configuration.HTTPAdditionalHeaders = Manager.defaultHTTPHeaders
     return Manager(configuration: configuration)
 }()

构造函数，我们要找的NSURLSession就在这里，它默认已经有了一个Class（SessionDelegate）来实现相应的delegate了：

required public init(configuration: NSURLSessionConfiguration? = nil) {
    self.delegate = SessionDelegate()
    self.session = NSURLSession(configuration: configuration, delegate: delegate, delegateQueue:
nil)
    self.delegate.sessionDidFinishEventsForBackgroundURLSession = { [weak self] session in
        if let strongSelf = self {
            strongSelf.backgroundCompletionHandler?()
        }
    }
}

这个构造函数看上去动不了什么，关键还在SessionDelegate类，它实现了所有了NSURLSessionDelegate：

public final class SessionDelegate: NSObject, NSURLSessionDelegate, NSURLSessionTaskDelegate, NSURLSessionDataDelegate, NSURLSessionDownloadDelegate {
    public var taskWillPerformHTTPRedirection: ((NSURLSession, NSURLSessionTask, NSHTTPURLResponse,NSURLRequest) -> NSURLRequest?)?
    public func URLSession(session: NSURLSession, task: NSURLSessionTask, willPerformHTTPRedirection response: NSHTTPURLResponse, newRequest request: NSURLRequest, completionHandler: ((NSURLRequest!) -> Void)) {
     var redirectRequest: NSURLRequest? = request
     if taskWillPerformHTTPRedirection != nil {
         redirectRequest = taskWillPerformHTTPRedirection!(session, task, response, request)
     }
     completionHandler(redirectRequest)
 }
}

仔细观察会发现，有一个public的变量（var）taskWillPerformHTTPRedirection、有一个重写方法（willperformHTTPRedirection）。

从这个方法中可以看出，它期望我们给taskWillPerformHTTPRedirection变量传一个自定义方法，如果我们赋值了，它就运行我们的自定义方法。

我们要给taskWillPerformHTTPRedirection变量赋值，参数是一个方法。

在Manager类中加入下面代码： 

public typealias TaskWillRedirectAction = ((NSURLSession, NSURLSessionTask, NSHTTPURLResponse,NSURLRequest) -> NSURLRequest?)
public func setTaskWillRedirectAction(action:TaskWillRedirectAction){
    self.delegate.taskWillPerformHTTPRedirection = action
}

对Alamofire库的修改就这样可以了！

我们需要在发送网络请求前，先调用setTaskWillRedirectAction方法，传入我们的自定义方法。

使用方法： 

var manager = Manager.sharedInstance
manager.setTaskWillRedirectAction { (session, task, response, request) -> NSURLRequest? in
    return nil
}
manager.request(Method.POST, url, parameters: authparam.toDictionary(), encoding: ParameterEncoding.URL).response { (request, response, data, err) -> Void in
    //由于上面的setTaskWillRedirectAction方法返回nil，所以在处理NSURLSessionDataDelegate的重写方法时，complectionHandler方法参数为nil，也就实现了拦截！
    println(response?.allHeaderFields["Location"])
}

注意，这里需要先从sharedInstance属性中拿到一个Manager对象，然后再用这个对象设置拦截的回调方法，再发送请求。

如果您还是使用Alamofire.request来发送请求的话，就没有作用了，因为你又重新创建了个Manager类对象。

参考资料

https://developer.apple.com/library/ios/documentation/Cocoa/Conceptual/URLLoadingSystem/Articles/RequestChanges.html

http://stackoverflow.com/questions/1446509/handling-redirects-correctly-with-nsurlconnection