如何让Swift结构体变量支持多个枚举类型？

这个问题一点都不奇怪！很多刚开始接触Swift枚举的开发者都会有类似的需求——想要让一个属性灵活地接受多种枚举类型。你的思路方向是对的，只是Swift里不能直接写PointCategory || BudgetCategory这种语法，下面给你几个实用的解决方案：

方案1：用一个“组合枚举”包装两种类型 #

这是最直接也最常用的方式，利用Swift枚举的关联值特性，创建一个新枚举来包含你现有的两个枚举：

// 定义组合枚举，把两种类型都包含进去
enum CombinedCategory {
    case budget(BudgetCategory)
    case point(PointCategory)
}

// 修改Points结构体的itemCategory类型
struct Points { 
    var user: String 
    var itemName: String 
    var itemCategory: CombinedCategory 
    var code: PointCode 
    var valuePerTap: Int 
    var itemDate: Double 
    var paid: Bool 
}

使用示例： #

赋值的时候可以直接指定具体的枚举类型：

// 用BudgetCategory赋值
let clothingPoint = Points(
    user: "Alice",
    itemName: "Buy Shirt",
    itemCategory: .budget(.clothing),
    code: /* 你的PointCode值 */,
    valuePerTap: 10,
    itemDate: Date().timeIntervalSince1970,
    paid: true
)

// 用PointCategory赋值
let paydayPoint = Points(
    user: "Bob",
    itemName: "Monthly Pay",
    itemCategory: .point(.payday),
    code: /* 你的PointCode值 */,
    valuePerTap: 100,
    itemDate: Date().timeIntervalSince1970,
    paid: true
)

处理这个属性的时候，用switch语句就能区分两种类型：

func handleCategory(_ category: CombinedCategory) {
    switch category {
    case .budget(let budgetCat):
        print("预算分类：\(budgetCat.rawValue)")
    case .point(let pointCat):
        print("积分分类：\(pointCat.rawValue)")
    }
}

方案2：用协议统一枚举类型 #

如果你的两个枚举有共同的特性（比如都遵守RawRepresentable且原始值是String），可以定义一个协议来统一它们，让属性接受协议类型：

// 定义统一的协议，这里要求原始值是String，你也可以根据需求调整
protocol Category: RawRepresentable where RawValue == String {}

// 让两个现有枚举遵守这个协议
enum BudgetCategory: String, Category { 
    case clothing = "Clothing" 
    case donations = "Donations" 
    // ... 其他case不变
}
enum PointCategory: String, Category { 
    case outsideIncome = "outside income" 
    case fees = "fees" 
    // ... 其他case不变
}

// 修改结构体属性类型为协议
struct Points { 
    var user: String 
    var itemName: String 
    var itemCategory: Category 
    var code: PointCode 
    var valuePerTap: Int 
    var itemDate: Double 
    var paid: Bool 
}

这种方案的好处是扩展性强——以后如果需要新增第三种枚举类型，只要让它遵守Category协议就行。但如果需要区分具体是哪个枚举类型，你需要用类型转换（as?）来判断：

func handleCategory(_ category: Category) {
    if let budgetCat = category as? BudgetCategory {
        print("预算分类：\(budgetCat.rawValue)")
    } else if let pointCat = category as? PointCategory {
        print("积分分类：\(pointCat.rawValue)")
    }
}

额外建议：要不要重新设计结构体？ #

如果这两个枚举对应的业务场景差异很大（比如一个是支出分类，一个是收入/积分分类），你可以考虑把Points拆分成两个更具体的结构体，比如BudgetPoint和PointTransaction，各自对应自己的枚举类型。这样代码的语义会更清晰，避免一个结构体承担太多不同的职责。

不过如果你的业务逻辑里确实需要这两种分类共存于同一个结构体中，前面的两个方案完全可以满足需求。

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