具有 T 类型和采用 T 的 \'Typed\' 函数的泛型

Question

有一个模块定义了一堆类型

type Thing1 struct {}

type Thing2 struct {}

以及将这些类型作为参数的函数类型

type T1 func(t *Thing1) {...}
type T2 func(t *Thing2) {...}

然后它使用这些函数类型定义一个映射

var (
ModThing1 = map[string] T1
ModThing2 = map[string] T2
)

在我使用此模块的应用程序中，我想对 Thing1 和 Thing2 使用泛型

就像是

func do[T any](in *T, inMap map[string]func(in *T)) {
  for _, val := range inMap {
     val(in)
  }
}
...
do[mod.Thing1](&mod.Thing1{}, mod.ModThing1)

当然问题是 go 不允许这样做，因为映射值的类型与 mod.ModThing1 值类型不同。 'func(in *T))' 与 'mod.T1'

无论如何让这个工作？

Answer 1

在函数do 中，声明一个额外的类型参数F，其近似约束也引用T。函数参数的类型统一将完成剩下的工作。

func main() {
    do(&foo.Thing1{}, foo.ModThing1)
    do(&foo.Thing2{}, foo.ModThing2)
}

func do[T any, F ~func(*T)](in *T, inMap map[string]F) {
    for _, val := range inMap {
        val(in)
    }
}

游乐场：https://go.dev/play/p/wiWWzXVDG7v

---

如果您可以控制导入的包，另一种解决方案是使用通用函数类型而不是 T1 和 T2，假设它们足够相似：

type F[T any] func(t *T)

// instead of
// type T1 func(t *Thing1)
// type T2 func(t *Thing2)

然后你用F[T]的特定实例声明映射变量：

var (
    ModThing1 = map[string]F[Thing1]{ ... }
    ModThing2 = map[string]F[Thing2]{ ... }
)

然后在 do 函数中使用类型参数实例化 F：

func do[T any](in *T, inMap map[string]F[T]) {
    for _, val := range inMap {
        val(in)
    }
}

游乐场：https://go.dev/play/p/ITBqiqjjVUz

---

顺便说一句，您还可以抽象地图类型：

type M[T any] map[string]F[T]

func do[T any](in *T, inMap M[T]) {
    for _, val := range inMap {
        val(in)
    }
}