如何在 EF Core 中实现 Select For Update

据我所知，EF（和 EF Core）中没有选项可以显式锁定我正在查询的资源，但我经常需要此功能，并且不想再重新编写每次需要时选择语句。

因为我只需要它用于 postgres 和根据规格 FOR UPDATE是查询中的最后一项，我想到的最简单的实现方法是获取如下所述的 select 语句：在 Linq to Entities 中，您可以将 IQueryable 转换为 SQL 字符串吗？并附加FOR UPDATE并直接执行。然而，这要么给我一个带有参数占位符的查询，要么不是一个准备好的查询，这意味着执行计划的缓存在 postgres 上不起作用，所以无论哪种方式都是不行的。

Linq to SQL 有方法DataContext.GetCommand但 EF 中似乎没有任何等效的东西，特别是 EF Core。我还查看了 EntityFramework.Extended 及其批量更新/删除，但由于他们必须将 select 语句转换为不同的语句，因此他们需要处理比我复杂得多的问题，因此我希望有一个更简单的解决方案。

Update:

如果描述不清楚，我想创建一个像这样的扩展方法：

public static IList<T> ForUpdate (this IQueryable<T> me)
{
    // this line is obviously what is missing for me :)
    var theUnderlyingCommand = me.GetTheUnderlyingDbCommandOrSimilar();

    theUnderlyingCommand.Text += "FOR UPDATE";
    return me.ToList();
}

这样，其他开发人员就可以像所有其他过程一样通过 Linq 使用 EF，而不是运行.ToList()他们会跑.ForUpdate()。 （对于Update故意执行查询是为了让实现更容易，也因为FOR UPDATE是 postgres 支持的最后一个选项，之后就不应该再有任何其他选项了）

这对我使用 SQL Server 有效（没有测试过的异步方法）：

首先，创建一个DbCommandInterceptor（我称之为HintInterceptor.cs）

using System;
using System.Data.Common;
using System.Data.Entity.Infrastructure.Interception;
using System.Text.RegularExpressions;

public class HintInterceptor : DbCommandInterceptor
{
    private static readonly Regex _tableAliasRegex = new Regex(@"(?<tableAlias>FROM +(\[.*\]\.)?(\[.*\]) AS (\[.*\])(?! WITH \(\*HINT\*\)))", RegexOptions.Multiline | RegexOptions.IgnoreCase | RegexOptions.Compiled);

    [ThreadStatic]
    public static string HintValue;

    private static string Replace(string input)
    {
        if (!String.IsNullOrWhiteSpace(HintValue))
        {
            if (!_tableAliasRegex.IsMatch(input))
            {
                throw new InvalidProgramException("Não foi possível identificar uma tabela para ser marcada para atualização(forupdate)!", new Exception(input));
            }
            input = _tableAliasRegex.Replace(input, "${tableAlias} WITH (*HINT*)");
            input = input.Replace("*HINT*", HintValue);
        }
        HintValue = String.Empty;
        return input;
    }

    public override void ScalarExecuting(DbCommand command, DbCommandInterceptionContext<object> interceptionContext)
    {
        command.CommandText = Replace(command.CommandText);
    }

    public override void ReaderExecuting(DbCommand command, DbCommandInterceptionContext<DbDataReader> interceptionContext)
    {
        command.CommandText = Replace(command.CommandText);
    }
}

所以在 Web.config 中注册你的拦截器类

<entityFramework>
<defaultConnectionFactory type="System.Data.Entity.Infrastructure.SqlConnectionFactory, EntityFramework" />
<providers>
  <provider invariantName="System.Data.SqlClient" type="System.Data.Entity.SqlServer.SqlProviderServices, EntityFramework.SqlServer" />
</providers>
<interceptors> 
  <interceptor type="Full.Path.Of.Class.HintInterceptor, Dll.Name" />
</interceptors>
</entityFramework>

现在我创建一个名为 HintExtension 的静态类

public static class HintExtension
{
    public static IQueryable<T> WithHint<T>(this IQueryable<T> set, string hint) where T : class
    {
        HintInterceptor.HintValue = hint;
        return set;
    }
    public static IQueryable<T> ForUpdate<T>(this IQueryable<T> set) where T : class
    {
        return set.WithHint("UPDLOCK");
    }
}

就这样，我可以在数据库事务中使用，例如：

using(var trans = context.Database.BeginTransaction())
{
        var query = context.mydbset.Where(a => a.name == "asd").ForUpdate();
        // not locked yet
        var mylist = query.ToList();
        // now are locked for update
        // update the props, call saveChanges() and finally call commit ( or rollback)
        trans.Commit();
        // now are unlocked
}

抱歉我的英语不好，希望我的例子能有所帮助。