登录
首页 >  Golang >  Go问答

gorm 预加载不适用于对象集合吗?

来源:stackoverflow

时间:2024-04-15 20:27:33 277浏览 收藏

偷偷努力,悄无声息地变强,然后惊艳所有人!哈哈,小伙伴们又来学习啦~今天我将给大家介绍《gorm 预加载不适用于对象集合吗?》,这篇文章主要会讲到等等知识点,不知道大家对其都有多少了解,下面我们就一起来看一吧!当然,非常希望大家能多多评论,给出合理的建议,我们一起学习,一起进步!

问题内容

我需要从数据库中获取所有模块,其中 status != complete

我的模块表与节表有多对多映射,节表与字段表有多对多映射。

我正在尝试使用下面的语句获取单个数据库查询中的所有模块。

dbconnection.set("gorm:auto_preload", true).where("status != ?", enum.completed).find(&modules)

但它不返回节和字段。 如果我使用下面的语句,那么它确实返回嵌套模型。

dbConnection.Set("gorm:auto_preload", true).Where("status != ?", enum.Completed).First(&modules)

这是否意味着它仅适用于单个记录而不适用于集合?


解决方案


使用 dbconnection.set("gorm:auto_preload", true),您只能加载子关系,但如果您还想加载子关系的子关系,则必须使用嵌套预加载。喜欢

db.preload("sections").preload("sections.fields").find(&modules)

我根据您的描述编写了示例代码,您可以在底部找到完整的代码。让我们看看不同的选项:

// use auto_preload and .first
db.set("gorm:auto_preload", true).where("name = ?", "m0").first(&modules)

output:
[{id:1 name:m0 status:0 sections:[{id:2 name:s1 fields:[]}]}]

因此,使用 .first 你会得到这些部分,但不会得到 sections.fields

// use auto_preload and .find
db.set("gorm:auto_preload", true).find(&modules)

output:
[{id:1 name:m0 status:0 sections:[{id:2 name:s1 fields:[]}]} {id:2 name:m1 status:0 sections:[{id:1 name:s0 fields:[]}]}]

使用 .find 你会得到与 .first 相同的行为

// use .preload
db.preload("sections").preload("sections.fields").find(&modules)

output:
[{id:1 name:m0 status:0 sections:[{id:2 name:s1 fields:[{id:1 name:f0} {id:2 name:f1}]}]} {id:2 name:m1 status:0 sections:[{id:1 name:s0 fields:[{id:1 name:f0} {id:3 name:f2}]}]}]

通过显式预加载,您可以加载 sectionssections.fields。使用 auto_preload 仅预加载直接字段,而不预加载更深的字段。 imo 使用显式预加载通常是更好的做法。

完整代码:

package main

import (
    "fmt"
    "github.com/jinzhu/gorm"
    _ "github.com/jinzhu/gorm/dialects/sqlite"
    "log"
)

type Module struct {
    ID       int `gorm:"primary_key"`
    Name     string
    Status   int
    Sections []Section `gorm:"many2many:module_sections;"`
}

type Section struct {
    ID     int `gorm:"primary_key"`
    Name   string
    Fields []Field `gorm:"many2many:section_fields;"`
}

type Field struct {
    ID   int `gorm:"primary_key"`
    Name string
}

func preloadingSample() error {
    db, err := gorm.Open("sqlite3", "test.db")
    if err != nil {
        return fmt.Errorf("open DB failed: %w", err)
    }
    defer db.Close()

    err = db.AutoMigrate(
        &Module{},
        &Section{},
        &Field{},
    ).Error
    if err != nil {
        return fmt.Errorf("migration failed: %w", err)
    }

    // Delete everything in DB
    db.Delete(&Module{})
    db.Delete(&Section{})
    db.Delete(&Field{})

    // Put some sample data in DB
    sampleFields := []Field{
        {Name: "f0"},
        {Name: "f1"},
        {Name: "f2"},
        {Name: "f3"},
    }
    for idx := range sampleFields {
        err = db.Create(&sampleFields[idx]).Error
        if err != nil {
            return fmt.Errorf("failed to create: %w", err)
        }
    }

    sampleSections := []Section{
        {Name: "s0", Fields: []Field{sampleFields[0], sampleFields[2]}},
        {Name: "s1", Fields: []Field{sampleFields[0], sampleFields[1]}},
    }
    for idx := range sampleSections {
        err = db.Create(&sampleSections[idx]).Error
        if err != nil {
            return fmt.Errorf("failed to create: %w", err)
        }
    }

    sampleModules := []Module{
        {Name: "m0", Sections: []Section{sampleSections[1]}},
        {Name: "m1", Sections: []Section{sampleSections[0]}},
    }
    for idx := range sampleModules {
        err = db.Create(&sampleModules[idx]).Error
        if err != nil {
            return fmt.Errorf("failed to create: %w", err)
        }
    }

    var modules []Module

    // Load just one module with auto_preload
    err = db.Set("gorm:auto_preload", true).Where("name = ?", "m0").First(&modules).Error
    if err != nil {
        return fmt.Errorf("auto_preload first: %w", err)
    }
    fmt.Printf("%+v\n", modules)

    // Load all modules with auto_preload
    err = db.Set("gorm:auto_preload", true).Find(&modules).Error
    if err != nil {
        return fmt.Errorf("auto_preload find: %w", err)
    }
    fmt.Printf("%+v\n", modules)

    // Explicitly load
    err = db.Preload("Sections").Preload("Sections.Fields").Find(&modules).Error
    if err != nil {
        return fmt.Errorf("preload find: %w", err)
    }
    fmt.Printf("%+v\n", modules)

    return nil
}

func main() {
    err := preloadingSample()
    if err != nil {
        log.Fatal(err)
    }
}

到这里,我们也就讲完了《gorm 预加载不适用于对象集合吗?》的内容了。个人认为,基础知识的学习和巩固,是为了更好的将其运用到项目中,欢迎关注golang学习网公众号,带你了解更多关于的知识点!

声明:本文转载于:stackoverflow 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>