将所有枚举值作为数组获取

Question

我有以下枚举.

enum EstimateItemStatus: Printable {
    case Pending
    case OnHold
    case Done

    var description: String {
        switch self {
        case .Pending: return "Pending"
        case .OnHold: return "On Hold"
        case .Done: return "Done"
        }
    }

    init?(id : Int) {
        switch id {
        case 1:
            self = .Pending
        case 2:
            self = .OnHold
        case 3:
            self = .Done
        default:
            return nil
        }
    }
}

我需要将所有原始值作为字符串数组(如此["Pending", "On Hold", "Done"]).

我将此方法添加到枚举中.

func toArray() -> [String] {
    var n = 1
    return Array(
        GeneratorOf<EstimateItemStatus> {
            return EstimateItemStatus(id: n++)!.description
        }
    )
}

但我收到以下错误.

无法为类型为'(( - - > _)'的类型'GeneratorOf'找到初始值设定项

我无法弄清楚如何解决这个问题.有帮助吗？或者请告诉我是否有更简单/更好/更优雅的方式来做到这一点.

谢谢.

Answer 1

对于Swift 4.2(Xcode 10)及更高版本

有一个CaseIterable协议:

enum EstimateItemStatus: String, CaseIterable {
    case pending = "Pending"
    case onHold = "OnHold"
    case done = "Done"

    init?(id : Int) {
        switch id {
        case 1: self = .pending
        case 2: self = .onHold
        case 3: self = .done
        default: return nil
        }
    }
}

for value in EstimateItemStatus.allCases {
    print(value)
}

对于Swift <4.2

不,您无法查询enum其包含的值.看到这篇文章.您必须定义一个列出所有值的数组.还可以看看Frank Valbuena的聪明解决方案.

enum EstimateItemStatus: String {
    case Pending = "Pending"
    case OnHold = "OnHold"
    case Done = "Done"

    static let allValues = [Pending, OnHold, Done]

    init?(id : Int) {
        switch id {
        case 1:
            self = .Pending
        case 2:
            self = .OnHold
        case 3:
            self = .Done
        default:
            return nil
        }
    }
}

for value in EstimateItemStatus.allValues {
    print(value)
}

Answer 2

Swift 4.2引入了一个名为的新协议CaseIterable

enum Fruit : CaseIterable {
    case apple , apricot , orange, lemon
}

当你符合时,你可以从这样的enum案例中得到一个数组

for fruit in Fruit.allCases {
    print("I like eating \(fruit).")
}

Answer 3

还有另一种方法,至少在编译时是安全的:

enum MyEnum {
    case case1
    case case2
    case case3
}

extension MyEnum {
    static var allValues: [MyEnum] {
        var allValues: [MyEnum] = []
        switch (MyEnum.case1) {
        case .case1: allValues.append(.case1); fallthrough
        case .case2: allValues.append(.case2); fallthrough
        case .case3: allValues.append(.case3)
        }
        return allValues
    }
}

请注意,这适用于任何枚举类型(RawRepresentable与否)以及如果添加新案例,那么您将收到一个好的编译器错误,因为这会强制您使其保持最新状态.

Answer 4

enum EstimateItemStatus: String, CaseIterable {
  case pending = "Pending"
  case onHold = "OnHold"
  case done = "Done"

  static var statusList: [String] {
    return EstimateItemStatus.allCases.map { $0.rawValue }
  }
}

[“待定”、“暂停”、“完成”]

Answer 5

我找到了这个代码:

protocol EnumCollection : Hashable {}


extension EnumCollection {

    static func cases() -> AnySequence<Self> {
        typealias S = Self
        return AnySequence { () -> AnyIterator<S> in
            var raw = 0
            return AnyIterator {
                let current : Self = withUnsafePointer(to: &raw) { $0.withMemoryRebound(to: S.self, capacity: 1) { $0.pointee }
                }
                guard current.hashValue == raw else { return nil }
                raw += 1
                return current
            }
        }
    }
}

使用:

enum YourEnum: EnumCollection { //code }

YourEnum.cases()

从YourEnum返回案例列表

Answer 6

将CaseIterable协议添加到枚举：

enum EstimateItemStatus: String, CaseIterable {
    case pending = "Pending"
    case onHold = "OnHold"
    case done = "Done"
}

用法：

let values: [String] = EstimateItemStatus.allCases.map { $0.rawValue }
//["Pending", "OnHold", "Done"]

Answer 7

要获得出于功能目的的列表，请使用EnumName.allCases返回数组的表达式，例如

EnumName.allCases.map{$0.rawValue} 

给定一个字符串列表 EnumName: String, CaseIterable

注意：使用allCases代替AllCases()。