EF Core PostgreSQL 提供程序中的UPDATE语句生成问题解析

在Entity Framework Core PostgreSQL提供程序（EF Core.PG）中，开发团队发现了一个关于UPDATE语句生成的特定问题。这个问题在执行批量更新操作时出现，特别是当目标表同时作为主表和连接表出现在查询中时。

问题背景

当使用EF Core的ExecuteUpdate/Delete方法进行批量更新时，PostgreSQL提供程序生成的SQL语句会出现表名重复的问题。具体表现为：目标表在UPDATE语句中同时作为主表和连接表出现，导致SQL语法错误。

问题复现

考虑以下测试用例：

public virtual async Task Replace_ColumnExpression_in_column_setter(bool async)
{
    var contextFactory = await InitializeAsync<Context28671>();
    await AssertUpdate(
        async,
        contextFactory.CreateContext,
        ss => ss.Set<Owner>().SelectMany(e => e.OwnedCollections),
        s => s.SetProperty(o => o.Value, "SomeValue"),
        rowsAffectedCount: 0);
}

当前实现会生成如下SQL：

UPDATE "OwnedCollection" AS o0
SET "Value" = 'SomeValue'
FROM "Owner" AS o,
    "OwnedCollection" AS o0
WHERE o."Id" = o0."OwnerId"

这个SQL语句的问题在于"OwnedCollection"表被重复声明了两次（都使用了o0别名），这在PostgreSQL中会导致语法错误。

技术分析

PostgreSQL的UPDATE语法与SQL Server有所不同。在PostgreSQL中：

1. 目标表必须直接跟在UPDATE关键字后面
2. FROM子句用于引入其他需要连接的表
3. 不能像SQL Server那样在FROM子句中指定目标表

相比之下，SQL Server生成的SQL更加简洁：

UPDATE [o0]  
SET [o0].[Value] = N'SomeValue'  
FROM [Owner] AS [o]  
INNER JOIN [OwnedCollection] AS [o0] ON [o].[Id] = [o0].[OwnerId]

解决方案

问题的根源在于EF Core的查询生成器中使用了引用相等性检查来判断表是否相同。在EF Core 9.0中，团队已经开始从引用标识转向值语义，这是为了解决许多引用完整性错误，并实现树的完全不可变性。

具体修复方法是修改查询生成器中的条件判断，从引用相等性检查改为比较表别名：

原始代码：

if (ReferenceEquals(updateExpression.Table, joinExpression?.Table ?? table))
{
    LiftPredicate(table);
    continue;
}

修改后：

if (updateExpression.Table.Alias == (joinExpression?.Table.Alias ?? table.Alias))
{
    LiftPredicate(table);
    continue;
}

这一改变使得当两个表表达式引用相同的表和别名时，即使不是同一个对象实例，也能被正确识别为相同表。

相关数据库行为

值得注意的是，不同数据库对UPDATE语句的处理方式有所不同：

1. SQL Server：支持在FROM子句中指定目标表
2. PostgreSQL/SQLite：目标表必须直接跟在UPDATE后，FROM用于连接其他表
3. Jet数据库：没有FROM子句概念，UPDATE后是整个表表达式

在某些数据库如Jet中，甚至可以实现在单个UPDATE语句中更新多个连接表的数据，这是EF Core目前不支持的语法特性。

总结

这个问题展示了EF Core在不同数据库提供程序间实现SQL生成时的挑战。通过将引用相等性检查改为基于别名的比较，可以更可靠地识别相同的表引用，从而生成正确的SQL语句。这一改进不仅解决了PostgreSQL提供程序中的特定问题，也符合EF Core向值语义转变的长期方向。