自定义SQL类型枚举字段的定义方法

知识点掌握了，还需要不断练习才能熟练运用。下面golang学习网给大家带来一个Golang开发实战，手把手教大家学习《自定义SQL类型枚举字段的定义方法》，在实现功能的过程中也带大家重新温习相关知识点，温故而知新，回头看看说不定又有不一样的感悟！

我想像这样定义自定义sql类型enum：

package db

import (
    "database/sql/driver"
    "github.com/spf13/cast"
)

type enum int64

func (e enum) value() (driver.value, error) {
    return int64(e), nil
}

func (e *enum) scan(v any) error {
    *e = enum(cast.toint64(v))
    return nil
}

我想使用以下内容：

package db_test

import (
    "db"
    "gorm.io/driver/sqlite"
    "gorm.io/gorm"
    "testing"
)

type StatusEnum = db.Enum

const (
    StatusEnumEnable StatusEnum = iota
    StatusEnumDisable
)

// this is error because this type alias can not define new method
//func (e *StatusEnum) String() string{
//  if v,ok := StatusEnumMap[*e];ok{
//      return v
//  }
//  return "unknown"
//}

var StatusEnumMap = map[StatusEnum]string{
    StatusEnumEnable:  "Enable",
    StatusEnumDisable: "Disable",
}

type module struct {
    Status StatusEnum
}

func TestEnum(t *testing.T) {
    db, _ := gorm.Open(sqlite.Open(":memory:"))

    db.AutoMigrate(&module{})

    db.Model(&module{}).Create(&module{Status: StatusEnumDisable})

    var out *module
    // SELECT * FROM `modules` WHERE status=1 ORDER BY `modules`.`status` LIMIT 1
    db.Debug().Model(&module{}).Where("status=?", StatusEnumDisable).First(&out)

    t.Log(out)
    t.Log(StatusEnumMap[out.Status])
    //t.Log(out.Status.String())
}

现在，当我想要字符串格式时，我像这样使用 statusenummap[enumvalue] 。我认为 value.string() 更加优雅和简单。 fmt 格式化还可以打印字符串而不是数字。

如何更好地实现这个功能？感谢您的建议。

以上问题已由@zeke lu回答。

还有一个小问题，如果我没有使用valuer并删除string方法，sql条件status=1将是status="1"，为什么？ mysql中类型不一致是否会影响sql的查询性能。

第二个问题：只是gorm调试信息。实际执行的sql为status=1。请参阅 gorm 问题

正确答案

如果底层类型是int64，则似乎不需要实现valuer和scanner接口。如果这是真的，则不需要 enum 类型。以下示例无需 enum 即可工作：

注意：我稍微修改了您的演示，以在 statusenum 类型（而不是 *statusenum）上实现 stringer 接口。

package db_test

import (
    "testing"

    "gorm.io/driver/sqlite"
    "gorm.io/gorm"
)

type statusenum int64

const (
    statusenumenable statusenum = iota
    statusenumdisable
)

var statusenummap = map[statusenum]string{
    statusenumenable:  "enable",
    statusenumdisable: "disable",
}

func (e statusenum) string() string {
    if v, ok := statusenummap[e]; ok {
        return v
    }
    return "unknown"
}

type module struct {
    status statusenum
}

func testenum(t *testing.t) {
    db, _ := gorm.open(sqlite.open(":memory:"))

    db.automigrate(&module{})

    db.model(&module{}).create(&module{status: statusenumdisable})

    var out *module
    // select * from `modules` where status=1 order by `modules`.`status` limit 1
    db.debug().model(&module{}).where("status=?", statusenumdisable).first(&out)

    t.log(out)
    t.logf("status value: %d, status string: %s", out.status, out.status)
}

输出为：

&{Disable}
status value: 1, status string: Disable

请注意，现在第一个是 &{disable}。之前是 &{1}。

以上就是《自定义SQL类型枚举字段的定义方法》的详细内容，更多关于的资料请关注golang学习网公众号！