feat: Add pdb.all (#57)

Author: isaacvando

src/AllQuery.cs

@@ -0,0 +1,9 @@
+using System.Diagnostics.CodeAnalysis;
+
+namespace ParadeDB.EntityFrameworkCore;
+
+[ExcludeFromCodeCoverage]
+public abstract class AllQuery : PdbQuery
+{
+    private protected AllQuery() { }
+}

src/Extensions/ParadeDbFunctionsExtensions.cs

@@ -80,11 +80,8 @@ int maxNumChars
     ) => throw new InvalidOperationException(CoreStrings.FunctionOnClient(nameof(Snippet)));
 
     [DbFunction]
-    public static bool Proximity<TProperty>(
-        this DbFunctions _,
-        TProperty property,
-        PdbQuery query
-    ) => throw new InvalidOperationException(CoreStrings.FunctionOnClient(nameof(Proximity)));
+    public static bool Query<TProperty>(this DbFunctions _, TProperty property, PdbQuery query) =>
+        throw new InvalidOperationException(CoreStrings.FunctionOnClient(nameof(Query)));
 
     [DbFunction]
     public static string Tokenize<TProperty>(

src/Internal/Query/Translator.cs

@@ -63,11 +63,18 @@ IRelationalTypeMappingSource typeMappingSource
                 argumentsPropagateNullability: [false],
                 returnType: typeof(float)
             ),
-            nameof(ParadeDbFunctionsExtensions.Snippet) => BuildSnippet(arguments),
-            nameof(ParadeDbFunctionsExtensions.Proximity)
-                when method.DeclaringType == typeof(ParadeDbFunctionsExtensions) => BuildProximity(
-                arguments
+            nameof(Pdb.All) => _sqlExpressionFactory.Function(
+                name: "all",
+                schema: "pdb",
+                nullable: false,
+                arguments: [],
+                argumentsPropagateNullability: [],
+                returnType: typeof(bool)
             ),
+            nameof(ParadeDbFunctionsExtensions.Snippet) => BuildSnippet(arguments),
+            nameof(ParadeDbFunctionsExtensions.Query)
+                when method.DeclaringType == typeof(ParadeDbFunctionsExtensions) =>
+                BuildQueryBuilder(arguments),
             nameof(Pdb.Proximity) => _sqlExpressionFactory.ApplyDefaultTypeMapping(arguments[0]),
             nameof(Pdb.ProximityRegex) => BuildProximityFunction("prox_regex", arguments),
             nameof(Pdb.ProximityArray) => BuildProximityFunction("prox_array", arguments),
@@ -163,7 +170,7 @@ PdbOperatorType operatorType
         );
     }
 
