gorm golang在postgres中的自引用单对多关系答案

【问题标题】：gorm golang self-referencing onetomany relationship in postgresgorm golang在postgres中的自引用单对多关系
【发布时间】：2020-10-19 09:20:15
【问题描述】：

我正在尝试使用 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: "JimmyJohnson@Test.com", Password: "1234", AuthLevel: 1, UserID: 0},
        {Mail: "HowardHills@Test.com", Password: "1234", AuthLevel: 2, UserID: 1},
        {Mail: "CraigColbin@Test.com", 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": "JimmyJohnson@Test.com",
  "Password": "1234",
  "AuthLevel": 1,
  "Created": [
    {
      "ID": 2,
      "Mail": "HowardHills@Test.com",
      "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")。

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

任何帮助将不胜感激:)

【问题讨论】：

您的用户创建方式确实是错误的，不清楚您想要什么。试着用一个例子来说明它。
您必须先创建父用户，然后再创建子用户。

标签： go go-gorm

【解决方案1】：

您正在使用来自旧版本库的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: "JimmyJohnson@Test.com", Password: "1234", AuthLevel: 1}
db.Create(&user1)

user2 := User{Mail: "HowardHills@Test.com", 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。

【讨论】：

【解决方案2】：

user2 初始化错误

cannot use &node1.ID (type *uint) as type uint in field value

【讨论】：