在 CosmosDb 上的 Linq.Where() 中使用 Linq.Any()

Use Linq.Any() inside a Linq.Where() on CosmosDb

我正在尝试将 .Any() 嵌套在 .Where() 子句中以查询本地 CosmosDb 模拟器。

代码如下所示;其中 permittedStundentIds 是变量 (List<long>),a 是 CosmosDb

中的 Document
.Where(a => permittedStudentIds.Any(sId => a.Students.Any(s => s.Id == sId)));

当我执行查询时,出现错误:

Method 'Any' is not supported. ActivityId: 800000a8-0002-d600-b63f-84710c7967bb, documentdb-dotnet-sdk/1.22.0 Host/64-bit MicrosoftWindowsNT/10.0.16299.0

我尝试了多种变体来获得等效的表达式,但无济于事。唯一有效的方法是使用 .Contains() 并对学生索引进行硬编码;这是不可行的,因为可能不知道学生的数量。

.Where(a => permittedStudentIds.Contains(a.Students[0].Id));

我知道 CosmosDb 的 Sql API 尚不支持某些 lambda 扩展,但是否有解决方法?

在尝试了各种 lambda 表达式的无数组合之后,这就是对我有用的方法。

我在 DocumentModel class 中添加了 StudentIds 属性;冗余但仅用于过滤。

此后,我 OR-ed 使用 .Contains() 进行查询,如下所示:

Expression<Func<MyDocumentModel, bool>> query = a => a.StudentIds.Contains(permittedStudentIds[0]);
foreach (var id in permittedStudentIds.Skip(1))
{
    query = query.Or(a => a.StudentIds.Contains(id));
}

然后使用如下查询:

.Where(query);

对于 query.Or() 部分,我使用了以下 classes:

// See: https://blogs.msdn.microsoft.com/meek/2008/05/02/linq-to-entities-combining-predicates/
public static class ExpressionExtensions
{
    public static Expression<T> Compose<T>(this Expression<T> first, Expression<T> second, Func<Expression, Expression, Expression> merge)
    {
        // build parameter map (from parameters of second to parameters of first)
        var map = first.Parameters.Select((f, i) => new { f, s = second.Parameters[i] }).ToDictionary(p => p.s, p => p.f);

        // replace parameters in the second lambda expression with parameters from the first
        var secondBody = ParameterVistor.ReplaceParameters(map, second.Body);

        // apply composition of lambda expression bodies to parameters from the first expression 
        return Expression.Lambda<T>(merge(first.Body, secondBody), first.Parameters);
    }

    public static Expression<Func<T, bool>> And<T>(this Expression<Func<T, bool>> first, Expression<Func<T, bool>> second)
    {
        return first.Compose(second, Expression.AndAlso);
    }

    public static Expression<Func<T, bool>> Or<T>(this Expression<Func<T, bool>> first, Expression<Func<T, bool>> second)
    {
        return first.Compose(second, Expression.OrElse);
    }
}


public class ParameterVistor : ExpressionVisitor
{
    private readonly Dictionary<ParameterExpression, ParameterExpression> map;

    public ParameterVistor(Dictionary<ParameterExpression, ParameterExpression> map)
    {
        this.map = map ?? new Dictionary<ParameterExpression, ParameterExpression>();
    }

    public static Expression ReplaceParameters(Dictionary<ParameterExpression, ParameterExpression> map, Expression exp)
    {
        return new ParameterVistor(map).Visit(exp);
    }

    protected override Expression VisitParameter(ParameterExpression p)
    {
        ParameterExpression replacement;
        if (map.TryGetValue(p, out replacement))
        {
            p = replacement;
        }
        return base.VisitParameter(p);
    }
}

所以你有一系列 permittedStudentIds 和一个包含一系列学生的文档。每个学生都有一个Id。

您想知道是否有任何 permittedStudentsId 也是您文档中一个(或多个)学生的 ID。

换句话说,如果 permittedStudentIds 的值为 1、2,您想知道 Document.Students 中是否有 ID 为 1 或 2 的 Student。

为什么不提取所有学生的 ID,将它们与您允许的 StudentIds 相交,看看结果是否为空?

var studentIds = Document.Students.Select(student => student.Id);
var intersection = studentIds.Intersect(permittedStudentIds);
var result = intersection.Any();
// TODO: make one statement.

如果两个序列都是 AsQueryable,这会起作用,但如果您的 Document.Students 是 IQueryable 并且您的 permittedStudentIds 是 IEnumerable,它也应该起作用。我最好的猜测是这将成为一个 SQL 包含。参见 Queryable.Intersect