-    private SqlExpression? BuildProximity(IReadOnlyList<SqlExpression> arguments)
+    private SqlExpression? BuildQueryBuilder(IReadOnlyList<SqlExpression> arguments)
     {
         var column = _sqlExpressionFactory.ApplyDefaultTypeMapping(arguments[1]);
         var query = _sqlExpressionFactory.ApplyDefaultTypeMapping(arguments[2]);

src/Pdb.cs

@@ -6,7 +6,7 @@
 namespace ParadeDB.EntityFrameworkCore;
 
 [ExcludeFromCodeCoverage]
-public static partial class Pdb
+public static class Pdb
 {
     public static Boost Boost(float factor) => new(factor);
 
@@ -68,4 +68,7 @@ public static PdbProximityQuery ProximityArray(params string[] tokens) =>
 
     public static PdbProximityQuery ProximityArray(params PdbProximityQuery[] operands) =>
         throw new InvalidOperationException(CoreStrings.FunctionOnClient(nameof(ProximityArray)));
+
+    public static AllQuery All() =>
+        throw new InvalidOperationException(CoreStrings.FunctionOnClient(nameof(AllQuery)));
 }

tests/QueryTests.cs

@@ -586,7 +586,7 @@ public async Task Proximity()
         await using var context = DbFixture.CreateContext();
 
         var query = context.MockItems.Where(p =>
-            EF.Functions.Proximity(p.Description, Pdb.Proximity("sleek").Within(1, "shoes"))
+            EF.Functions.Query(p.Description, Pdb.Proximity("sleek").Within(1, "shoes"))
         );
 
         var sql = """
@@ -605,7 +605,7 @@ public async Task Proximity_Ordered()
         await using var context = DbFixture.CreateContext();
 
         var query = context.MockItems.Where(p =>
-            EF.Functions.Proximity(
+            EF.Functions.Query(
                 p.Description,
                 Pdb.Proximity("sleek").Within(1, "shoes", ordered: true)
             )
@@ -631,7 +631,7 @@ public async Task Proximity_WithVariableArguments()
         int distance = 1;
 
         var query = context.MockItems.Where(p =>
-            EF.Functions.Proximity(p.Description, Pdb.Proximity(left).Within(distance, right))
+            EF.Functions.Query(p.Description, Pdb.Proximity(left).Within(distance, right))
         );
 
         var sql = """
@@ -653,7 +653,7 @@ public async Task Proximity_WithRegex()
         await using var context = DbFixture.CreateContext();
 
         var query = context.MockItems.Where(p =>
-            EF.Functions.Proximity(p.Description, Pdb.ProximityRegex("sl.*").Within(1, "shoes"))
+            EF.Functions.Query(p.Description, Pdb.ProximityRegex("sl.*").Within(1, "shoes"))
         );
 
         var sql = """
@@ -674,7 +674,7 @@ public async Task Proximity_WithRegexVariablePattern()
         string pattern = "sl.*";
 
         var query = context.MockItems.Where(p =>
-            EF.Functions.Proximity(p.Description, Pdb.ProximityRegex(pattern).Within(1, "shoes"))
+            EF.Functions.Query(p.Description, Pdb.ProximityRegex(pattern).Within(1, "shoes"))
         );
 
         var sql = """
@@ -694,10 +694,7 @@ public async Task Proximity_WithRegexAndMaxExpansions()
         await using var context = DbFixture.CreateContext();
 
         var query = context.MockItems.Where(p =>
-            EF.Functions.Proximity(
-                p.Description,
-                Pdb.ProximityRegex("sl.*", 100).Within(1, "shoes")
-            )
+            EF.Functions.Query(p.Description, Pdb.ProximityRegex("sl.*", 100).Within(1, "shoes"))
         );
 
         var sql = """
@@ -716,7 +713,7 @@ public async Task Proximity_WithArrayOfTokens()
         await using var context = DbFixture.CreateContext();
 
         var query = context.MockItems.Where(p =>
-            EF.Functions.Proximity(
+            EF.Functions.Query(
                 p.Description,
                 Pdb.ProximityArray("sleek", "white").Within(1, "shoes")
             )
@@ -741,7 +738,7 @@ public async Task Proximity_WithArrayOfVariableTokens()
         string t2 = "white";
 
         var query = context.MockItems.Where(p =>
-            EF.Functions.Proximity(p.Description, Pdb.ProximityArray(t1, t2).Within(1, "shoes"))
+            EF.Functions.Query(p.Description, Pdb.ProximityArray(t1, t2).Within(1, "shoes"))
         );
 
         var sql = """
@@ -762,7 +759,7 @@ public async Task Proximity_WithArrayOfMixedOperands()
         await using var context = DbFixture.CreateContext();
 
         var query = context.MockItems.Where(p =>
-            EF.Functions.Proximity(
+            EF.Functions.Query(
                 p.Description,
                 Pdb.ProximityArray(Pdb.ProximityRegex("sl.*"), Pdb.ProximityArray("white"))
                     .Within(1, "shoes")
@@ -785,7 +782,7 @@ public async Task Proximity_Chained()
         await using var context = DbFixture.CreateContext();
 
         var query = context.MockItems.Where(p =>
-            EF.Functions.Proximity(
+            EF.Functions.Query(
                 p.Description,
                 Pdb.Proximity("sleek").Within(1, "running").Within(2, "shoes")
             )
@@ -825,6 +822,23 @@ FROM mock_items AS m
         await query.ToListAsync();
     }
 
+    [Test]
+    public async Task All()
+    {
+        await using var context = DbFixture.CreateContext();
+
+        var query = context.MockItems.Where(p => EF.Functions.Query(p.Id, Pdb.All()));
+
+        var sql = """
+            SELECT m.id, m.category, m.created_at, m.description, m.in_stock, m.last_updated_date, m.latest_available_time, m.metadata, m.rating, m.weight_range
+            FROM mock_items AS m
+            WHERE m.id @@@ pdb."all"()
+            """;
+
+        AssertSql(query, sql);
+        await query.ToListAsync();
+    }
+
     [Test]
     public async Task Snippet()
     {