Golang如何将struct类型作为map键构建函数映射替代type-switch

实现思路 #

Go 中 reflect.Type 类型本身是可比较的，完全可以作为 map 的键，你不需要遍历 map，直接通过类型索引就能拿到对应的处理函数，效率更高。

完整可运行代码 #

package main

import (
	"fmt"
	"reflect"
)

// 定义三个自定义结构体
type A struct {
	a1 int
	a2 string
}

type B struct {
	b1 int
	b2 string
}

type C struct {
	c1 int
	c2 string
}

// 各结构体对应的处理函数
func processA(a A) {
	fmt.Printf("处理A类型: a1=%d, a2=%s\n", a.a1, a.a2)
}

func processB(b B) {
	fmt.Printf("处理B类型: b1=%d, b2=%s\n", b.b1, b.b2)
}

func processC(c C) {
	fmt.Printf("处理C类型: c1=%d, c2=%s\n", c.c1, c.c2)
}

// 定义全局funcMap，键为结构体类型，值为对应的处理函数包装
var funcMap = map[reflect.Type]func(arg any){
	reflect.TypeOf(A{}): func(arg any) {
		// 类型断言转换后调用实际处理函数
		processA(arg.(A))
	},
	reflect.TypeOf(B{}): func(arg any) {
		processB(arg.(B))
	},
	reflect.TypeOf(C{}): func(arg any) {
		processC(arg.(C))
	},
	// 如果需要支持指针类型，添加对应项即可，示例：
	// reflect.TypeOf(&A{}): func(arg any) {
	// 	processA(*arg.(*A))
	// },
}

func testFunc(arg any) {
	argType := reflect.TypeOf(arg)
	handler, ok := funcMap[argType]
	if !ok {
		fmt.Printf("未找到类型 %s 的处理函数\n", argType)
		return
	}
	// 调用对应处理函数
	handler(arg)
}

func main() {
	// 测试用例
	a := A{a1: 1, a2: "测试A"}
	b := B{b1: 2, b2: "测试B"}
	c := C{c1: 3, c2: "测试C"}
	unknown := "其他类型"

	testFunc(a)
	testFunc(b)
	testFunc(c)
	testFunc(unknown)
}

运行输出 #

处理A类型: a1=1, a2=测试A
处理B类型: b1=2, b2=测试B
处理C类型: c1=3, c2=测试C
未找到类型 string 的处理函数

注意事项 #

• 如果你的场景中会传入结构体指针，需要在funcMap中额外添加指针类型的映射项，参考代码中的注释示例即可。
• 该方案依赖反射，性能相比原生type switch会有一定损耗，适合业务逻辑中需要频繁新增类型、扩展性要求高的场景，如果是性能敏感场景更推荐用type switch实现。

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