如何让函数接收实现泛型接口的结构体？Go泛型适配问题

问题场景 #

定义泛型接口Worker：

type Worker[input any, output any] interface {
    Process(input) output
}

用结构体IntWorker实现该接口：

type IntWorker[i int, o int] struct{}

func (w *IntWorker[i, o]) Process(input i) o {
    fmt.Println("running i")
    return 1
}

调用WorkManager的RegisterWorker方法时触发编译错误：

mgr := internal.NewWorkManager()
iwkr := &IntWorker[int, int]{}
mgr.RegisterWorker(iwkr)

错误信息：

cannot use iwkr (variable of type *IntWorker[int, int]) as internal.Worker[any, any] value in argument to : *IntWorker[int, int] does not implement internal.Worker[any, any] (wrong type for method Process)
        have Process(input int) int
        want Process(any) any

其中RegisterWorker定义：

func (m *WorkManager) RegisterWorker(f Worker[any, any]) uuid.UUID

错误原因 #

Go的泛型接口类型参数是不变的，Worker[any, any]是一个具体的接口类型，要求Process方法必须能接收任意any类型的输入、返回任意any类型的输出。而*IntWorker[int, int]的Process仅能处理int输入、返回int，无法满足接收任意类型的要求，因此编译器报错。

解决方案 #

方案1：将RegisterWorker改为泛型方法 #

修改WorkManager的RegisterWorker为泛型方法，支持任意输入输出类型的Worker：

func (m *WorkManager) RegisterWorker[I any, O any](f Worker[I, O]) uuid.UUID {
    // 业务逻辑实现
}

调用时无需额外修改，编译器会自动推导类型参数：

mgr := internal.NewWorkManager()
iwkr := &IntWorker[int, int]{}
mgr.RegisterWorker(iwkr) // 正常编译

方案2：引入非泛型基础接口统一管理 #

如果需要统一管理不同类型的Worker，可以先定义一个非泛型的基础接口，让泛型Worker继承它：

type BaseWorker interface {
    // 可添加通用方法，或作为标记接口
}

type Worker[input any, output any] interface {
    BaseWorker
    Process(input) output
}

修改RegisterWorker的参数类型为BaseWorker：

func (m *WorkManager) RegisterWorker(f BaseWorker) uuid.UUID {
    // 业务逻辑实现
}

后续调用Process时需要通过类型断言确定具体类型：

// 假设从WorkManager中取出worker
if intWorker, ok := worker.(Worker[int, int]); ok {
    result := intWorker.Process(100)
    // 处理结果
}

方案3：调整IntWorker实现以适配Worker[any, any]（不推荐） #

仅当业务确实需要处理任意类型输入时，才修改IntWorker的Process方法，通过类型断言处理输入：

type IntWorker struct{}

func (w *IntWorker) Process(input any) any {
    val, ok := input.(int)
    if !ok {
        // 处理类型不匹配的情况，如返回默认值或错误
        return 0
    }
    fmt.Println("running i:", val)
    return val + 1
}

此时*IntWorker可直接实现Worker[any, any]，满足RegisterWorker的参数要求。

总结 #

• 若WorkManager无需关心Worker的具体输入输出类型，优先选择方案1，代码简洁且类型安全。
• 若需要统一管理多种类型的Worker，选择方案2，但需额外处理类型断言。
• 方案3仅适用于必须处理任意类型输入的场景，会丢失部分类型安全性。

内容的提问来源于stack exchange，提问作者tmp dev