非常抱歉,我之前的回答有误。确实,在插入节点时需要更新其父节点的子节点列表,包括新增插入节点后的数据库ID。以下是修改后的代码示例:
type NginxResourceGroup struct {
ID int
Pid int
Name string
ChildNodes []NginxResourceGroup // 子节点列表
}
func insertNode(db *gorm.DB, parentNodeID int, nodeName string) error {
var parentNode NginxResourceGroup
// 查询父节点是否存在
if err := db.Where("ID = ?", parentNodeID).First(&parentNode).Error; err != nil {
return errors.New("父节点不存在")
}
newNode := NginxResourceGroup{
Pid: parentNode.ID,
Name: nodeName,
}
// 插入新节点记录到数据库
if err := db.Create(&newNode).Error; err != nil {
return err
}
// 更新父节点的子节点列表,添加新节点
parentNode.ChildNodes = append(parentNode.ChildNodes, newNode)
if err := db.Save(&parentNode).Error; err != nil {
return err
}
return nil
}
在上述代码中,在 insertNode()
函数中,首先查询到指定的父节点,并创建一个新的子节点对象。然后将该子节点插入数据库,并将其添加到父节点的子节点列表中。最后保存父节点以更新其子节点列表。
请根据你的具体数据结构和需求进行适当修改和调整。
内容由零声教学AI助手提供,问题来源于学员提问