如何将SQL Server高开销查询关联到C#/EF Core生成代码？

针对你提出的需求，以下是成熟且易实现的解决方案：

一、给SQL添加自定义意图注释（关联Azure查询到代码意图） #

EF Core没有内置的WithIntent方法，但可以通过扩展方法+命令拦截器实现，让生成的SQL带上自定义注释，方便在Azure门户中识别并关联代码：

1. 定义DbContext扩展方法 #

用于存储查询意图：

public static class DbContextExtensions
{
    private const string QueryIntentKey = "QueryIntent";

    public static DbContext WithIntent(this DbContext context, string intent)
    {
        context.Items[QueryIntentKey] = intent;
        return context;
    }

    internal static string? GetIntent(this DbContext context)
    {
        context.Items.TryGetValue(QueryIntentKey, out var intent);
        return intent as string;
    }
}

2. 实现SQL注释拦截器 #

在命令执行前将注释注入SQL：

public class QueryIntentInterceptor : DbCommandInterceptor
{
    public override InterceptionResult<DbDataReader> ReaderExecuting(
        DbCommand command,
        CommandEventData eventData,
        InterceptionResult<DbDataReader> result)
    {
        AddIntentComment(command, eventData.Context);
        return base.ReaderExecuting(command, eventData, result);
    }

    public override ValueTask<InterceptionResult<DbDataReader>> ReaderExecutingAsync(
        DbCommand command,
        CommandEventData eventData,
        InterceptionResult<DbDataReader> result,
        CancellationToken cancellationToken = default)
    {
        AddIntentComment(command, eventData.Context);
        return base.ReaderExecutingAsync(command, eventData, result, cancellationToken);
    }

    private void AddIntentComment(DbCommand command, DbContext? context)
    {
        if (context == null) return;
        
        var intent = context.GetIntent();
        if (!string.IsNullOrEmpty(intent))
        {
            // 转义注释中的特殊字符，避免破坏SQL结构
            var safeIntent = intent.Replace("--", "---").Replace("\n", " ");
            command.CommandText = $"--{safeIntent}\n{command.CommandText}";
            // 清空意图，防止后续查询复用
            context.Items.Remove(DbContextExtensions.QueryIntentKey);
        }
    }
}

3. 注册拦截器 #

在DbContext配置中添加拦截器：

builder.Services.AddDbContext<MyDbContext>(options =>
{
    options.UseSqlServer(Configuration.GetConnectionString("Default"))
           .AddInterceptors(new QueryIntentInterceptor());
});

使用方式 #

代码中调用WithIntent即可生成带注释的SQL：

db.WithIntent("Cache the whole SomeMassiveTable locally into redis")
  .SomeMassiveTable.ToList();

生成的SQL会包含注释，在Azure门户的慢查询列表中可以直接看到，通过注释文本就能快速搜索到对应代码位置。

二、拦截无WHERE子句的查询并记录调用栈 #

通过自定义拦截器检测无过滤的查询，在本地开发环境记录调用栈，直接定位代码：

实现查询拦截器 #

public class UnfilteredQueryInterceptor : DbCommandInterceptor
{
    private readonly ILogger<UnfilteredQueryInterceptor> _logger;

    public UnfilteredQueryInterceptor(ILogger<UnfilteredQueryInterceptor> logger)
    {
        _logger = logger;
    }

    public override InterceptionResult<DbDataReader> ReaderExecuting(
        DbCommand command,
        CommandEventData eventData,
        InterceptionResult<DbDataReader> result)
    {
        CheckUnfilteredQuery(command);
        return base.ReaderExecuting(command, eventData, result);
    }

    public override ValueTask<InterceptionResult<DbDataReader>> ReaderExecutingAsync(
        DbCommand command,
        CommandEventData eventData,
        InterceptionResult<DbDataReader> result,
        CancellationToken cancellationToken = default)
    {
        CheckUnfilteredQuery(command);
        return base.ReaderExecutingAsync(command, eventData, result, cancellationToken);
    }

    private void CheckUnfilteredQuery(DbCommand command)
    {
        var sql = command.CommandText.Trim();
        // 精准匹配：针对SomeMassiveTable的无WHERE查询
        if (sql.StartsWith("SELECT", StringComparison.OrdinalIgnoreCase) &&
            sql.Contains("* FROM SomeMassiveTable", StringComparison.OrdinalIgnoreCase) &&
            // 忽略注释中的WHERE关键字
            !sql.Split(new[] {"--"}, StringSplitOptions.RemoveEmptyEntries)[0]
                .Contains("WHERE", StringComparison.OrdinalIgnoreCase))
        {
            // 生成带文件行号的调用栈
            var stackTrace = new StackTrace(true);
            _logger.LogWarning(
                "Unfiltered query on SomeMassiveTable detected!\nSQL: {Sql}\nCall Stack:\n{StackTrace}",
                sql,
                stackTrace.ToString());
        }
    }
}

注册拦截器 #

将拦截器加入DbContext配置，仅限开发环境启用：

if (builder.Environment.IsDevelopment())
{
    builder.Services.AddDbContext<MyDbContext>(options =>
    {
        options.UseSqlServer(Configuration.GetConnectionString("Default"))
               .AddInterceptors(new UnfilteredQueryInterceptor(
                   builder.Services.BuildServiceProvider().GetRequiredService<ILogger<UnfilteredQueryInterceptor>>()));
    });
}

本地运行时，只要触发无过滤查询，日志就会输出详细调用栈，直接定位到代码行。

三、EF Core内置日志辅助定位 #

在开发环境开启EF Core的命令日志，也能间接关联查询到代码：

// Program.cs中配置日志过滤
builder.Logging.AddFilter("Microsoft.EntityFrameworkCore.Database.Command", LogLevel.Debug);

开启后，日志会输出每个查询的执行时间、SQL语句，结合日志的调用栈信息（需日志框架支持），也能大致定位查询的生成位置，但精度不如自定义拦截器。

总结 #

EF Core没有完全内置的"一键关联"功能，但通过自定义拦截器是最成熟、灵活的解决方案：

• 生产环境用QueryIntentInterceptor给SQL加注释，在Azure中关联代码意图；
• 开发环境用UnfilteredQueryInterceptor直接捕获无过滤查询并输出调用栈，快速定位代码；
• 内置日志作为辅助手段，适合快速排查简单问题。

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