gorm关联查询的坑

AuroraLZDF · · 79 次点击 · · 开始浏览    

背景:最近在学习 golang,尝试将之前的PHP项目用 go改写一下。
涉及的表模型如下三个:

// 文章
type Topics struct {
    Id              int    `gorm:"primary_key"`
    Title           string `gorm:"not null"`
    Body            string `gorm:"not null"`
    UserId          int    `gorm:"not null"`
    CategoryId      int    `gorm:"not null"`
    ReplyCount      int
    ViewCount       int
    LastReplyUserId int
    Order           int
    Excerpt         string
    Slug            string
    Status          int `gorm:"default:1"`
    CreatedAt       time.Time
    UpdatedAt       time.Time

    Category Categories `gorm:"foreignkey:CategoryId"`
    User     Users      `gorm:"foreignkey:UserId"`
}
// 用户
type Users struct {
    Id                int    `gorm:"primary_key"`
    Name              string `gorm:"not null"`
    Email             string `gorm:"not null"`
    Password          string `gorm:"not null"`
    Avatar            string
    Introduction      string
    RememberToken     string
    NotificationCount int
    CreatedAt         time.Time
    UpdatedAt         time.Time
    LastActivedAt     time.Time
}
// 分类
type Categories struct {
    Id          int    `gorm:"primary_key"`
    Name        string `gorm:"not null"`
    Description string `gorm:"not null"`
    PostCount   int
    CreatedAt   time.Time
    UpdatedAt   time.Time
}

首先找到 gorm 官方文档的相关介绍:
http://doc.gorm.io/associations.html#belongs-to

clipboard.png

跃跃欲试(code V1):

func (Topics) TopicByID(id int) (*Topics, error) {
    db := DB()
    defer db.Close()

    var topic Topics

    // 1. 
    //result := db.Model(&topic).Where("id=?", id).Related(&topic.User)

    //2. 
    //db.Where("id=?", id).First(&topic)
    //result := db.Model(&topic).Related(&topic.User)
    
    if err := result.Error; err != nil {
        return &topic, err
    }

    if result.RecordNotFound() == true {
        return &topic, utils.Error("文章不存在!")
    }

    return &topic, nil
}

代码中的两种形式基本上是我再百度之后找到的结果。尝试了一番,都报错:(invalid association [])
然后在考虑是否我的外键定义的有问题:

clipboard.png

(也没问题)

继续百度,找到 [gorm的关联问题] (http://licyhust.com/golang/20...。其中:

clipboard.png

这里 Reload方法加了第二个参数,于是查看gorm关于这个方法的介绍:

clipboard.png

可以加对应的外键参数,测试成功。

最终代码(code V2):

func (Topics) TopicByID(id int) (*Topics, error) {
    db := DB()
    defer db.Close()

    var topic Topics

    //db.Where("id=?", id).First(&topic)
    // 关联的关键代码
    //db.Model(&topic).Related(&topic.Category, "CategoryId")
    //result := db.Model(&topic).Related(&topic.User, "UserId")

    result := db.Where("id=?", id).Preload("Category").Preload("User").First(&topic)
    if err := result.Error; err != nil {
        return &topic, err
    }

    if result.RecordNotFound() == true {
        return &topic, utils.Error("文章不存在!")
    }

    return &topic, nil
}

输出结果: // 包含了文章内容,文章对应用户信息,以及文章分类信息

{
6 测试话题2 <p>who are you?</p> 1 3 0 0 0 0   1 2018-09-11 14:30:39 +0800 CST 2018-09-11 14:43:44 +0800 CST 
{3 问答 请保持友善,互帮互助 0 0001-01-01 00:00:00 +0000 UTC 0001-01-01 00:00:00 +0000 UTC} 
{1 aurora 18862324237@163.com 3ffd3de33d3c8bf09bf835d8f7f9b0e0 http://local.beego.com/static/upload/2018/09/07/deepin-cc.png 啦啦啦啦啦~嗝 b506cf029de29c626a1a27bc4d70f5b0 0 2018-08-28 15:56:39 +0800 CST 2018-09-07 14:32:46 +0800 CST 0001-01-01 00:00:00 +0000 UTC}
}

本文来自:Segmentfault

感谢作者:AuroraLZDF

查看原文:gorm关联查询的坑

79 次点击  
加入收藏 微博
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传