Golang枚举实现，并制成字典反给前端使用

Sun, 18 Oct 2020 00:00:00 +0000

方法

package t

import (
	"reflect"
	"strconv"
)

type EnumType int8

func GetEnumMap(obj interface{}) (statusMap map[EnumType]string) {
	ta := reflect.TypeOf(obj)
	statusMap = make(map[EnumType]string)
	for i := 0; i < ta.NumField(); i++ {
		fl := ta.Field(i)
		statusNum, _ := strconv.Atoi(fl.Tag.Get("enum"))
		statusMap[EnumType(statusNum)] = fl.Tag.Get("desc")
	}
	return
}

func GetEnumByDesc(obj interface{}, desc string) EnumType {
	ta := reflect.TypeOf(obj)
	statusMap := make(map[string]EnumType)
	for i := 0; i < ta.NumField(); i++ {
		fl := ta.Field(i)
		statusNum, _ := strconv.Atoi(fl.Tag.Get("enum"))
		statusMap[fl.Tag.Get("desc")] = EnumType(statusNum)
	}
	return statusMap[desc]
}

func GetAllStatus() (statusList map[string]interface{}) {
	statusList = make(map[string]interface{})
	for _, v := range GetStatusObj() {
		statusList[reflect.TypeOf(v).Name()] = GetEnumMap(v)
	}
	return
}

枚举类

package t

func GetStatusObj() []interface{} {
	return []interface{}{
		UserStatus{},
		TaskStatus{},
	}
}

type UserStatus struct {
	InActivated EnumType `enum:"0" desc:"禁用"`
	Activated   EnumType `enum:"1" desc:"激活"`
}

func GetUserStatus() *UserStatus {
	return &UserStatus{
		InActivated: 0,
		Activated: 1,
	}
}

type TaskStatus struct {
	Waiting EnumType `enum:"0" desc:"排队中"`
	Working EnumType `enum:"1" desc:"处理中"`
	Finish  EnumType `enum:"2" desc:"已完成"`
}

func GetTaskStatus() *TaskStatus {
	return &TaskStatus{
		Waiting: 0,
		Working: 1,
		Finish:  2,
	}
}

使用

使用时，将后端的枚举返回给前端，如前端查询到的状态值为1,从字典中获取到UserStatus[1]
前端返回后端时，直接使用UserStatus[1]返给后端，后端通过GetEnumByDesc()方法来获取到需要存入数据库的值

Golang on Yazzyk's Blog

Golang枚举实现，并制成字典反给前端使用

方法

枚举类

使用