如何在运行时定位生成SQL的LINQ语句并自动注入标识

方法一：重写DbContext的Set方法 #

直接在你的TryDbContext中重写Set<T>方法，每次获取实体集合时自动触发SetQueryName扩展：

public class TryDbContext : DbContext
{
    // 你的DbSet定义
    public DbSet<Employee> Employees { get; set; }

    // 重写泛型Set方法
    public override DbSet<T> Set<T>() where T : class
    {
        return base.Set<T>().SetQueryName();
    }

    // 可选：重写非泛型Set方法，覆盖所有实体类型
    public override DbSet Set(Type entityType)
    {
        var originalSet = base.Set(entityType);
        // 通过反射调用SetQueryName扩展
        var method = typeof(ExtensionMethods).GetMethod("SetQueryName")
            .MakeGenericMethod(entityType);
        return (DbSet)method.Invoke(null, new object[] { originalSet });
    }
}

这样无论你调用context.Employees还是context.Set<Employee>()，都会自动带上查询追踪逻辑，无需手动调用扩展方法。

方法二：自定义包装DbSet类 #

如果需要更精细化的控制，可以自定义TrackedDbSet<T>包装原生DbSet：

public class TrackedDbSet<T> : DbSet<T>, IQueryable<T> where T : class
{
    private readonly DbSet<T> _originalSet;

    public TrackedDbSet(DbSet<T> originalSet)
    {
        _originalSet = originalSet;
    }

    // 实现IQueryable接口，自动应用SetQueryName
    public IQueryProvider Provider => _originalSet.SetQueryName().Provider;
    public Expression Expression => _originalSet.SetQueryName().Expression;
    public Type ElementType => _originalSet.SetQueryName().ElementType;

    // 重写DbSet核心操作方法，转发给原生集合
    public override T Add(T entity) => _originalSet.Add(entity);
    public override T Remove(T entity) => _originalSet.Remove(entity);
    public override T Attach(T entity) => _originalSet.Attach(entity);
    // 按需重写其他DbSet方法（如Update、Find等）

    public IEnumerator<T> GetEnumerator() => _originalSet.SetQueryName().GetEnumerator();
    IEnumerator IEnumerable.GetEnumerator() => GetEnumerator();
}

然后在DbContext中替换原生DbSet：

public class TryDbContext : DbContext
{
    private DbSet<Employee> _employees;

    public DbSet<Employee> Employees
    {
        get => _employees ??= new TrackedDbSet<Employee>(base.Set<Employee>());
        set => _employees = value;
    }
}

优化原SetQueryName扩展方法 #

原方法可以合并条件、简化逻辑，同时确保追踪信息能体现在生成的SQL中：

public static class ExtensionMethods
{
    public static IQueryable<T> SetQueryName<T>(this IQueryable<T> source,
        [CallerMemberName] string callerName = null,
        [CallerFilePath] string sourcePath = "",
        [CallerLineNumber] int lineNumber = 0)
    {
        var traceInfo = $"QuerySource: {callerName} | File: {Path.GetFileName(sourcePath)} | Line: {lineNumber}";
        // 创建恒真条件（不影响查询结果，但会被EF带入SQL）
        var conditionExpr = Expression.NotEqual(
            Expression.Constant(traceInfo),
            Expression.Constant(null, typeof(string))
        );

        var param = Expression.Parameter(typeof(T), "t");
        var lambda = Expression.Lambda<Func<T, bool>>(conditionExpr, param);

        // 输出追踪信息到控制台或日志系统
        Console.WriteLine(traceInfo);
        return source.Where(lambda);
    }
}

优化后，EF生成的SQL会包含类似WHERE N'QuerySource: GetAllEmployees | File: EmployeeService.cs | Line: 15' IS NOT NULL的条件，你可以通过SQL Profiler或EF日志直接定位到出错查询的调用位置。

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