我有一个结构,我想使用任何类型的数字,但某些方法只能使用浮点数(这是所需的)。有什么方法可以键入仅适用于浮点数的方法吗?
假设我有类似的东西:
type number interface { integer | float } type integer interface { uint | uint8 | uint16 | uint32 | uint64 | int | int8 | int32 | int64 } type float interface { float32 | float64 }
我有一个结构定义为:
type mystruct[t number] struct { data []t }
我想要一个仅适用于浮点数的方法:
func (ms MyStruct[T]) SomeMethod() MyStruct[T] { result := ms.Clone() for i, v := range result.Data { result.Data[i] = helper[Float](v) } return result } func helper[T Float] (v T) T { return v*v }
我希望 somemethod 仅在“t”是“float”类型泛型时才起作用。 go 中的泛型可以实现这一点吗?
或者更正式的东西,例如,假设类型 mystruct 使用 mystruct[float32]{} 或 mystruct[float64]{} 实例化,允许调用者调用“somemethod”,否则给出类型错误。
我也愿意接受其他方式来做到这一点,但这正是我想要做的。我正在写一个张量库。我需要整数和浮点数,以及一些仅处理浮点数的方法。我可以让它与整数一起工作,但这实际上没有意义。如果用户尝试使用某些带有整数的方法来偏离它,我宁愿用户收到错误。
不,方法使用与定义结构相同的类型参数运行。您无法专门化它们、添加或删除类型参数等。
您可以做的是将方法编写为简单函数,而不是使用结构体作为参数。然后,使用 integer
实例化的结构的调用将根本无法编译。
// not a method, takes the struct as argument func SomeMethod[T Float](ms MyStruct[T]) MyStruct[T] { } func main() { m1 := MyStruct[float64]{} // compiles SomeMethod(m1) m2 := MyStruct[uint8]{} // doesn't compile SomeMethod(m2) }
区别在于,通用函数是在调用站点实例化的(在本例中),您可以在每次调用时提供不同的 t
,而方法是与类型一起实例化的。
另一种方法是检查 t
的具体类型,如果它不是所需的类型之一,则会出现恐慌或返回错误,但这违反了最小惊喜原则:如果您定义 mystruct[t number]
,则预计它的方法与 number
中的所有类型的工作方式相同。