对Swift协议RawRepresentable的理解

前言

对于枚举原始值，系统默认只能是字符串、整型、浮点型字面量，那么咱们能不能增长其它类型呢？

答案是能够的，经过协议RawRepresentable就能够实现

正文

首先定义一个枚举

enum Terrain: String {
    case forest = "F"
    case mountain = "M"
    case water = "W"
}
复制代码

一、RawRepresentable协议

该协议定义了一个初始化构造器，当传入无效的原始值时构造器会返回nil；还定义了一个rawValue计算属性。

当定义上述枚举时，编译器将生成以下等价的代码：

enum Terrain {
    case forest
    case mountain
    case water
}

extension Terrain: RawRepresentable {
    typealias RawValue = String

    init?(rawValue: RawValue) {
        switch rawValue {
        case "F": self = .forest
        case "M": self = .mountain
        case "W": self = .water
        default: return nil
        }
    }

    var rawValue: RawValue {
        switch self {
        case .forest: return "F"
        case .mountain: return "M"
        case .water: return "W"
        }
    }
}
复制代码

二、rawValue本质

rawValue本质是一个计算属性

enum Terrain: String {
    case forest = "F"
    case mountain = "M"
    case water = "W"
}
let tt = Terrain.forest
print(tt.rawValue)
复制代码

0x104e32fe2 <+130>: movq   %rdx, -0x50(%rbp)
0x104e32fe6 <+134>: callq  0x104e330b0               ; Test.Terrain.rawValue.getter : Swift.String at <compiler-generated>
0x104e32feb <+139>: movq   0x6056(%rip), %rsi        ; (void *)0x00007fff879c41c8: type metadata for Swift.String
复制代码

经过断点调试汇编能够看到，底层call调用了rawValue的getter方法

而对rawValue进行赋值则会报错

let tt = Terrain.forest
tt.rawValue = "M" //Cannot assign to property: 'rawValue' is immutable
复制代码

三、原始值不会改变枚举的内存

一、原始值是不会存储到枚举变量中去的，而且只占用一个字节（严格的说当枚举成员个数小于256）

方法（1）经过函数MemoryLayout能够验证

MemoryLayout.size(ofValue: Terrain.forest) // → 1个字节
MemoryLayout.size(ofValue: "F") // → 16个字节
复制代码

方法（2）经过xcode调试View Memory能够查看枚举变量在内存中具体存放的内容（已有很多文章介绍，就不细说了）

四、增长原始值类型

RawRepresentable的RawValue关联类型能够是任何类型，前提是字面量，即它们在编译时必须是静态已知的。

而且实际原始值能够在运行时动态建立。举一个用于颜色的枚举示例：

import UIKit

// 默认这里不能直接用UIColor做为原始值
enum Color {
    case red
    case green
    case blue
}

// 经过遵照协议RawRepresentable能够把UIColor做为原始值
extension Color: RawRepresentable {
    typealias RawValue = UIColor

    init?(rawValue: RawValue) {
        switch rawValue {
        case UIColor.red: self = .red
        case UIColor.green: self = .green
        case UIColor.blue: self = .blue
        default: return nil
        }
    }

    var rawValue: RawValue {
        switch self {
        case .red: return UIColor.red
        case .green: return UIColor.green
        case .blue: return UIColor.blue
        }
    }
}

// 可使用UIColor
Color.blue.rawValue is UIColor
复制代码

最后

一、RawRepresentable协议不单单用在枚举上，还有Swift中的选项集合OptionSet（该协议继承自RawRepresentable），应用于位掩码操做

二、原始值和关联值是不一样的；原始值是在定义枚举时被预先填充的值，像上述三个字符串。对于一个特定的枚举成员，它的原始值始终不变。关联值是建立一个基于枚举成员的常量或变量时才设置的值，枚举成员的关联值能够变化。底层以及应用它们的区别还有不少，这里就不细说了。

参考：oleb.net/blog/2016/1…