如何重写HotChocolate生成的过滤LINQ查询以处理大量ID？

问题场景 #

当你通过HotChocolate构建的GraphQL服务需要过滤数万条ID时，默认生成的EF Core查询会用staticListOfIds.Contains(c.Id)的形式，这会带来两个麻烦：

• 性能拉胯：大量ID的Contains会生成带着几百上千参数的SQL，执行计划差，跑起来慢得离谱
• SQL Server限制：SQL Server对IN子句的参数数量有大约32000条的上限，超过直接报错

解决思路 #

用能把内存ID列表转成数据库临时表的NuGet包（比如EFCore.QueryableValues），把原查询里的staticListOfIds.Contains(c.Id)改成_dbContext.AsQueryableValues(staticListOfIds).Contains(c.Id)。核心是重写HotChocolate的中间件，自动把符合条件的Contains表达式替换成临时表查询的形式。

具体操作步骤 #

1. 安装NuGet包 #

先把支持临时表查询的包装上，用.NET CLI的话：

dotnet add package EFCore.QueryableValues

或者用Package Manager：

Install-Package EFCore.QueryableValues

2. 编写自定义表达式访问器 #

这个访问器的作用是拦截EF的表达式树，把针对内存列表的Contains换成临时表的Contains：

public class QueryableValuesRewriter : ExpressionVisitor
{
    private readonly YourDbContext _dbContext;
    private readonly int _threshold = 1000; // 超过这个数量的ID才触发替换，可按需调整

    public QueryableValuesRewriter(YourDbContext dbContext)
    {
        _dbContext = dbContext;
    }

    protected override Expression VisitMethodCall(MethodCallExpression node)
    {
        // 识别Enumerable.Contains的调用（针对内存列表的Contains）
        if (node.Method.DeclaringType == typeof(Enumerable) && 
            node.Method.Name == nameof(Enumerable.Contains) &&
            node.Arguments[0] is ConstantExpression constantExpr)
        {
            var idList = constantExpr.Value as IEnumerable<Guid>; // 替换成你的ID类型，比如int
            if (idList?.Count() > _threshold)
            {
                // 构建_dbContext.AsQueryableValues(idList)的表达式
                var asQueryableValuesMethod = typeof(QueryableValuesExtensions)
                    .GetMethod(nameof(QueryableValuesExtensions.AsQueryableValues))
                    .MakeGenericMethod(typeof(Guid));
                
                var tempTableQuery = Expression.Call(
                    asQueryableValuesMethod,
                    Expression.Constant(_dbContext),
                    Expression.Constant(idList));

                // 构建Queryable.Contains的调用，替换原有的Enumerable.Contains
                var containsMethod = typeof(Queryable)
                    .GetMethods()
                    .First(m => m.Name == nameof(Queryable.Contains) && m.GetParameters().Length == 2)
                    .MakeGenericMethod(typeof(Guid));

                return Expression.Call(containsMethod, tempTableQuery, node.Arguments[1]);
            }
        }

        return base.VisitMethodCall(node);
    }
}

3. 集成到HotChocolate过滤中间件 #

在GraphQL服务配置里，把刚才写的表达式访问器挂到过滤环节：

builder.Services.AddGraphQLServer()
    .AddQueryType<Query>()
    .AddProjections()
    .AddFiltering(options =>
    {
        options.UseExpressionVisitor((context, next) =>
        {
            var dbContext = context.Services.GetRequiredService<YourDbContext>();
            var rewriter = new QueryableValuesRewriter(dbContext);
            context.Expression = rewriter.Visit(context.Expression);
            return next(context);
        });
    });

4. 验证效果 #

改完之后，EF会生成基于临时表的JOIN查询，不再是堆一堆参数的IN子句。既绕开了SQL Server的32000条限制，查询性能也会有明显提升——尤其是ID数量多的时候，临时表的索引会帮你把查询速度拉上来。

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