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

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: "[e-mail]", Password: "1234", AuthLevel: 1, UserID: 0},

        {Mail: "[e-mail]", Password: "1234", AuthLevel: 2, UserID: 1},

        {Mail: "[e-mail]", 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": "[e-mail]",

  "Password": "1234",

  "AuthLevel": 1,

  "Created": [

    {

      "ID": 2,

      "Mail": "[e-mail]",

      "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

user2 初始化错误cannot use &node1.ID (type *uint) as type uint in field value

Answer 2

您正在使用db.Related哪个来自旧版本的库。V2 具有更强大的关系处理方法。如果您被困在 V1 中，以下可能仍然有效（我发现Related在 v1 中有错误，但Preload应该可以）。在 V2 中，这可以按您的预期工作：type User struct {    ID        uint    Mail      string    Password  string    AuthLevel uint    Created   []User `gorm:"foreignkey:UserID"`    UserID    *uint}//...user1 := User{Mail: "[e-mail]", Password: "1234", AuthLevel: 1}db.Create(&user1)user2 := User{Mail: "[e-mail]", Password: "1234", AuthLevel: 2, UserID: &user1.ID}db.Create(&user2)var result Userdb.Preload("Created").First(&result, user1.ID)// result.Created[0] is user2请注意，我创建了 UserID*uint并且在创建时没有给它一个值user1，这样可以避免在尝试为此列插入 0 时违反外键。此外，通过使用，Preload您不必再次调用Relatedor Association。