通用加法函数 Swift

Question

我想创建一个可以接受两个参数的函数，并根据输入的类型执行二进制运算（例如，'int' 简单加法，对于字符串，它应该连接等）并返回结果。对于以下方法，我收到诸如“二进制运算符‘+’不能应用于两个‘T’操作数”之类的错误

func commonAdd <T>(paramA:T,paramB:T)->T

Answer 1

一种可能的方法。

1) 可添加协议

您定义了一个Addable 协议。

protocol Addable {
    func add(other:Self) -> Self
}

2) commonAdd 函数

接下来你定义你的功能这种方式

func commonAdd <T: Addable>(paramA:T,paramB:T) -> T {
    return paramA.add(paramB)
}

3) 使 Int 符合 Addable

接下来选择一个类型并使其符合Addable

extension Int: Addable {
    func add(other: Int) -> Int {
        return self + other
    }
}

4) 用法

现在您可以将您的函数与Int 一起使用。

commonAdd(1, paramB: 2) // 3

更多

您应该重复第 3 步，使 Addable 成为您要在函数中使用的每个类型。

Answer 2

符合 AdditiveArithmetic 添加通用数据类型的协议。

func commonAdd<T: AdditiveArithmetic>(paramA:T,paramB:T) ->T{
    return item1 + item2
  }

如果您需要支持字符串，请使其符合 AdditiveArithmetic，但您还需要实现减法：

extension String: AdditiveArithmetic {
    public static func -= (lhs: inout String, rhs: String) {
            var set = Set(rhs)
        lhs.removeAll{ !set.insert($0).inserted }
    }
    public static func - (lhs: String, rhs: String) -> String {
        var set = Set(rhs)
        return lhs.filter{set.insert($0).inserted}
    }
    public static var zero: String { "" }
}

Answer 3

要添加的通用函数（Int、Double、String）

func add<T: Any >(itemA: T, itemB: T) -> T {

    if itemA is Int && itemB is Int {
        debugPrint("Int")
        let intNum1 = itemA as! Int
        let intNum2 = itemB as! Int
        return intNum1 + intNum2 as! T
        
    } else if itemA is Double && itemB is Double {
        debugPrint("Double")
        let doubleNum1 = itemA as! Double
        let doubleNum2 = itemB as! Double
        return doubleNum1 + doubleNum2 as! T
    } else {
        debugPrint("String")
        let string1 = itemA as! String
        let string2 = itemB as! String
        return string1 + string2 as! T
        
    }
    
}