iOS token刷新机制

原创

mob64ca12f1c6f8 2024-10-10 05:21:46 ©著作权

©著作权归作者所有：来自51CTO博客作者mob64ca12f1c6f8的原创作品，请联系作者获取转载授权，否则将追究法律责任

iOS Token 刷新机制

在现代移动应用中，安全性和用户体验至关重要。为此，许多平台采用了以 Token 为基础的身份验证机制。iOS 应用通常会使用这种机制来处理用户的登录状态。本文将深入探讨 iOS 中的 Token 刷新机制，并通过代码示例和状态图帮助大家更好地理解这一过程。

什么是 Token

Token 是一种包含用户身份信息的字符串，通常在用户成功登录后由服务器生成并返回给客户端。Token 可以存储在用户的设备上，用于后续的身份验证。如果 Token 过期或即将过期，客户端可以通过调用刷新接口来获取一个新的 Token。

Token 刷新机制

在 iOS 应用中，通常采用 JWT（JSON Web Token）或 OAuth 2.0 作为 Token 的实现方式。Token 刷新机制通常包括以下几个步骤：

用户登录：用户输入凭据后，应用向服务器发送登录请求，服务器验证凭据并返回 Token 和刷新 Token。
Token 使用：应用使用 Token 访问受保护的资源。
Token 过期：Token 会在一段时间后过期，通常在用户有一段时间不活动后。
Token 刷新：当应用检测到 Token 已过期时，会使用刷新 Token 向服务器请求新的 Token。
重新使用：获取新的 Token 后，应用继续使用。

状态图

以下是 Token 刷新机制的状态图，用于展示各个状态之间的转换关系：

stateDiagram
    [*] --> login
    login --> token_received : successful login
    token_received --> token_used : use token
    token_used --> token_expired : token expired
    token_expired --> token_refreshed : refresh token
    token_refreshed --> token_used
    token_used --> [*]

实现代码示例

下面的代码示例展示了如何在 iOS 中实现 Token 刷新机制：

1. 登录并获取 Token

import Foundation

struct AuthResponse: Codable {
    let token: String
    let refreshToken: String
}

func login(username: String, password: String, completion: @escaping (AuthResponse?) -> Void) {
    guard let url = URL(string: " else { return }
    
    var request = URLRequest(url: url)
    request.httpMethod = "POST"
    let bodyData = ["username": username, "password": password]
    request.httpBody = try? JSONSerialization.data(withJSONObject: bodyData)
    request.setValue("application/json", forHTTPHeaderField: "Content-Type")
    
    let task = URLSession.shared.dataTask(with: request) { data, response, error in
        guard let data = data, error == nil else {
            print("Error: \(error?.localizedDescription ?? "Unknown error")")
            completion(nil)
            return
        }
        
        let responseModel = try? JSONDecoder().decode(AuthResponse.self, from: data)
        completion(responseModel)
    }
    task.resume()
}

2. 刷新 Token

struct RefreshResponse: Codable {
    let token: String
}

func refreshToken(refreshToken: String, completion: @escaping (RefreshResponse?) -> Void) {
    guard let url = URL(string: " else { return }
    
    var request = URLRequest(url: url)
    request.httpMethod = "POST"
    let bodyData = ["refreshToken": refreshToken]
    request.httpBody = try? JSONSerialization.data(withJSONObject: bodyData)
    request.setValue("application/json", forHTTPHeaderField: "Content-Type")
    
    let task = URLSession.shared.dataTask(with: request) { data, response, error in
        guard let data = data, error == nil else {
            print("Error: \(error?.localizedDescription ?? "Unknown error")")
            completion(nil)
            return
        }

        let responseModel = try? JSONDecoder().decode(RefreshResponse.self, from: data)
        completion(responseModel)
    }
    task.resume()
}

3. 使用 Token

在发送请求之前，应用需要检查 Token 是否过期，并决定是否需要刷新：

var authToken: String?
var refreshToken: String?

func makeApiRequest() {
    guard let token = authToken else {
        print("No token found.")
        return
    }

    // Mock function to check if token is expired
    if isTokenExpired(token) {
        refreshToken(refreshToken: refreshToken!) { newAuth in
            if let newToken = newAuth?.token {
                self.authToken = newToken
                // Proceed with API request
            }
        }
    } else {
        // Proceed with API request using the token
    }
}

func isTokenExpired(_ token: String) -> Bool {
    // Implement token expiration check logic
    return false // Placeholder
}