使用GORM选择相关字段

时间:2019-04-22 09:39:27

标签: go go-gorm

我是Golang的新手,它来自python背景,因此尝试了解新的和不同的概念。 我正在尝试创建相关字段,然后从数据库中选择它们。

型号:

type Company struct {
    gorm.Model
    Name string
}

type CreditCard struct {
    gorm.Model
    Number    int
    Company   Company
    CompanyId uint
}

type User struct {
    gorm.Model
    Name         string
    CreditCard   CreditCard
    CreditCardID uint
}

创建表和行并从数据库中选择

common.DB.AutoMigrate(
    &Company{},
    &CreditCard{},
    &User{},
)

user := User{
    Name:       "Alice",
    CreditCard: CreditCard{Number: 123456, Company: Company{Name: "Bank A"}},
}

common.DB.Create(&user)

var retrivedUser User
var creditCard CreditCard
var company Company
common.DB.First(&retrivedUser, 1).Related(&creditCard).Related(&company)

fmt.Println("user name", retrivedUser.Name)
fmt.Println("creditcard number", retrivedUser.CreditCard.Number)
fmt.Println("creditcard number related", creditCard.Number)
fmt.Println("company name", retrivedUser.CreditCard.Company.Name)
fmt.Println("company name related", company.Name)

此打印:

user name Alice
creditcard number 0
creditcard number related 123456
company name 
company name related 

两个问题:

  1. 为什么我需要选择Related信用卡作为单独的变量而不是点符号?
  2. 我在invalid association []公司遇到错误Related,并且没有点号或相关工作。我该如何找回?

1 个答案:

答案 0 :(得分:5)

根据documentation,您需要将auto_preload设置为true才能使自动预加载起作用。因此,将检索用户的行更改为:

common.DB.Set("gorm:auto_preload", true).First(&retrivedUser, 1)

应该给您预期的结果。


在旁注中,从不忽略错误。您始终可以将.Error附加到gorm中的几乎所有操作中,并检查返回的错误。例如:

err = common.DB.AutoMigrate(...).Error
if err != nil {
    // handle err
}

err = common.DB.Create(&user).Error
// Check for err
err = common.DB.Set("gorm:auto_preload", true).First(&retrivedUser, 1).Error
// Check for err