gorm golang在postgres中自引用onetomany关系

Question

我正在尝试使用 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").

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

任何帮助将不胜感激 ：）

Answer 1

您正在使用的db.Related是旧版本的库。V2 有更稳健的方法来处理关系。如果您陷入 V1 中，以下内容可能仍然有效（我发现Relatedv1 中有错误，但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 设置为 an*uint并且在创建时没有给它赋值user1，这避免了尝试为此列插入 0 时发生外键冲突。

此外，通过使用，Preload您不必再次调用Related或Association。