Go泛型结构体类型推断：如何简化复杂约束表达式？

问题背景 #

原代码通过两个泛型参数约束结构体A的字段m必须是键类型为key1或key2、值类型为value的map（包括自定义map类型），代码如下：

import "fmt"

type key1 int
type key2 string
type value int

type m1 map[key1]value
type m2 map[key2]value

type A[key key1 | key2, t ~map[key]value] struct {
    m t
}

// 遍历map示例，强调约束中定义map的必要性
func (a A[t, key]) foo() {
    for k, v := range a.m {
        fmt.Println(k, v)
    }
}

func main() {
    A[key1, m1]{
        m: m1{1: 1},
    }.foo() // 输出: 1 1

    A[key2, m2]{
        m: m2{"one": 1},
    }.foo() // 输出: one 1
}

但在复杂场景中，重复编写key key1 | key2, t ~map[key]value这类长泛型表达式过于繁琐。尝试用以下方式简化时编译失败：

type constrainKey interface {
    key1 | key2
}

type constrainMap interface {
    map[constrainKey]value
}

type A[t constrainMap] struct {
    m t
}

func (a A[constrainMap]) foo() {
    for k, v := range a.m {
        fmt.Println(k, v)
    }
}

问题原因 #

上述简化写法失败的核心原因：

• constrainMap中map[constrainKey]value的写法不合法，constrainKey是联合类型，不能直接作为map的键类型用于底层类型约束（map的键类型必须是具体类型）。
• 方法接收者A[constrainMap]错误地直接使用约束类型，而非泛型参数。

正确解决方案 #

使用带类型参数的泛型接口，将原有的两个约束合并为一个简洁的泛型接口约束，代码如下：

import "fmt"

type key1 int
type key2 string
type value int

type m1 map[key1]value
type m2 map[key2]value

// 定义泛型接口，约束map的键类型为key1|key2，值类型为value
type MapConstraint[K key1 | key2] interface {
    ~map[K]value
}

// 结构体仅需两个泛型参数，实例化时K会自动推导
type A[t MapConstraint[K], K key1 | key2] struct {
    m t
}

// 方法定义无需重复复杂约束
func (a A[t, K]) foo() {
    for k, v := range a.m {
        fmt.Println(k, v)
    }
}

func main() {
    // 实例化时可自动推导K，无需手动指定
    A[m1]{
        m: m1{1: 1},
    }.foo() // 输出: 1 1

    A[m2]{
        m: m2{"one": 1},
    }.foo() // 输出: one 1
}

代码说明 #

1. 泛型接口MapConstraint：
   • 带类型参数K，约束K必须是key1或key2；
   • 接口本身约束实现类型必须是基于map[K]value的类型（包括原生map和自定义map类型，~符号表示匹配底层类型）。
2. 结构体A：
   • 仅需两个泛型参数t和K，但实例化时Go会自动推导K，无需手动指定，大幅简化代码。
3. 方法定义：
   • 接收者A[t, K]只需使用泛型参数，无需重复写冗长的约束表达式。

内容的提问来源于stack exchange，提问作者Remi.b