如何扩展 K:String 和 V:String 的 Swift 字典

Question

我在想下面的语法：

extension Dictionary where Key:String,Value:String{
    func addIfNew(key:String,value:String){
      print("new item added: \(key) with value: \(value)")
    }
}

例如：该函数将记录任何新添加的数据

Answer 1

这就够了……

protocol P {}
extension String: P {}
extension Dictionary where Key:P, Value:P {
    func addIfNew(key:String, value:String){
        print("new item added: \(key) with value: \(value)")
    }
}

let d:[String:String] = [:]

别做其他符合P的类型就好了

在 Swift3 中

extension Dictionary where Key == String, Value == String {
    func addIfNew(key:String, value:String){
        print("new item added: \(key) with value: \(value)")
    }
}
let d:[String:String] = [:]
d.addIfNew(key: "a", value: "A")

Answer 2

不可能按照您想要的方式进行。您也只能将Dictionary 的extensions 约束为protocols Key 和/或Value。否则你只会得到更多的错误。

但你可以这样做：

extension String : CustomStringConvertible {

    public var description : String { return self }

}

extension Dictionary where Key : CustomStringConvertible, Value : CustomStringConvertible {
    func addIfNew(key:Key,value:Value){
        print("new item added: \(key.description) with value: \(value.description)")
    }
}

var dict : [String:String] = [:]
dict["blah"] = "foo"
dict.addIfNew("blah", value: "foo2")

---

重要的是问问自己你想要什么。

• 仅在Strings 上允许此功能
• Equatable 的任何内容都足够好

如果Equatable 足够好，您可以这样做：

extension Dictionary where Value : Equatable {
    mutating func addIfNew(key:Key,value:Value){

        if self.values.contains(value) {
            print("\(value) exists")
        } else {
            print("new item added: \(key) with value: \(value)")
            self[key] = value
        }
    }
}

var dict : [String:String] = [:]
dict["blah"] = "foo"
dict.addIfNew("blah1", value: "foo2")
dict.addIfNew("blah2", value: "foo2")

虽然此功能适用于Value 为Equatable 的任何Dictionary，但您可以选择仅对String 为Value 的Value 使用它

---

如果你真的需要它只在String 上工作，你需要创建一个自定义protocol 并且只有String 符合它。 protocol 需要有一个返回 self 的计算属性，因为否则您将无法 print 它，因为它在 extension 中不被视为 String。

protocol StringType : Hashable {

    var value : String { get }

}

extension String : StringType {

    var value : String { return self }

}

extension Dictionary where Key : StringType, Value : StringType {
    mutating func addIfNew(key:Key,value:Value){

        if self.values.contains(value) {
            print("\(value.value) exists")
        } else {
            print("new item added: \(key.value) with value: \(value.value)")
            self[key] = value
        }
    }
}

var dict : [String:String] = [:]
dict["blah"] = "foo"
dict.addIfNew("blah1", value: "foo2")
dict.addIfNew("blah2", value: "foo2") 

---

旁注：

由于Dictionary 是struct，函数需要mutating 才能更改self。