使用 GORM 在 Golang 中实现 PostgreSQL 中的自我引用 One-to-Many 关系

对于一个Golang开发者来说，牢固扎实的基础是十分重要的，golang学习网就来带大家一点点的掌握基础知识点。今天本篇文章带大家了解《使用 GORM 在 Golang 中实现 PostgreSQL 中的自我引用 One-to-Many 关系》，主要介绍了，希望对大家的知识积累有所帮助，快点收藏起来吧，否则需要时就找不到了！

我正在尝试使用 golang gorm 创建一个自引用的一对多关系。每个用户可以创建多个其他用户 (created)，但一个用户始终由另一个用户 (userid) 创建。

type user struct {
    id        uint
    mail      string
    password  string
    authlevel uint
    created   []user `gorm:"foreignkey:userid"`
    userid    uint
}

var db *gorm.db
var err error
var (
    users = []user{
        {mail: "[email protected]", password: "1234", authlevel: 1, userid: 0},
        {mail: "[email protected]", password: "1234", authlevel: 2, userid: 1},
        {mail: "[email protected]", password: "1234", authlevel: 2, userid: 1},
    }
)

func main() {

    //connection to db

    db.automigrate(&user{})
    for index := range users {
        db.create(&users[index])
    }

    router.handlefunc("/users/{id}", getuser).methods("get")

    handler := cors.default().handler(router)

    log.fatal(http.listenandserve(":8080", handler))
}

//getuser returns one user by id
func getuser(w http.responsewriter, r *http.request) {
    params := mux.vars(r)
    var user user
    var users []user
    db.first(&user, params["id"])
    db.debug().model(&user).related(&users)
    user.created = users

    json.newencoder(w).encode(&user)
}

使用此代码，getuser 函数针对 id = 1 返回以下内容：

{
  "ID": 1,
  "Mail": "[email protected]",
  "Password": "1234",
  "AuthLevel": 1,
  "Created": [
    {
      "ID": 2,
      "Mail": "[email protected]",
      "Password": "1234",
      "AuthLevel": 2,
      "Created": null,
      "UserID": 1
    }
  ],
  "UserID": 2
}

看来我的关系建立方式是错误的。

userid 应该引用创建者的 id，当调用 db.debug().model(&user).related(&users) 时，我希望它查询： select * from "users" where ("user_id" = "id") 而不是 select * from "users" where ("id" = "user_id")。

到目前为止，文档或其他类似的问题还无法帮助我，因为它们正在解决多对多的关系。

任何帮助将不胜感激:)

解决方案

您正在使用 db.related，它来自旧版本的库。 v2 有更稳健的方法来处理关系。如果您陷入 v1 中，以下内容可能仍然有效（我发现 v1 中的 related 有问题，但 preload 应该可以工作）。

在 v2 中，这将按您的预期工作：

type User struct {
    ID        uint
    Mail      string
    Password  string
    AuthLevel uint
    Created   []User `gorm:"foreignkey:UserID"`
    UserID    *uint
}

//...

user1 := User{Mail: "[email protected]", Password: "1234", AuthLevel: 1}
db.Create(&user1)

user2 := User{Mail: "[email protected]", Password: "1234", AuthLevel: 2, UserID: &user1.ID}
db.Create(&user2)

var result User
db.Preload("Created").First(&result, user1.ID)
// result.Created[0] is user2

请注意，我将 userid 设置为 *uint，并且在创建 user1 时没有给它赋值，这样可以避免在尝试为此列插入 0 时发生外键冲突。

此外，通过使用 preload，您不必再次调用 related 或 association。

今天关于《使用 GORM 在 Golang 中实现 PostgreSQL 中的自我引用 One-to-Many 关系》的内容介绍就到此结束，如果有什么疑问或者建议，可以在golang学习网公众号下多多回复交流；文中若有不正之处，也希望回复留言以告知！