如何在gorm中添加enum？

Question

我正在写PostgreSQL表模式。

type TestTable struct {
    ID        int    `gorm:"column:id;primaryKey;autoIncrement"`
    CarType   string `gorm:"column:car_type"`
}

那么我如何将“轿车”、“掀背车”、“小型货车”等汽车类型添加为枚举数据类型

Answer 1

为了扩展Nick的回答，我为自动化添加了以下内容:

假设您有DB客户端结构，您可以创建一个创建此汽车类型的方法：

func (psqlClient *DBClient) CreateCarTypeEnum() error {
    result := psqlClient.db.Exec("SELECT 1 FROM pg_type WHERE typname = 'car_type';")

    switch {
    case result.RowsAffected == 0:
        if err := psqlClient.db.Exec("CREATE TYPE car_type AS ENUM ('SEDAN', 'HATCHBACK', 'MINIVAN');").Error; err != nil {
            log.Error().Err(err).Msg("Error creating car_type ENUM")
            return err
        }

        return nil
    case result.Error != nil:
        return result.Error

    default:
        return nil
    }
}

Answer 2

附带说明-如果您决定采用稍微不同的方法：您可以将枚举定义为int，并利用iota。然后，您可以使用代码生成器创建sql＜code＞Scaner/Valuer＜/code＞以及json/text表示。例如：https://github.com/dmarkham/enumer

Answer 3

假设你正在使用 GORM 和 PostgreSQL。首先在数据库中创建一个类型。

CREATE TYPE car_type AS ENUM (
    'SEDAN',
    'HATCHBACK',
    'MINIVAN');

然后，您需要定义以下模型：

import "database/sql/driver"

type carType string

const (
    SEDAN  carType = "SEDAN"
    HATCHBACK carType = "HATCHBACK"
    MINIVAN carType = "MINIVAN"
)

func (ct *carType) Scan(value interface{}) error {
    *ct = carType(value.([]byte))
    return nil
}

func (ct carType) Value() (driver.Value, error) {
    return string(ct), nil
}

type MyTable struct {
    gorm.Model
    CarType carType `gorm:"type:car_type"`
}

func (MyTable) TableName() string {
    return "my_table"
}

请注意，对于MySQL用户，您可以添加结构标签gorm：或sql：，这样您就不必运行原始查询即可在数据库中创建枚举。

CarType carType `gorm:"type:enum('SEDAN', 'HATCHBACK', 'MINIVAN')";"column:car_type"`

运筹学

CarType carType `sql:"type:ENUM('SEDAN', 'HATCHBACK', 'MINIVAN')" gorm:"column:car_type"`