使用GORM在GoLang中循环二叉树

小伙伴们有没有觉得学习Golang很有意思？有意思就对了！今天就给大家带来《使用GORM在GoLang中循环二叉树》，以下内容将会涉及到，若是在学习中对其中部分知识点有疑问，或许看了本文就能帮到你！

我正在尝试使用 gorm (gorm.io/gorm v1.21.9) 在 golang (go 1.16) 中对数据库进行建模。

我的架构的一部分包括带有 2 个可选外键的表 a，实际上每条记录仅使用 1 个外键。表 b 包含对表 a 的 2 个引用，就像二叉树一样。表c是叶节点。

图 1. 实体关系图

图 2. 用法示例

我的 gorm 代码设置如下。

type A struct {
    ID   string
    BID  *string
    B    *B
    CID  *string
    C    *C
}

type B struct {
    ID       string
    LeftAID  string
    LeftA    A
    RightAID string
    RightA   A
}

type C struct {
    ID string
}

当我使用示例数据加载应用程序时，表 b 的外键始终为空值。

但是，表 a 的每条记录都有一个 table_b_id 值（当 table_c_id 存在时应为 null）。

我猜测，由于表 a 具有表 b 的外键，因此它在所有情况下都使用该关联，而不是选取 right_a_id 和 left_a_id 关联。是否有 gorm 注释可以解决此问题，或者这是 gorm 固有的限制？

请注意，我已经删除了不相关的架构信息并使用了抽象的表/列名称。

正确答案

也许你应该使用 自引用有一个 Self-Referential Has One 喜欢关注

type a struct {
    id   string
    leftid  *string
    lefta    *a `gorm:"foreignkey:leftid"`
    rightid *string
    righta    *a `gorm:"foreignkey:rightid"`
    leafid *string
    leafa    *a `gorm:"foreignkey:leafid"`
}

测试代码：

func main() {
    db := config.CreateMysql()
    db.AutoMigrate(&A{})
    a1 := A{ID: "a1"}
    a2 := A{ID: "a2"}
    a3 := A{ID: "a3"}
    a4 := A{ID: "a4"}
    a5 := A{ID: "a5"}

    a1.LeftA = &a2
    a1.RightA = &a3

    a4.LeafA = &a5

    db.Create(&a1)
    db.Create(&a4)
}

在数据库中像这样：

今天关于《使用GORM在GoLang中循环二叉树》的内容就介绍到这里了，是不是学起来一目了然！想要了解更多关于的内容请关注golang学习网公众号！