NSURLProtocol在自定義網絡請求處理中的使用示例

NSURLProtocol是一個用于攔截和處理網絡請求的類，可以實現自定義的網絡請求處理邏輯。下面是一個使用NSURLProtocol的示例，實現對網絡請求的攔截和修改：

1. 創建一個自定義的NSURLProtocol子類，例如CustomURLProtocol：

import Foundation

class CustomURLProtocol: NSURLProtocol {
    
    override class func canInit(with request: URLRequest) -> Bool {
        // 在這里判斷是否要攔截當前的請求
        return true // 攔截所有請求
    }
    
    override class func canonicalRequest(for request: URLRequest) -> URLRequest {
        return request
    }
    
    override func startLoading() {
        let newRequest = self.request // 獲取原始請求
        
        // 在這里可以對請求進行修改，例如修改請求頭
        let modifiedRequest = newRequest
        
        let session = URLSession(configuration: .default, delegate: self, delegateQueue: nil)
        let task = session.dataTask(with: modifiedRequest)
        task.resume()
    }
    
    override func stopLoading() {
        // 停止加載
    }
}

extension CustomURLProtocol: URLSessionDataDelegate {
    
    func urlSession(_ session: URLSession, dataTask: URLSessionDataTask, didReceive data: Data) {
        // 處理接收到的數據
        self.client?.urlProtocol(self, didLoad: data)
    }
    
    func urlSession(_ session: URLSession, task: URLSessionTask, didCompleteWithError error: Error?) {
        // 請求完成或出錯處理
        self.client?.urlProtocolDidFinishLoading(self)
    }
}

2. 注冊CustomURLProtocol類，以便在應用啟動時能夠使用該類來處理網絡請求：

URLProtocol.registerClass(CustomURLProtocol.self)

通過以上示例，我們可以自定義處理網絡請求，例如可以在startLoading方法中修改請求頭，處理請求返回的數據等。需要注意的是，在使用完自定義的NSURLProtocol后，記得調用URLProtocol.unregisterClass方法來取消注冊。