added support for $allElementsTrue and $anyElementTrue to aggregation builder.

Author: craiggwilson

src/MongoDB.Driver.Tests/Linq/Translators/AggregateProjectionTranslatorTests_Project.cs

@@ -107,6 +107,26 @@ public async Task Should_translate_add_flattened()
             result.Value.Result.Should().Be(43);
         }
 
+        [Test]
+        public async Task Should_translate_allElementsTrue()
+        {
+            var result = await Project(x => new { Result = x.G.All(g => g.E.F > 30) });
+
+            result.Projection.Should().Be("{ Result: { \"$allElementsTrue\" : { \"$map\": { input: \"$G\", as: \"g\", in: { \"$gt\": [\"$$g.E.F\", 30 ] } } } }, _id: 0 }");
+
+            result.Value.Result.Should().BeTrue();
+        }
+
+        [Test]
+        public async Task Should_translate_anyElementTrue()
+        {
+            var result = await Project(x => new { Result = x.G.Any(g => g.E.F > 40) });
+
+            result.Projection.Should().Be("{ Result: { \"$anyElementTrue\" : { \"$map\": { input: \"$G\", as: \"g\", in: { \"$gt\": [\"$$g.E.F\", 40 ] } } } }, _id: 0 }");
+
+            result.Value.Result.Should().BeTrue();
+        }
+
         [Test]
         public async Task Should_translate_and()
         {

src/MongoDB.Driver/Linq/Translators/AggregateProjectionTranslator.cs

@@ -419,6 +419,20 @@ private bool TryBuildLinqMethodCall(MethodCallExpression node, out BsonValue res
                 result = null;
                 switch (node.Method.Name)
                 {
+                    case "All":
+                        if(TryBuildMap(node, out result))
+                        {
+                            result = new BsonDocument("$allElementsTrue", result);
+                            return true;
+                        }
+                        break;
+                    case "Any":
+                        if(TryBuildMap(node, out result))
+                        {
+                            result = new BsonDocument("$anyElementTrue", result);
+                            return true;
+                        }
+                        break;
                     case "Count":
                     case "LongCount":
                         if (node.Arguments.Count == 1)
@@ -450,29 +464,8 @@ private bool TryBuildLinqMethodCall(MethodCallExpression node, out BsonValue res
                         }
                         break;
                     case "Select":
-                        var sourceSerializationExpression = node.Arguments[0] as IBsonSerializationInfoExpression;
-                        if (sourceSerializationExpression != null)
+                        if(TryBuildMap(node, out result))
                         {
-                            var lambda = MongoExpressionVisitor.GetLambda(node.Arguments[1]);
-                            if (lambda.Body is IBsonSerializationInfoExpression)
-                            {
-                                result = ResolveValue(lambda.Body);
-                                return true;
-                            }
-
-                            var inputValue = ResolveValue(node.Arguments[0]);
-                            var asValue = lambda.Parameters[0].Name;
-
-                            // HACK: need to add a leading $ sign to the replacement because of how we resolve values.
-                            var body = FieldNameReplacer.Replace(lambda.Body, sourceSerializationExpression.SerializationInfo.ElementName, "$" + asValue);
-                            var inValue = ResolveValue(body);
-
-                            result = new BsonDocument("$map", new BsonDocument
-                            {
-                                { "input", inputValue },
-                                { "as", asValue },
-                                { "in", inValue }
-                            });
                             return true;
                         }
                         break;
@@ -492,6 +485,38 @@ private bool TryBuildLinqMethodCall(MethodCallExpression node, out BsonValue res
                 return false;
             }
 
+            private bool TryBuildMap(MethodCallExpression node, out BsonValue result)
+            {
+                result = null;
+                var sourceSerializationExpression = node.Arguments[0] as IBsonSerializationInfoExpression;
+                if (sourceSerializationExpression != null)
+                {
+                    var lambda = MongoExpressionVisitor.GetLambda(node.Arguments[1]);
+                    if (lambda.Body is IBsonSerializationInfoExpression)
+                    {
+                        result = ResolveValue(lambda.Body);
+                        return true;
+                    }
+
+                    var inputValue = ResolveValue(node.Arguments[0]);
+                    var asValue = lambda.Parameters[0].Name;
+
+                    // HACK: need to add a leading $ sign to the replacement because of how we resolve values.
+                    var body = FieldNameReplacer.Replace(lambda.Body, sourceSerializationExpression.SerializationInfo.ElementName, "$" + asValue);
+                    var inValue = ResolveValue(body);
+
+                    result = new BsonDocument("$map", new BsonDocument
+                            {
+                                { "input", inputValue },
+                                { "as", asValue },
+                                { "in", inValue }
+                            });
+                    return true;
+                }
+
+                return false;
+            }
+
             private bool TryBuildStringMethodCall(MethodCallExpression node, out BsonValue result)
             {
                 result = null;