GoLang GORM循环二叉树

type TreeNode struct {
    gorm.Model
    ID       uint
    Value    string
    LeftID   uint
    RightID  uint
    LeftNode *TreeNode `gorm:"foreignkey:LeftID"`
    RightNode *TreeNode `gorm:"foreignkey:RightID"`
}

// 创建循环二叉树节点
func CreateTreeNode(value string, left *TreeNode, right *TreeNode) *TreeNode {
    node := &TreeNode{
        Value:    value,
        LeftNode: left,
        RightNode: right,
    }
    db.Create(node) // 将节点保存到数据库中
    return node
}

// 查询循环二叉树节点
func GetTreeNode(id uint) (*TreeNode, error) {
    var node TreeNode
    if err := db.First(&node, id).Error; err != nil {
        return nil, err
    }
    return &node, nil
}

// 更新循环二叉树节点
func UpdateTreeNode(node *TreeNode) error {
    return db.Save(node).Error
}

// 删除循环二叉树节点
func DeleteTreeNode(node *TreeNode) error {
    return db.Delete(node).Error
}