EF Core Code First如何根据OperatorId动态切换MainData查询逻辑？

需求回顾 #

现有两张数据库表：

• MainData：包含Name、EmailAddress、Id字段
• Operator：包含Name、EmailAddress、OperatorId字段

还有一个OperatorService服务，其OperatorId为可空long类型，查询逻辑要求：

• 当OperatorId为null时，直接查询MainData的所有字段映射到MainData对象
• 当OperatorId不为null时，查询结果的Id取自MainData，Name和EmailAddress取自对应OperatorId的Operator表数据

要求通过一处配置，让所有MainData实体查询自动应用该逻辑，当前使用EF Core 3.1，即将升级到.NET 7，可通过Code First实现，以下是分版本的具体方案：

1. 基础实体类定义 #

先确保实体类与表结构对应：

public class MainData
{
    public long Id { get; set; }
    public string Name { get; set; }
    public string EmailAddress { get; set; }
}

public class Operator
{
    public long OperatorId { get; set; }
    public string Name { get; set; }
    public string EmailAddress { get; set; }
}

在DbContext中添加对应DbSet：

public class AppDbContext : DbContext
{
    public DbSet<MainData> MainData { get; set; }
    public DbSet<Operator> Operators { get; set; }

    // 构造函数及数据库连接配置...
}

2. EF Core 3.1 兼容方案：封装全局查询扩展 #

由于EF Core 3.1不支持在全局查询过滤器中注入动态服务参数，最稳妥的方式是封装统一的查询扩展方法，确保所有查询都走该方法：

public static class MainDataQueryExtensions
{
    public static IQueryable<MainData> GetMainData(this AppDbContext dbContext, long? operatorId)
    {
        if (!operatorId.HasValue)
        {
            return dbContext.MainData.Select(m => new MainData
            {
                Id = m.Id,
                Name = m.Name,
                EmailAddress = m.EmailAddress
            });
        }
        else
        {
            return dbContext.MainData
                .Join(dbContext.Operators.Where(o => o.OperatorId == operatorId),
                    _ => 1, _ => 1, (m, o) => new MainData
                    {
                        Id = m.Id,
                        Name = o.Name,
                        EmailAddress = o.EmailAddress
                    });
        }
    }
}

使用时直接调用dbContext.GetMainData(operatorService.OperatorId)即可，所有查询都会自动应用逻辑，实现一处配置。

3. .NET 7+ 优雅方案：利用ToQuery全局映射 #

升级到.NET 7后，EF Core支持在实体配置中使用ToQuery方法，结合依赖注入的服务，实现完全自动的全局查询逻辑：

修改DbContext，注入OperatorService并配置实体映射：

public class AppDbContext : DbContext
{
    private readonly OperatorService _operatorService;

    public AppDbContext(DbContextOptions<AppDbContext> options, OperatorService operatorService)
        : base(options)
    {
        _operatorService = operatorService;
    }

    public DbSet<MainData> MainData { get; set; }
    public DbSet<Operator> Operators { get; set; }

    protected override void OnModelCreating(ModelBuilder modelBuilder)
    {
        modelBuilder.Entity<MainData>().ToQuery(() =>
        {
            var operatorId = _operatorService.OperatorId;
            if (!operatorId.HasValue)
            {
                return MainData.Select(m => new MainData
                {
                    Id = m.Id,
                    Name = m.Name,
                    EmailAddress = m.EmailAddress
                });
            }
            else
            {
                return MainData.Join(Operators.Where(o => o.OperatorId == operatorId),
                    _ => 1, _ => 1, (m, o) => new MainData
                    {
                        Id = m.Id,
                        Name = o.Name,
                        EmailAddress = o.EmailAddress
                    });
            }
        });
    }
}

配置完成后，直接查询dbContext.MainData即可自动应用逻辑，无需额外调用扩展方法，完全实现全局自动映射。

补充说明 #

• Code First模式下，上述实体类配置可直接生成对应数据库表，若需要自定义主键等规则，可在OnModelCreating中添加Fluent API配置。
• 测试时可验证两种场景生成的SQL：
  • OperatorId为null时，生成SELECT Name, EmailAddress, Id FROM MainData
  • OperatorId不为null时，生成关联查询，取MainData.Id和Operator的Name、EmailAddress

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