Swift选项集合(OptionSet)

Objective-C中的NS_OPTION宏

对于iOS开发者来说,NS_OPTION这个宏应该都不会陌生吧,这个宏给我们的开发带来了极大的便利。

typedef NS_OPTIONS(NSUInteger, UIViewAutoresizing) {
    UIViewAutoresizingNone                 = 0,
    UIViewAutoresizingFlexibleLeftMargin   = 1 << 0,
    UIViewAutoresizingFlexibleWidth        = 1 << 1,
    UIViewAutoresizingFlexibleRightMargin  = 1 << 2,
    UIViewAutoresizingFlexibleTopMargin    = 1 << 3,
    UIViewAutoresizingFlexibleHeight       = 1 << 4,
    UIViewAutoresizingFlexibleBottomMargin = 1 << 5
};

Objective-C中可以通过|来组合选项,通过&来判断给定的选项中是否包含某个成员。

UIViewAutoresizing autoresizing = UIViewAutoresizingFlexibleWidth | UIViewAutoresizingFlexibleHeight;
if (autoresizing & UIViewAutoresizingFlexibleWidth) {
    ...
}

通过|=来追加选项。

autoresizing |= UIViewAutoresizingFlexibleLeftMargin;

OptionSet协议

Swift使用struct来遵从OptionSet协议,以引入选项集合,而非enum。为什么这样处理呢?当枚举成员互斥的时候,比如说,一次只有一个选项可以被选择的情况下,枚举是非常好的。但是和 C 不同,在 Swift 中,你无法把多个枚举成员组合成一个值,而 C 中的枚举对编译器来说就是整型,可以接受任意整数值。

struct Hobbies: OptionSet {
    var rawValue: UInt

    static let none         = Hobbies(rawValue: 0) // 这人很懒,居然没任何爱好😂
    static let basketball   = Hobbies(rawValue: 1 << 0)
    static let football     = Hobbies(rawValue: 1 << 1)
    static let baseball     = Hobbies(rawValue: 1 << 2)
    static let swim         = Hobbies(rawValue: 1 << 3)
    static let badminton    = Hobbies(rawValue: 1 << 4)
    static let all: Hobbies = [.basketball, .football, .baseball, .swim, .badminton]
}

和C一样,Swift中的选项集合结构体使用了高效的位域来表示,但是这个结构体本身表现为一个集合,它的成员则为被选择的选项。这允许你使用标准的集合运算来维护位域,比如使用contains(_:)来检验集合中是否有某个成员,或者是用union(_:)来组合两个位域,通过remove(_:)来移除某个成员。另外,由于OptionSet继承于ExpressibleByArrayLiteral,你可以使用数组字面量来生成一个选项集合。

var hobbies: Hobbies = [.football, .swim]
_ = hobbies.contains(.football) // true
_ = hobbies.contains(.baseball) // false
// 增加`.badminton`选项
hobbies = hobbies.union(.badminton)
// 移除`.football`选项
hobbies.remove(.football)

源码

你可以在GitHub上查看到OptionSet的源码swift/stdlib/public/core/OptionSet.swift,如果你有兴趣。