我想使一个函数在Swift中接受任何数字(Int,Float,Double,…)
- func myFunction <T : "What to put here"> (number : T) -> {
- //...
- }
而不使用NSNumber
这实际上是不可能在Swift开箱即用。为此,您需要创建一个新的协议,通过在通用函数中使用的任何方法和操作符进行声明。这个过程将适用于您,但具体细节将取决于您的通用功能的功能。以下是如何为获取数字n并返回(n – 1)^ 2的函数执行此操作。
首先,定义你的协议,与运算符和一个初始化器接受Int(这样我们可以减去一个)。
- protocol NumericType {
- func +(lhs: Self,rhs: Self) -> Self
- func -(lhs: Self,rhs: Self) -> Self
- func *(lhs: Self,rhs: Self) -> Self
- func /(lhs: Self,rhs: Self) -> Self
- func %(lhs: Self,rhs: Self) -> Self
- init(_ v: Int)
- }
所有的数字类型都已经实现了这些,但是在这一点上,编译器不知道它们符合新的NumericType协议。你必须明确这一点 – 苹果称之为“通过扩展声明协议”。我们将为Double,Float和所有整数类型执行此操作:
- extension Double : NumericType { }
- extension Float : NumericType { }
- extension Int : NumericType { }
- extension Int8 : NumericType { }
- extension Int16 : NumericType { }
- extension Int32 : NumericType { }
- extension Int64 : NumericType { }
- extension UInt : NumericType { }
- extension UInt8 : NumericType { }
- extension UInt16 : NumericType { }
- extension UInt32 : NumericType { }
- extension UInt64 : NumericType { }
现在我们可以写我们的实际函数,使用NumericType协议作为通用约束。
- func minusOneSquared<T : NumericType> (number : T) -> T {
- let minusOne = number - T(1)
- return minusOne * minusOne
- }
- minusOneSquared(5) // 16
- minusOneSquared(2.3) // 1.69
- minusOneSquared(2 as UInt64) // 1