chore: updated matches implementation for visual fetch

Bendomey · Feb 25, 2025 · f229451 · f229451
1 parent 8c5edcb
commit f229451
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 16 deletions.
diff --git a/services/main/Controllers/Content.cs b/services/main/Controllers/Content.cs
@@ -294,7 +294,7 @@ public async Task<IActionResult> VisualSearch(
 
             var matches = await _searchContentService.AskRekognitionForMatch(media);
 
-            if (matches == null)
+            if (matches == null || matches.Length == 0)
             {
                 var res = HttpLib.GeneratePagination<OutputContent, Models.Content>(
                     [],

diff --git a/services/main/Domains/Content/SearchContentService.cs b/services/main/Domains/Content/SearchContentService.cs
@@ -54,17 +54,16 @@ GetContentsInput input
         FilterDefinitionBuilder<Content> builder = Builders<Content>.Filter;
         var filter = builder.Empty;
 
+        var validIds = matches.Where(match => ObjectId.TryParse(match, out _)).ToList();
 
-        matches.ToList().ForEach(match =>
+        if (validIds.Count > 0)
         {
-            if (!ObjectId.TryParse(match, out _))
-            {
-                return;
-            }
+            var idFilter = Builders<Content>.Filter.Or(
+                validIds.ConvertAll(id => Builders<Content>.Filter.Eq(r => r.Id, id))
+            );
+            filter = idFilter;
+        }
 
-            var idFilter = builder.Eq(r => r.Id, match);
-            filter |= idFilter;
-        });
 
         if (filter == builder.Empty)
         {
@@ -100,14 +99,7 @@ public async Task<List<Content>> VisualSearch(
         GetContentsInput input
     )
     {
-        FilterDefinitionBuilder<Content> builder = Builders<Content>.Filter;
-
         var filter = filterLogicForVisualSearch(matches, input);
-        if (filter == builder.Empty)
-        {
-            return [];
-        }
-
 
         var contents = await _contentsCollection
             .Find(filter)