字典下标或运算符，如果密钥不存在，则添加指定的默认&值是SWIFT

发布于11月23日

我需要编写类似以下内容的代码:

var dict = [SomeEnum: SomeObject]()
var someValue: SomeObject! = dict[.case1]
if someValue == nil {
  someValue = someDefaultValue
  dict[.case1] = someValue
}

Nil-coalescing运算符已经存在，但它不会改变原始字典.dict[.case1, default: someDefaultValue也是如此.

您能解释一下如何编写一个新的下标/运算符来包装上面的代码吗？

推荐答案

通过使用关键字subscript而不是关键字func声明新方法，可以创建自定义子脚本.

根据您的要求，您似乎只需要您的自定义下标具有一个getter，而不需要为它设置一个setter-您可以使用默认的setter来设置值.

extension Dictionary {
  /// Subscript which returns the previously set `Value` for `key` in case such value exists.
  /// Otherwise, sets `defaultValue` as the value for `key` and returns `defaultValue`.
  subscript(_ key: Key, defaultValue defaultValue: Value) -> Value {
    mutating get {
      // In case the dictionary already contained a value for key, return the existing value
      if let value = self[key] {
        return value
      } else { // Otherwise, set the key to have defaultValue as its value and return defaultValue
        self[key] = defaultValue
        return defaultValue
      }
    }
  }
}

使用示例:

let defaultValue = "default"
var dict = [Int:String]()
dict[0] = "0"
dict[2] = "2"
let valueFor1 = dict[1, defaultValue: defaultValue] // "default"
print(dict) // [0: "0", 1: "default", 2: "2"]
dict[2, defaultValue: defaultValue] // this returns 2 as the key had a previously set value