Ent-Go框架Postgres下O2M关联Upsert重复创建及关联NULL问题

核心问题分析 #

1. 子表关联字段为NULL的根源：第二次执行upsert时，子表数据未正确绑定主表实例，导致ent自动生成的外键字段virtual_machine_id为NULL。PostgreSQL中NULL不参与唯一约束匹配，直接绕过了你设置的key+sysId+virtualMachine唯一索引，最终生成重复数据。
2. ent-go O2M关联upsert的误区：单独对子表执行upsert时，若未通过主表关联操作器绑定关系，ent无法自动填充外键字段，唯一约束自然无法触发。

具体修复步骤 #

1. 确认子表唯一索引的正确性 #

检查ent schema中子表的索引定义，确保使用的是数据库实际字段名（ent会将关联边virtualMachine映射为蛇形命名的virtual_machine_id）：

// 在CmdbCiServerVmwareVirtualMachineCustomField的schema文件中
func (CmdbCiServerVmwareVirtualMachineCustomField) Indexes() []ent.Index {
    return []ent.Index{
        Index{
            Name:    "custom_field_unique_constraint",
            Columns: []string{"key", "sys_id", "virtual_machine_id"}, // 必须用数据库实际字段名
            Unique:  true,
        },
    }
}

2. 通过主表关联操作器执行子表upsert #

不要单独操作子表，而是借助主表的关联方法绑定外键，确保子表数据正确关联主表：

func UpsertVMWithCustomFields(ctx context.Context, client *ent.Client, vmID string, customFields []CustomField) error {
    // 先执行主表的upsert
    vm, err := client.CmdbCiServerVmwareVirtualMachine.
        UpsertOneID(vmID). // 假设主表以sys_id作为主键
        SetName("示例虚拟机"). // 替换为你的主表字段设置
        OnConflictColumns("sys_id").
        UpdateNewValues().
        Save(ctx)
    if err != nil {
        return err
    }

    // 遍历子表数据，通过主表关联操作器执行upsert
    for _, cf := range customFields {
        _, err := vm.UpdateCmdbCiServerVmwareVirtualMachineCustomFields().
            Upsert().
            SetKey(cf.Key).
            SetValue(cf.Value).
            SetSysID(vmID).
            OnConflictColumns("key", "sys_id", "virtual_machine_id"). // 匹配唯一索引
            UpdateNewValues(). // 冲突时更新新值，可按需改为SetXXX指定更新字段
            Save(ctx)
        if err != nil {
            return err
        }
    }
    return nil
}

关键：通过vm.UpdateCmdbCiServerVmwareVirtualMachineCustomFields()获取关联操作器后，ent会自动填充virtual_machine_id字段，唯一约束就能正常触发冲突检测。

3. 清理已存在的无效重复数据 #

如果数据库中已经存在virtual_machine_id为NULL的重复子表数据，先执行清理：

-- PostgreSQL清理脚本
DELETE FROM cmdb_ci_server_vmware_virtual_machine_custom_field
WHERE virtual_machine_id IS NULL
AND EXISTS (
    SELECT 1 FROM cmdb_ci_server_vmware_virtual_machine_custom_field t2
    WHERE t2.key = cmdb_ci_server_vmware_virtual_machine_custom_field.key
    AND t2.sys_id = cmdb_ci_server_vmware_virtual_machine_custom_field.sys_id
    AND t2.virtual_machine_id IS NOT NULL
);

4. 重新生成ent代码 #

修改schema后，执行代码生成命令确保变更生效：

go generate ./ent

常见误区提醒 #

• 禁止直接调用client.CmdbCiServerVmwareVirtualMachineCustomField.UpsertOne()，除非手动指定VirtualMachineID，否则外键会为NULL。
• PostgreSQL唯一约束中，NULL值被视为互不相等，所以只要virtual_machine_id为NULL，即使key和sys_id完全相同，也不会触发冲突。

内容的提问来源于stack exchange，提问作者Rygo