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