如何在Swift环境下为Firebase Firestore生成自定义且唯一的文档ID？

好的，我来帮你搞定这个问题——要生成location1、location2这种格式的唯一自定义ID，同时保证在Firestore中不重复，核心是要原子性地生成序号，避免多请求并发时的冲突。下面结合Swift代码给你两种可行方案：

这种方式通过一个独立的计数器文档来记录当前最大的序号，再用Firestore事务保证“读取计数器→生成新ID→更新计数器→写入数据”这一系列操作的原子性，从根本上避免ID重复。

具体步骤： #

1. 先在Firestore里创建一个counters集合，添加一个名为locationCounter的文档，里面存一个count字段，初始值设为0（也可以在代码里判断如果文档不存在就自动创建）。
2. 用事务读取这个计数器，计算新序号，生成自定义ID，同时更新计数器，最后写入数据。

Swift代码示例： #

import FirebaseFirestore

let dbLocations = Firestore.firestore()
let yourData: [String: Any] = ["name": "My Location", "coords": [39.9, 116.3]] // 替换成你的实际数据

// 执行Firestore事务
dbLocations.runTransaction({ (transaction, errorPointer) -> Any? in
    // 指向计数器文档的引用
    let counterRef = dbLocations.collection("counters").document("locationCounter")
    
    // 读取计数器文档
    guard let counterDoc = try? transaction.getDocument(counterRef) else {
        errorPointer?.pointee = NSError(
            domain: "FirestoreError",
            code: -1,
            userInfo: [NSLocalizedDescriptionKey: "Failed to fetch counter document"]
        )
        return nil
    }
    
    // 获取当前计数，默认0（如果文档刚创建还没数据）
    let currentCount = counterDoc.data()?["count"] as? Int ?? 0
    let newCount = currentCount + 1
    let newCustomID = "location\(newCount)"
    
    // 事务内更新计数器
    transaction.setData(["count": newCount], forDocument: counterRef)
    
    // 事务内写入数据到自定义ID的文档
    let newDocRef = dbLocations.collection("Data").document(newCustomID)
    transaction.setData(yourData, forDocument: newDocRef)
    
    return newCustomID
}) { (result, error) in
    if let error = error {
        print("Transaction failed: \(error.localizedDescription)")
    } else if let newID = result as? String {
        print("Successfully created document with custom ID: \(newID)")
    }
}

如果你的应用并发请求很少，也可以直接查询所有以location开头的文档，提取序号找到最大值，再加1生成新ID。但要注意：这种方式在高并发下可能出现多个请求同时查到相同的最大值，导致ID重复，所以仅推荐低流量场景使用。

Swift代码示例： #

import FirebaseFirestore

let dbLocations = Firestore.firestore()
let yourData: [String: Any] = ["name": "My Location", "coords": [39.9, 116.3]] // 替换成你的实际数据

// 查询所有以"location"开头的文档（用documentID范围查询优化性能）
dbLocations.collection("Data")
    .whereField(FieldPath.documentID(), isGreaterThanOrEqualTo: "location")
    .whereField(FieldPath.documentID(), isLessThan: "locationz")
    .getDocuments { (querySnapshot, error) in
        if let error = error {
            print("Error fetching existing documents: \(error)")
            return
        }
        
        var maxSeqNumber = 0
        querySnapshot?.documents.forEach { doc in
            // 提取ID里的数字部分
            if let number = Int(doc.documentID.replacingOccurrences(of: "location", with: "")) {
                maxSeqNumber = max(maxSeqNumber, number)
            }
        }
        
        // 生成新的自定义ID
        let newCustomID = "location\(maxSeqNumber + 1)"
        
        // 写入数据
        dbLocations.collection("Data").document(newCustomID).setData(yourData) { error in
            if let error = error {
                print("Error writing document: \(error)")
            } else {
                print("Document created with ID: \(newCustomID)")
            }
        }
    }

额外说明： #

• 如果担心删除文档后序号断层（比如删了location2，新生成的是location3而不是补location2），一般来说不需要额外处理——ID的唯一性才是关键，序号连续与否对大多数场景不重要。如果一定要连续，那需要更复杂的逻辑（比如维护可用序号列表），但通常没必要。
• 事务方案是Firestore官方推荐的处理这类原子性操作的方式，能完美应对多用户同时创建文档的场景。

内容的提问来源于stack exchange，提问作者doxsi