Swift开发中请求URL获取数据时是否可以添加自定义请求头？

API 探路者

2026-10-6

解答

你当前使用的Data(contentsOf:)便捷接口不支持自定义请求头，该接口仅用于简单的资源快速读取场景，无法配置请求头、超时时间、缓存策略等自定义参数，且你当前写法直接强制解包URL、未做错误兜底，线上运行有崩溃风险。

推荐实现方案

改用URLSession发起请求，可自由配置所有请求参数，包括自定义请求头，参考实现如下：

static private func createPDFLocalURL(urlString:String, fileName:String) -> FileManagementResponse {
    // 先校验URL合法性，避免强制解包崩溃
    guard let url = URL(string: urlString) else {
        // 此处返回你自定义的错误状态即可
        return .failed
    }
    // 创建可变请求对象
    var request = URLRequest(url: url)
    // 按需添加自定义请求头，示例如下
    request.setValue("自定义User-Agent", forHTTPHeaderField: "User-Agent")
    request.setValue("application/pdf", forHTTPHeaderField: "Accept")
    // 如需鉴权可添加Authorization头
    // request.setValue("Bearer 你的鉴权token", forHTTPHeaderField: "Authorization")
    
    // 若要兼容你原有的同步逻辑，可通过信号量实现（注意不要在主线程调用，避免阻塞UI）
    var pdfData: Data?
    var requestError: Error?
    let semaphore = DispatchSemaphore(value: 0)
    let task = URLSession.shared.dataTask(with: request) { data, _, error in
        pdfData = data
        requestError = error
        semaphore.signal()
    }
    task.resume()
    // 可根据需求调整请求超时时长
    _ = semaphore.wait(timeout: .now() + 30)
    
    if let error = requestError {
        // 处理请求错误，返回对应状态
        return .failed
    }
    guard let data = pdfData else {
        return .failed
    }
    // 此处保留你原有将PDF数据写入本地、返回对应状态的逻辑
    // ...
}