首页 > 文章列表 > 有哪些情况下泛型方法只适用于类型化泛型的子集?

有哪些情况下泛型方法只适用于类型化泛型的子集?

234 2024-02-09
问题内容

我有一个结构,我想使用任何类型的数字,但某些方法只能使用浮点数(这是所需的)。有什么方法可以键入仅适用于浮点数的方法吗?

假设我有类似的东西:

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 中的所有类型的工作方式相同。