在使用 GORM + Postgres + Go Gin + Graphql 创建之前自动生成 UUID

Question

有什么方法可以在 go 的数据库中保存 object 的同时使用 GORM 自动生成 UUID？

我有 ROR 迁移的经验，默认情况下会自动生成 ID 和 PK。

这是我的代码

Todo.go

package model

type Todo struct {
    ID     string `json:"id"`
    Text   string `json:"text"`
    Done   bool   `json:"done"`
}

架构.resolvers.go

func (r *mutationResolver) CreateTodo(ctx context.Context, input model.NewTodo) (*model.Todo, error) {
    db, _ := database.connect()
    defer db.Close()

    todo := &model.Todo{
        Text:   input.Text,
        ID:     fmt.Sprintf("%d", rand.Int()),  // I don't want to provide ID like this
    }
    db.Create(&todo)
    ...
}

models_gen.go

# GraphQL schema example
#
# https://gqlgen.com/getting-started/

type Todo {
  id: ID!
  text: String!
  done: Boolean!
}

input NewTodo {
  text: String!
  userId: String!
}


type Mutation {
  createTodo(input: NewTodo!): Todo!
}

任何帮助将非常感激。

Answer 1

假设您使用的是 gorm v2，您可以使用hooks来实现这一点。 关于钩子的文档在这里

特别是，您可以查看BeforeCreate钩子。 应用于model ，它提供了一个容器，用于在数据库中创建之前运行的函数作为其名称状态。

GORM 文档还提供了一个示例，用于在我们使用 Gorm 在数据库中创建用户实体时生成 UUID，这似乎是您要查找的内容：

func (u *User) BeforeCreate(tx *gorm.DB) (err error) {
  u.UUID = uuid.New()

  if !u.IsValid() {
    err = errors.New("can't save invalid data")
  }
  return
}

Answer 2

GORM Hooks是创建 object 时生成 UUID 的最明显的解决方案。

func (t *Todo) BeforeCreate(tx *gorm.DB) (err error) {
    t.ID = uuid.New().String()
    return
}

但是如果你想为每个 object 生成 UUID，你可以定义一个基本结构并将它嵌入到 object 结构中。

type Base struct {
    ID string `json:"id"`
}

func (b *Base) BeforeCreate(tx *gorm.DB) (err error) {
    b.ID = uuid.New().String()
    return
}

type Todo struct {
    Base
    Text string `json:"text"`
    Done bool   `json:"done"`
}

type User struct {
    Base
    Name string `json:"name"`
}

或者直接使用GORM插件为对象生成 UUID。

import "github.com/invzhi/next"

plugin := NewPlugin()
// register uuid generate function
plugin.Register("uuid", func(_, zero bool) (interface{}, error) {
    if !zero {
        return nil, SkipField
    }
    return uuid.New().String()
})

_ = db.Use(plugin)

type Todo struct {
    ID   string `gorm:"type:varchar(20);column:id;next:uuid"` // next:uuid meaning generate uuid when create
    Text string `gorm:"type:varchar(100);column:text`
    Done bool   `gorm:"type:boolean;column:done`
}

type User struct {
    ID   string `gorm:"type:varchar(20);column:id;next:uuid"` // next:uuid meaning generate uuid when create
    Name string `gorm:"type:varchar(100);column:name"`
}