511 lines
18 KiB
Swift
511 lines
18 KiB
Swift
import Foundation
|
||
|
||
// 添加登出通知
|
||
extension Notification.Name {
|
||
static let userDidLogoutNotification = Notification.Name("UserDidLogoutNotification")
|
||
}
|
||
|
||
// 请求标识符
|
||
private struct RequestIdentifier {
|
||
static var currentId: Int = 0
|
||
static var lock = NSLock()
|
||
|
||
static func next() -> Int {
|
||
lock.lock()
|
||
defer { lock.unlock() }
|
||
currentId += 1
|
||
return currentId
|
||
}
|
||
}
|
||
|
||
public protocol NetworkServiceProtocol {
|
||
func postWithToken<T: Decodable>(
|
||
path: String,
|
||
parameters: [String: Any],
|
||
completion: @escaping (Result<T, NetworkError>) -> Void
|
||
)
|
||
|
||
@discardableResult
|
||
func upload(
|
||
request: URLRequest,
|
||
fileData: Data,
|
||
onProgress: @escaping (Double) -> Void,
|
||
completion: @escaping (Result<(Data?, URLResponse), Error>) -> Void
|
||
) -> URLSessionUploadTask?
|
||
}
|
||
|
||
extension NetworkService: NetworkServiceProtocol {
|
||
public func postWithToken<T: Decodable>(
|
||
path: String,
|
||
parameters: [String: Any],
|
||
completion: @escaping (Result<T, NetworkError>) -> Void
|
||
) {
|
||
var headers = [String: String]()
|
||
if let token = KeychainHelper.getAccessToken() {
|
||
headers["Authorization"] = "Bearer \(token)"
|
||
}
|
||
|
||
post(path: path, parameters: parameters, headers: headers, completion: completion)
|
||
}
|
||
|
||
@discardableResult
|
||
public func upload(
|
||
request: URLRequest,
|
||
fileData: Data,
|
||
onProgress: @escaping (Double) -> Void,
|
||
completion: @escaping (Result<(Data?, URLResponse), Error>) -> Void
|
||
) -> URLSessionUploadTask? {
|
||
var request = request
|
||
|
||
// Set content length header if not already set
|
||
if request.value(forHTTPHeaderField: "Content-Length") == nil {
|
||
request.setValue("\(fileData.count)", forHTTPHeaderField: "Content-Length")
|
||
}
|
||
|
||
var progressObserver: NSKeyValueObservation?
|
||
|
||
let task = URLSession.shared.uploadTask(with: request, from: fileData) { [weak self] data, response, error in
|
||
// Invalidate the progress observer when the task completes
|
||
progressObserver?.invalidate()
|
||
|
||
if let error = error {
|
||
completion(.failure(error))
|
||
return
|
||
}
|
||
|
||
guard let response = response else {
|
||
completion(.failure(NetworkError.invalidURL))
|
||
return
|
||
}
|
||
|
||
completion(.success((data, response)))
|
||
}
|
||
|
||
// Add progress tracking if available
|
||
if #available(iOS 11.0, *) {
|
||
progressObserver = task.progress.observe(\.fractionCompleted) { progressValue, _ in
|
||
DispatchQueue.main.async {
|
||
onProgress(progressValue.fractionCompleted)
|
||
}
|
||
}
|
||
}
|
||
|
||
task.resume()
|
||
return task
|
||
}
|
||
}
|
||
|
||
public enum NetworkError: Error {
|
||
case invalidURL
|
||
case noData
|
||
case decodingError(Error)
|
||
case serverError(String)
|
||
case unauthorized
|
||
case other(Error)
|
||
case networkError(Error)
|
||
case unknownError(Error)
|
||
case invalidParameters
|
||
|
||
public var localizedDescription: String {
|
||
switch self {
|
||
case .invalidURL:
|
||
return "无效的URL"
|
||
case .noData:
|
||
return "没有接收到数据"
|
||
case .decodingError(let error):
|
||
return "解码错误: \(error.localizedDescription)"
|
||
case .serverError(let message):
|
||
return "服务器错误: \(message)"
|
||
case .unauthorized:
|
||
return "未授权，请重新登录"
|
||
case .other(let error):
|
||
return error.localizedDescription
|
||
case .networkError(let error):
|
||
return "网络错误: \(error.localizedDescription)"
|
||
case .unknownError(let error):
|
||
return "未知错误: \(error.localizedDescription)"
|
||
case .invalidParameters:
|
||
return "无效的参数"
|
||
}
|
||
}
|
||
}
|
||
|
||
class NetworkService {
|
||
static let shared = NetworkService()
|
||
|
||
// 默认请求头
|
||
private let defaultHeaders: [String: String] = [
|
||
"Content-Type": "application/json",
|
||
"Accept": "application/json"
|
||
]
|
||
|
||
private var isRefreshing = false
|
||
private var requestsToRetry: [(URLRequest, (Result<Data, NetworkError>) -> Void, Int)] = []
|
||
|
||
private init() {}
|
||
|
||
// MARK: - 基础请求方法
|
||
private func request<T: Decodable>(
|
||
_ method: String,
|
||
path: String,
|
||
parameters: Any? = nil,
|
||
headers: [String: String]? = nil,
|
||
completion: @escaping (Result<T, NetworkError>) -> Void
|
||
) {
|
||
// 生成请求ID
|
||
let requestId = RequestIdentifier.next()
|
||
|
||
// 构建URL
|
||
let fullURL = APIConfig.baseURL + path
|
||
guard let url = URL(string: fullURL) else {
|
||
print("❌ [Network][#\(requestId)][\(method) \(path)] 无效的URL")
|
||
completion(.failure(.invalidURL))
|
||
return
|
||
}
|
||
|
||
// 创建请求
|
||
var request = URLRequest(url: url)
|
||
request.httpMethod = method
|
||
|
||
// 设置请求头 - 合并默认头、认证头和自定义头
|
||
defaultHeaders.forEach { key, value in
|
||
request.setValue(value, forHTTPHeaderField: key)
|
||
}
|
||
|
||
// 添加认证头 - 排除登录接口
|
||
if !path.contains("/iam/login/") {
|
||
APIConfig.authHeaders.forEach { key, value in
|
||
request.setValue(value, forHTTPHeaderField: key)
|
||
}
|
||
}
|
||
|
||
// 添加自定义头（如果提供）
|
||
headers?.forEach { key, value in
|
||
request.setValue(value, forHTTPHeaderField: key)
|
||
}
|
||
|
||
// 设置请求体（如果是POST/PUT请求）
|
||
if let parameters = parameters, (method == "POST" || method == "PUT") {
|
||
do {
|
||
if JSONSerialization.isValidJSONObject(parameters) {
|
||
request.httpBody = try JSONSerialization.data(withJSONObject: parameters)
|
||
} else {
|
||
print("❌ [Network][#\(requestId)][\(method) \(path)] 参数不是有效的JSON对象")
|
||
completion(.failure(.invalidParameters))
|
||
return
|
||
}
|
||
} catch {
|
||
print("❌ [Network][#\(requestId)][\(method) \(path)] 参数序列化失败: \(error.localizedDescription)")
|
||
completion(.failure(.other(error)))
|
||
return
|
||
}
|
||
}
|
||
|
||
// 打印请求信息
|
||
print("""
|
||
🌐 [Network][#\(requestId)][\(method) \(path)] 开始请求
|
||
🔗 URL: \(url.absoluteString)
|
||
📤 Headers: \(request.allHTTPHeaderFields ?? [:])
|
||
📦 Body: \(request.httpBody.flatMap { String(data: $0, encoding: .utf8) } ?? "")
|
||
""")
|
||
|
||
// 创建任务
|
||
let startTime = Date()
|
||
let task = URLSession.shared.dataTask(with: request) { [weak self] data, response, error in
|
||
let duration = String(format: "%.3fs", Date().timeIntervalSince(startTime))
|
||
|
||
// 处理响应
|
||
self?.handleResponse(
|
||
requestId: requestId,
|
||
method: method,
|
||
path: path,
|
||
data: data,
|
||
response: response,
|
||
error: error,
|
||
request: request,
|
||
duration: duration,
|
||
completion: { (result: Result<T, NetworkError>) in
|
||
completion(result)
|
||
}
|
||
)
|
||
}
|
||
|
||
// 开始请求
|
||
task.resume()
|
||
}
|
||
|
||
private func handleResponse<T: Decodable>(
|
||
requestId: Int,
|
||
method: String,
|
||
path: String,
|
||
data: Data?,
|
||
response: URLResponse?,
|
||
error: Error?,
|
||
request: URLRequest,
|
||
duration: String,
|
||
completion: @escaping (Result<T, NetworkError>) -> Void
|
||
) {
|
||
// 打印响应信息
|
||
if let httpResponse = response as? HTTPURLResponse {
|
||
let statusCode = httpResponse.statusCode
|
||
let statusMessage = HTTPURLResponse.localizedString(forStatusCode: statusCode)
|
||
|
||
// 处理401未授权
|
||
if statusCode == 401 {
|
||
print("""
|
||
🔑 [Network][#\(requestId)][\(method) \(path)] 检测到未授权，尝试刷新token...
|
||
⏱️ 耗时: \(duration)
|
||
""")
|
||
|
||
// 将请求加入重试队列
|
||
let dataResult = data.flatMap { Result<Data, NetworkError>.success($0) } ?? .failure(.noData)
|
||
self.requestsToRetry.append((request, { result in
|
||
switch result {
|
||
case .success(let data):
|
||
do {
|
||
let decoder = JSONDecoder()
|
||
let result = try decoder.decode(T.self, from: data)
|
||
print("""
|
||
✅ [Network][#\(requestId)][\(method) \(path)] 重试成功
|
||
⏱️ 总耗时: \(duration) (包含token刷新时间)
|
||
""")
|
||
completion(.success(result))
|
||
} catch let decodingError as DecodingError {
|
||
print("""
|
||
❌ [Network][#\(requestId)][\(method) \(path)] JSON解析失败
|
||
🔍 错误: \(decodingError.localizedDescription)
|
||
📦 原始数据: \(String(data: data, encoding: .utf8) ?? "")
|
||
""")
|
||
completion(.failure(.decodingError(decodingError)))
|
||
} catch {
|
||
print("""
|
||
❌ [Network][#\(requestId)][\(method) \(path)] 未知错误
|
||
🔍 错误: \(error.localizedDescription)
|
||
""")
|
||
completion(.failure(.unknownError(error)))
|
||
}
|
||
case .failure(let error):
|
||
print("""
|
||
❌ [Network][#\(requestId)][\(method) \(path)] 重试失败
|
||
🔍 错误: \(error.localizedDescription)
|
||
""")
|
||
completion(.failure(error))
|
||
}
|
||
}, requestId))
|
||
|
||
// 如果没有正在刷新的请求，则开始刷新token
|
||
if !isRefreshing {
|
||
refreshAndRetryRequests()
|
||
}
|
||
return
|
||
}
|
||
|
||
// 处理其他错误状态码
|
||
if !(200...299).contains(statusCode) {
|
||
let errorMessage = data.flatMap { String(data: $0, encoding: .utf8) } ?? ""
|
||
print("""
|
||
❌ [Network][#\(requestId)][\(method) \(path)] 请求失败
|
||
📊 状态码: \(statusCode) (\(statusMessage))
|
||
⏱️ 耗时: \(duration)
|
||
🔍 错误响应: \(errorMessage)
|
||
""")
|
||
completion(.failure(.serverError("状态码: \(statusCode), 响应: \(errorMessage)")))
|
||
return
|
||
}
|
||
|
||
// 成功响应
|
||
print("""
|
||
✅ [Network][#\(requestId)][\(method) \(path)] 请求成功
|
||
📊 状态码: \(statusCode) (\(statusMessage))
|
||
⏱️ 耗时: \(duration)
|
||
""")
|
||
}
|
||
|
||
// 处理网络错误
|
||
if let error = error {
|
||
print("""
|
||
❌ [Network][#\(requestId)][\(method) \(path)] 网络请求失败
|
||
⏱️ 耗时: \(duration)
|
||
🔍 错误: \(error.localizedDescription)
|
||
""")
|
||
completion(.failure(.networkError(error)))
|
||
return
|
||
}
|
||
|
||
// 检查数据是否存在
|
||
guard let data = data else {
|
||
print("""
|
||
❌ [Network][#\(requestId)][\(method) \(path)] 没有收到数据
|
||
⏱️ 耗时: \(duration)
|
||
""")
|
||
completion(.failure(.noData))
|
||
return
|
||
}
|
||
|
||
// 打印响应数据（调试用）
|
||
if let responseString = String(data: data, encoding: .utf8) {
|
||
print("""
|
||
📥 [Network][#\(requestId)][\(method) \(path)] 响应数据:
|
||
\(responseString.prefix(1000))\(responseString.count > 1000 ? "..." : "")
|
||
""")
|
||
}
|
||
|
||
do {
|
||
// 解析JSON数据
|
||
let decoder = JSONDecoder()
|
||
let result = try decoder.decode(T.self, from: data)
|
||
completion(.success(result))
|
||
} catch let decodingError as DecodingError {
|
||
print("""
|
||
❌ [Network][#\(requestId)][\(method) \(path)] JSON解析失败
|
||
🔍 错误: \(decodingError.localizedDescription)
|
||
📦 原始数据: \(String(data: data, encoding: .utf8) ?? "")
|
||
""")
|
||
completion(.failure(.decodingError(decodingError)))
|
||
} catch {
|
||
print("""
|
||
❌ [Network][#\(requestId)][\(method) \(path)] 未知错误
|
||
🔍 错误: \(error.localizedDescription)
|
||
""")
|
||
completion(.failure(.unknownError(error)))
|
||
}
|
||
}
|
||
|
||
private func refreshAndRetryRequests() {
|
||
guard !isRefreshing else { return }
|
||
|
||
isRefreshing = true
|
||
let refreshStartTime = Date()
|
||
|
||
print("🔄 [Network] 开始刷新Token...")
|
||
|
||
TokenManager.shared.refreshToken { [weak self] success, _ in
|
||
guard let self = self else { return }
|
||
|
||
let refreshDuration = String(format: "%.3fs", Date().timeIntervalSince(refreshStartTime))
|
||
|
||
if success {
|
||
print("""
|
||
✅ [Network] Token刷新成功
|
||
⏱️ 耗时: \(refreshDuration)
|
||
🔄 准备重试\(self.requestsToRetry.count)个请求...
|
||
""")
|
||
|
||
// 重试所有待处理的请求
|
||
let requestsToRetry = self.requestsToRetry
|
||
self.requestsToRetry.removeAll()
|
||
|
||
for (request, completion, requestId) in requestsToRetry {
|
||
var newRequest = request
|
||
if let token = KeychainHelper.getAccessToken() {
|
||
newRequest.setValue("Bearer \(token)", forHTTPHeaderField: "Authorization")
|
||
}
|
||
|
||
let task = URLSession.shared.dataTask(with: newRequest) { data, response, error in
|
||
if let data = data {
|
||
completion(.success(data))
|
||
} else if let error = error {
|
||
completion(.failure(.networkError(error)))
|
||
} else {
|
||
completion(.failure(.noData))
|
||
}
|
||
}
|
||
task.resume()
|
||
}
|
||
} else {
|
||
print("""
|
||
❌ [Network] Token刷新失败
|
||
⏱️ 耗时: \(refreshDuration)
|
||
🚪 清除登录状态...
|
||
""")
|
||
|
||
// 清除token并通知需要重新登录
|
||
TokenManager.shared.clearTokens()
|
||
DispatchQueue.main.async {
|
||
NotificationCenter.default.post(name: .userDidLogoutNotification, object: nil)
|
||
}
|
||
|
||
// 所有待处理的请求都返回未授权错误
|
||
self.requestsToRetry.forEach { _, completion, _ in
|
||
completion(.failure(.unauthorized))
|
||
}
|
||
self.requestsToRetry.removeAll()
|
||
}
|
||
|
||
self.isRefreshing = false
|
||
}
|
||
}
|
||
|
||
// MARK: - 公共方法
|
||
|
||
/// GET 请求
|
||
func get<T: Decodable>(
|
||
path: String,
|
||
parameters: [String: Any]? = nil,
|
||
headers: [String: String]? = nil,
|
||
completion: @escaping (Result<T, NetworkError>) -> Void
|
||
) {
|
||
request("GET", path: path, parameters: parameters, headers: headers, completion: completion)
|
||
}
|
||
|
||
/// POST 请求
|
||
func post<T: Decodable>(
|
||
path: String,
|
||
parameters: Any? = nil,
|
||
headers: [String: String]? = nil,
|
||
completion: @escaping (Result<T, NetworkError>) -> Void
|
||
) {
|
||
var params: Any?
|
||
if let parameters = parameters {
|
||
if let dict = parameters as? [String: Any] {
|
||
params = dict
|
||
} else if let array = parameters as? [Any] {
|
||
params = array
|
||
} else {
|
||
print("❌ [Network] POST 请求参数类型不支持")
|
||
completion(.failure(.invalidParameters))
|
||
return
|
||
}
|
||
}
|
||
request("POST", path: path, parameters: params, headers: headers, completion: completion)
|
||
}
|
||
|
||
/// POST 请求（带Token）
|
||
func postWithToken<T: Decodable>(
|
||
path: String,
|
||
parameters: Any? = nil,
|
||
headers: [String: String]? = nil,
|
||
completion: @escaping (Result<T, NetworkError>) -> Void
|
||
) {
|
||
var headers = headers ?? [:]
|
||
if let token = KeychainHelper.getAccessToken() {
|
||
headers["Authorization"] = "Bearer \(token)"
|
||
}
|
||
post(path: path, parameters: parameters, headers: headers, completion: completion)
|
||
}
|
||
|
||
/// DELETE 请求
|
||
func delete<T: Decodable>(
|
||
path: String,
|
||
parameters: [String: Any]? = nil,
|
||
headers: [String: String]? = nil,
|
||
completion: @escaping (Result<T, NetworkError>) -> Void
|
||
) {
|
||
var headers = headers ?? [:]
|
||
if let token = KeychainHelper.getAccessToken() {
|
||
headers["Authorization"] = "Bearer \(token)"
|
||
}
|
||
request("DELETE", path: path, parameters: parameters, headers: headers, completion: completion)
|
||
}
|
||
|
||
/// PUT 请求
|
||
func put<T: Decodable>(
|
||
path: String,
|
||
parameters: [String: Any]? = nil,
|
||
headers: [String: String]? = nil,
|
||
completion: @escaping (Result<T, NetworkError>) -> Void
|
||
) {
|
||
request("PUT", path: path, parameters: parameters, headers: headers, completion: completion)
|
||
}
|
||
}
|