如何将Go接口方法参数约束为仅允许的几个指定结构体类型？

可以通过Go 1.18及以上版本的泛型类型集约束实现，全程无需使用interface{}/any，所有类型校验在编译期完成，不会有运行时类型错误。

核心实现步骤 #

• 第一步：预先定义所有允许的模块输入结构体

// 示例：预先指定的两个合法输入类型
type ModuleAInputs struct {
    ConfigPath string
    Timeout    int
}

type ModuleBInputs struct {
    TargetAddr string
    RetryCount int
}

• 第二步：定义输入类型约束，通过类型集并集限定仅上述结构体可作为模块入参

// AllowedModuleInput 类型约束，仅允许列表内的结构体作为模块Run方法入参
type AllowedModuleInput interface {
    ModuleAInputs | ModuleBInputs
}

• 第三步：定义泛型版Module接口，绑定类型参数约束

// Module 泛型模块接口，类型参数T必须符合AllowedModuleInput约束
type Module[T AllowedModuleInput] interface {
    Run(moduleInput T) error
}

• 第四步：各模块实现接口时，直接绑定自身对应的输入类型即可

// 模块A实现
type ModuleA struct{}

// Run方法入参固定为ModuleAInputs，传其他类型会直接编译报错
func (m ModuleA) Run(input ModuleAInputs) error {
    // 模块A业务逻辑
    return nil
}

// 模块B实现
type ModuleB struct{}

// Run方法入参固定为ModuleBInputs
func (m ModuleB) Run(input ModuleBInputs) error {
    // 模块B业务逻辑
    return nil
}

统一存储模块实例的补充方案 #

由于不同类型参数的Module[T]属于不同接口类型，如果需要统一收纳所有合法模块实例，可以额外定义一个非泛型标记接口，避免外部非法类型混入：

// AnyModule 仅用于统一存储合法模块实例的标记接口
type AnyModule interface {
    isAllowedModule()
}

// 给合法模块实现标记方法，方法为小写外部无法自行实现
func (ModuleA) isAllowedModule() {}
func (ModuleB) isAllowedModule() {}

后续新增允许的输入类型时，只需要在AllowedModuleInput的类型集里追加对应结构体，再给对应模块实现Run方法和标记方法即可，全程编译期拦截非法入参类型，不需要写运行时类型判断逻辑。

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