如何创建谓词来过滤Swift中具有关联值的枚举数组？

Question

enum EnumType {
    case WithString(String)
}

var enums = [EnumType]()

enums.append(EnumType.WithString("A"))
enums.append(EnumType.WithString("B"))
enums.append(EnumType.WithString("C"))
enums.append(EnumType.WithString("D"))
enums.append(EnumType.WithString("E"))
enums.append(EnumType.WithString("F"))

如何过滤我的enums数组以找到相关值相等的数组C.我需要使用什么谓词？

Answer 1

过滤器函数可以作为数组上的全局函数调用,也可以作为实例方法调用(我更喜欢后者,因为它更像是OO).

它接受一个带有一个参数(被评估的元素)的闭包,它返回一个布尔值(表示该元素是否与所需条件匹配).

由于在明确的情况下它是一个简单的闭包,所以可以使用缩写形式.

我猜其他"With"案例会添加到你的枚举中,所以你可以选择以下内容:

let filteredArray = enums.filter { 
    switch $0 {
      case let .WithString(value):
        return value == "C"
      default:
        return false
    }
 }

这应该是你的例子中的诀窍.

Answer 2

正如有人提到的那样,对于Swift> 2.0,如果有案例陈述可用:

enums.filter {
  if case .WithString("C") = $0 {
    return true
  }
  return false
}

但是,它看起来并不好看,特别是如果你要再次重复相同的逻辑.我们在这里可以做的是使EnumType符合Equatable:

extension EnumType: Equatable {
}

func ==(lhs: EnumType, rhs: EnumType) -> Bool {
    switch (lhs, rhs) {
    case (.WithString(let lStr), .WithString(let rStr)):
        return lStr == rStr
    }
}

现在你可以:

enums.filter { $0 == .WithString("C") }