如何在 gorm 中添加枚举？

Question

我正在编写后greSQL表架构。type TestTable struct {    ID        int    `gorm:"column:id;primaryKey;autoIncrement"`    CarType   string `gorm:"column:car_type"`}那么我怎么能添加汽车类型，如“轿车”，“掀背车”，“小型货车”作为枚举数据类型

Answer 1

假设您正在使用戈尔姆。首先在数据库中创建一个类型。

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 `sql:"car_type"`

}

func (MyTable) TableName() string {

    return "my_table"

}

Answer 2

在旁注中 - 如果您决定采用稍微不同的方法：您可以将枚举定义为int，并利用.然后，您可以使用代码生成器来创建sql，也可以创建json /文本表示形式。例如：https://github.com/dmarkham/enumeriotaScaner/Valuer




Answer 3

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

假设您有结构，则可以创建一个创建此汽车类型的方法：DBClient

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 4

更新，它将不工作，而是使用 。一旦您在数据库中手动创建了自定义枚举类型，也是如此。sql:"car_type"gorm:"car_type"