ARTICLE DETAIL

建站实战干货

来自一线的建站与推广经验沉淀,每一条都经过真实交付验证。

GO GORM 自定义数据类型-枚举

2026/8/9 2:47:43 拓冰建站 浏览量
GO GORM 自定义数据类型-枚举

文章目录

  • 枚举
    • 枚举的终极办法,原文3.0版本
      • 枚举的实现举例

学习来源: 枫枫知道

枚举

  很多时候,由于字符串的所占据空间较大,而某些状态的值是一定的,所以我们希望用枚举来固定对应的值。这样不仅可以节省空间,更关键的是可以方便后期的维护
  我们这里以主机管理为例,状态有 Running 运行中, OffLine 离线, Except 异常
  我们跳过原文章中方法逐步优化的过程,直接到最完善的版本,即使用数字表示状态和使用类型别名的方法,最后通过手动逻辑方法来实现golang中的枚举

枚举的终极办法,原文3.0版本

package mainimport ("encoding/json""fmt"
)type Status int // 类型别名// 定义running、except、offline对应的常量,本质为整数数据
// 通过定义常量我们可以在查询时直接返回这里的常量名,而不是数字等容产生歧义
const ( //在线、异常、离线Running Status = iota + 1 // iota为golang中常量的行索引,起步为0,每行+1ExceptOffline
)type Host struct {ID     uint   `gorm:"json:id"`Name   string `gorm:"json:name"`Status Status `gorm:"json:status"`
}// 在status的json序列化过程中进行常量和字符串的转化
func (status Status) MarshalJSON() ([]byte, error) {var str string //定义一个字符串方便对status的json转换switch status {case Running:str = "running"case Except:str = "except"case Offline:str = "offline"}return json.Marshal(str)}func main() {//定义一个host主机实例var host = Host{Status: Running,Name:   "wang2",}result, _ := json.Marshal(host) //序列化host为json以存储fmt.Printf(string(result))//查询结果如下,可以看到status的值这里查询为字符//{"ID":0,"Name":"wang2","Status":"running"}
}

枚举的实现举例

在需要输出的时候(print,json),自定义类型就变成了字符串

复习阶段:数据类型的转换
Go 语言类型转换基本格式如下:

type_name(expression)

type_name 为类型,expression 为表达式。
字符串类型的数据转换可以用到strconv,strconv.atoi将字符串转换为int;strconv.itoa将int转换为字符串

package mainimport ("encoding/json""fmt""time"
)type WeekDay int //定义一个数值的类型别名,方便显示// 定义weekday与整数的对应
const (Sunday WeekDay = iota + 1MondayTuesdayWednesdayThursdayFridaySaturday
)var WeekStringList = []string{"Sunday", "Monday", "Tuesday", "Thursday", "Wednesday", "Friday", "Saturday"}
var WeekTypeList = []WeekDay{Sunday, Monday, Tuesday, Thursday, Wednesday, Friday, Saturday}// 通过定义weekday的string方法将其本质int映射为对应的string 自定义》字符
// 实现了自定义weekday类型的string接口,会在输出、打印weekday时自动调用,其被包含在stringer包中,具体可查阅该包
func (W WeekDay) String() string {s := WeekStringList[W]//fmt.Println("from string 方法", s)return s
}func (W WeekDay) MarshalJSON() ([]byte, error) {s, _ := json.Marshal(W.String())//fmt.Println("marshal启用")return s, nil}// ParseStrWeekday 将字符串类型转化为自定义类型 字符串》自定义
func (W *WeekDay) ParseStrWeekday(week string) {for i, v := range WeekStringList {if v == week {*W = WeekTypeList[i]fmt.Println(*W)}}
}// ParseIntWeekday Int》weekday自定义类型
func (W *WeekDay) ParseIntWeekday(week int) {*W = WeekTypeList[week-1]
}type DayInfo struct {WeekDay WeekDay   `json:"WeekDay"`Data    time.Time `json:"Data"`
}func main() {December7th := DayInfo{WeekDay: Wednesday,Data:    time.Now(),}re, _ := json.Marshal(December7th)fmt.Println(string(re))//通过ParseStrWeekday和ParseIntWeekday这两个方法,实现了输入字符串和int都可以转换为weekday类型,同时通过该类型的string接口的实现可以在输出时自动以对应的字符串了类型进行输出var w WeekDayw.ParseStrWeekday("Sunday")fmt.Println("我输入的字符串“Sunday”,输出结果为:", w)w.ParseIntWeekday(2)fmt.Println("我输入的int类型 2,输出结果为: ", w)}