Swift中基于NSMutableDictionary实现RSVP首字母分组的问询

我来帮你完善并优化这段按首字母分组RSVP数据的代码，先解决你代码未完成的部分，再给出更安全、更符合Swift风格的实现方案。

补全你的原始代码（可运行版本） #

你的代码最后未完成追加数据的逻辑，这里帮你补全，同时修复了强制解包可能导致崩溃的问题：

let RSVPDirectory = NSMutableDictionary() 
for i in 0..<self.RSVPs.count { 
    let tmp = self.RSVPs[i] 
    var firstLetter: String
    
    // 优先处理姓氏字段，避免强制解包
    if let lastName = tmp["lastname"] as? String, !lastName.isEmpty {
        firstLetter = lastName.prefix(1).uppercased()
    } 
    // 姓氏为空或不存在时，处理邮箱字段
    else if let email = tmp["email"] as? String, !email.isEmpty {
        firstLetter = email.prefix(1).uppercased()
    } 
    // 极端情况：姓氏和邮箱都无效，用兜底字符
    else {
        firstLetter = "#"
    } 
    
    // 向对应首字母的数组中添加RSVP数据
    if var group = RSVPDirectory[firstLetter] as? [AnyHashable] {
        group.append(tmp)
        RSVPDirectory[firstLetter] = group
    } else {
        RSVPDirectory[firstLetter] = [tmp]
    }
}

更优雅的Swift原生实现 #

上面的代码依赖NSMutableDictionary和AnyHashable，不够符合Swift的类型安全特性，这里推荐使用原生Swift字典来实现，代码更简洁且安全：

假设你的RSVP数据是[[String: Any]]类型（字典数组） #

// 使用原生Swift字典，类型明确更安全
var rsvpGroups: [String: [[String: Any]]] = [:]

// 直接遍历数组，无需下标索引
for rsvp in self.RSVPs {
    let firstLetter: String
    
    // 处理姓氏：过滤掉空字符串和纯空格的情况
    if let lastName = rsvp["lastname"] as? String, 
       !lastName.trimmingCharacters(in: .whitespaces).isEmpty {
        firstLetter = lastName.prefix(1).uppercased()
    } 
    // 姓氏无效时，处理邮箱
    else if let email = rsvp["email"] as? String, 
            !email.trimmingCharacters(in: .whitespaces).isEmpty {
        firstLetter = email.prefix(1).uppercased()
    } 
    // 兜底处理，避免无有效首字母的情况
    else {
        firstLetter = "#"
    }
    
    // 使用字典的default下标，简化数组的创建与追加操作
    rsvpGroups[firstLetter, default: []].append(rsvp)
}

如果有自定义RSVP模型类（推荐） #

如果你的RSVP数据有对应的模型结构体，代码会更清晰安全，比如：

// 自定义RSVP模型
struct RSVP {
    let lastName: String?
    let email: String
    // 根据需求添加其他字段，比如name、status等
}

// 分组代码
var rsvpGroups: [String: [RSVP]] = [:]

for rsvp in self.RSVPs {
    let firstLetter: String
    
    if let lastName = rsvp.lastName, 
       !lastName.trimmingCharacters(in: .whitespaces).isEmpty {
        firstLetter = lastName.prefix(1).uppercased()
    } else {
        // 邮箱字段设为非可选，直接取首字母
        firstLetter = rsvp.email.prefix(1).uppercased()
    }
    
    rsvpGroups[firstLetter, default: []].append(rsvp)
}

优化点说明 #

• 避免强制解包：用可选绑定if let处理可能为nil的字段，防止运行时崩溃
• 类型安全：使用原生Swift字典替代NSMutableDictionary，明确存储的类型，减少类型转换的麻烦
• 简化逻辑：用default下标直接处理分组数组的创建和追加，省去判断数组是否存在的代码
• 鲁棒性：处理了空字符串、纯空格的情况，添加兜底逻辑避免无有效首字母的异常

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