Go语言如何基于Map字符串键实现类型切换？

错误原因 #

Go语言中结构体嵌入是组合关系，不是继承。ParamsRequest和OtherRequest是包含RequestType的全新类型，并非RequestType的子类型，因此无法直接将它们赋值给map[string]RequestType的value。

解决方案 #

根据你的需求（访问公共字段+特定字段），提供两种可行方案：

方案1：使用interface{}作为Map值类型（快速实现） #

将Map的value类型改为interface{}，存储所有请求类型实例，之后通过类型断言区分具体类型，分别访问公共字段和特定字段：

package main

import "fmt"

type RequestType struct {
	Type string `json:"type"`
}

type ParamsRequest struct {
	RequestType
	Param1 string
	Param2 string
}

type OtherRequest struct {
	RequestType
	Param3 string
	Param4 string
}

var requestTypes = map[string]interface{}{
	"base":  ParamsRequest{RequestType: RequestType{Type: "params"}, Param1: "val1"},
	"base2": OtherRequest{RequestType: RequestType{Type: "other"}, Param3: "val3"},
}

func main() {
	// 访问base请求的公共字段和特定字段
	if req, ok := requestTypes["base"].(ParamsRequest); ok {
		fmt.Println("公共Type字段:", req.Type)
		fmt.Println("特定Param1字段:", req.Param1)
	}

	// 访问base2请求的公共字段和特定字段
	if req, ok := requestTypes["base2"].(OtherRequest); ok {
		fmt.Println("公共Type字段:", req.Type)
		fmt.Println("特定Param3字段:", req.Param3)
	}
}

方案2：定义公共接口（规范写法） #

定义一个包含公共方法的接口，让所有请求类型实现该接口，Map的value类型设为这个接口，统一处理公共逻辑，再通过类型断言访问特定字段：

package main

import "fmt"

// 定义公共接口，声明公共字段的访问方法
type Request interface {
	GetType() string
}

type RequestType struct {
	Type string `json:"type"`
}

type ParamsRequest struct {
	RequestType
	Param1 string
	Param2 string
}

// 给ParamsRequest实现Request接口
func (p ParamsRequest) GetType() string {
	return p.Type
}

type OtherRequest struct {
	RequestType
	Param3 string
	Param4 string
}

// 给OtherRequest实现Request接口
func (o OtherRequest) GetType() string {
	return o.Type
}

// Map的value类型为Request接口
var requestTypes = map[string]Request{
	"base":  ParamsRequest{RequestType: RequestType{Type: "params"}, Param1: "val1"},
	"base2": OtherRequest{RequestType: RequestType{Type: "other"}, Param3: "val3"},
}

func main() {
	// 统一遍历访问所有请求的公共Type字段
	for key, req := range requestTypes {
		fmt.Printf("请求[%s]的公共Type字段: %s\n", key, req.GetType())
	}

	// 访问特定请求的专属字段
	if pReq, ok := requestTypes["base"].(ParamsRequest); ok {
		fmt.Println("base请求的Param1字段:", pReq.Param1)
	}
}

注意事项 #

如果需要修改结构体字段的值，建议在Map中存储指针类型，并将接口方法的接收者改为指针：

func (p *ParamsRequest) GetType() string {
	return p.Type
}

var requestTypes = map[string]Request{
	"base":  &ParamsRequest{RequestType: RequestType{Type: "params"}, Param1: "val1"},
	"base2": &OtherRequest{RequestType: RequestType{Type: "other"}, Param3: "val3"},
}

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