swift – 你可以扩展枚举吗?

前端之家收集整理的这篇文章主要介绍了swift – 你可以扩展枚举吗?前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。
我使用枚举来存储这样的字符串值:
enum Animals: String {
        case descCat = "I has attitude"
        case descDog = "how can I help"
        case descGator = "I will eat you"
        var s: String {
            get {
                return self.rawValue as String
            }
        }
    }

然后我访问他们:

print("Dogs be like:" + Animals.descDog.s)

我的问题是可以像任何其他struct或对象一样扩展枚举,所以我不必为每个枚举添加var s:String {}属性

您要为其原始值是字符串的所有枚举添加属性?这听起来像是受约束的协议扩展的情况!
extension RawRepresentable where RawValue == String {
    var description: String {
        return rawValue
    }
}

这是因为具有原始值的所有枚举自动符合RawRepresentable协议,并且所述协议具有关联的类型RawValue,它告诉您原始值的类型。

现在你的动物枚举会自动继承它:

print(Animals.descCat.description) // -> "I has attitude"

请注意,字符串枚举本身已经是CustomStringConvertible,因此它们已经具有描述属性(返回枚举大小写的名称),并且您的代码不会覆盖它:

print(Animals.descCat) // -> "descCat"

如果您希望您的描述覆盖默认值,只需添加一个CustomStringConvertible一致性声明到您的枚举:

private enum Animals: String,CustomStringConvertible { /*...*/ }
print(Animals.descCat) // -> "I has attitude"

您也可以扩展这个想法来涵盖其他原始值类型。例如:

extension RawRepresentable where RawValue: CustomStringConvertible {
    var description: String {
        return rawValue.description
    }
}

现在,您可以获取原始值为Int或自定义类型的枚举的自动描述(只要该类型具有自己的描述)。

猜你在找的Swift相关文章