gorm:填充新创建的数据的相关字段

Question

我有以下相关表格：

type Person struct {
ID      uint64 `json:"id" gorm:"primary_key;auto_increment"`
Name    string `json:"name"`
Surname string `json:"surname"`
}
type Book struct {
ID        uint64    `json:"id" gorm:"primary_key;auto_increment"`
Title     string    `json:"title" binding:"required,min=2,max=100" gorm:"type:varchar(100)"`
Author    Person    `json:"author" binding:"required" gorm:"foreignkey:AuthorID"` // * here
AuthorID  uint64    `json:"-"`                                                    // * here
WrittenIn string    `json:"written_in" gorm:"type:varchar(5)"`
CreatedAt time.Time `json:"created_at" gorm:"default:CURRENT_TIMESTAMP"`
UpdatedAt time.Time `json:"updated_at" gorm:"default:CURRENT_TIMESTAMP"`
}

我可以使用以下功能成功地使用gorm的Create()方法创建数据：

func CreateBook(ctx *gin.Context) {
// validating input
var inputData CreateBookInput
if err := ctx.ShouldBindJSON(&inputData); err != nil {
ctx.JSON(401, gin.H{"status": "fail", "error": err.Error()})
}
// create book
book := models.Book{Title: inputData.Title, AuthorID: inputData.AuthorID, WrittenIn: inputData.WrittenIn}
database.DB.Create(&book).Preload("Author")
// database.DB.Preload("Author").Create(&book)
// database.DB.Set("gorm:auto_preload", true).Create(&book)

ctx.JSON(201, gin.H{"status": "success", "book": book})
}

我想把这本新创作的书和作者一起归还。预期响应：

"book": {
"id": 10,
"title": "Chinor ostidagi duel",
"author": {
"id": 3,
"name": "John",
"surname": "Smith"
},
"written_in": "1983",
"created_at": "2022-01-07T17:07:50.84473824+05:00",
"updated_at": "2022-01-07T17:07:50.84473824+05:00"
}

但我找不到填充相关"作者"的方法。所以我得到的是：

"book": {
"id": 10,
"title": "Chinor ostidagi duel",
"author": {
"id": 0,       // empty
"name": "",    // empty
"surname": ""  // empty
},
"written_in": "1983",
"created_at": "2022-01-07T17:07:50.84473824+05:00",
"updated_at": "2022-01-07T17:07:50.84473824+05:00"
}

反复尝试这些方法都没有成功：

database.DB.Create(&book).Preload("Author")
database.DB.Preload("Author").Create(&book)
database.DB.Set("gorm:auto_preload", true).Create(&book)
database.DB.Create(&book).Set("gorm:auto_preload", true)

如何填充新建数据的相关字段？

Answer 1

您可以尝试的一种可能的解决方案是使用AfterCreate挂钩。

func (b *Book) AfterCreate(tx *gorm.DB) (err error) {
return tx.Model(b).Preload("Author").Error
}

你可以在这里找到更多关于钩子的信息。

Answer 2

Preload是链式方法，Create是finisher方法。只有finisher方法将生成并执行SQL。

所以。。。

1创建书籍后通过id查找作者

if err ;= database.DB.Create(&book).Error; err != nil {
return err
}
// will not throw not found error
database.DB.Limit(1).Find(&book.Author, book.AuthorID)
ctx.JSON(201, gin.H{"status": "success", "book": book})

2每次加载作者数据，使用钩子

// create and update
// func (b *Book) AfterSave(tx *gorm.DB) (err error) {
// just create
func (b *Book) AfterCreate(tx *gorm.DB) (err error) {
// handle error if you want
tx.Limit(1).Find(&b.Author, b.AuthorID)
return
}