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"`

}

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

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

    },

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

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 errordatabase.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 createfunc (b *Book) AfterCreate(tx *gorm.DB) (err error) {  // handle error if you want  tx.Limit(1).Find(&b.Author, b.AuthorID)  return}