Adding limit parameter to searchFor

spnkr · Jan 3, 2023 · 288998d · 288998d
1 parent 198be73
commit 288998d
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/Sources/CoreDataPlus/ManagedObjectSearchable.swift b/Sources/CoreDataPlus/ManagedObjectSearchable.swift
@@ -9,11 +9,18 @@ public protocol ManagedObjectSearchable where Self: NSFetchRequestResult {
 }
 
 public extension ManagedObjectSearchable {
-    static func searchFor(_ predicate: NSPredicate?, sortBy: [NSSortDescriptor]? = nil, context: NSManagedObjectContext) -> [Self] {
+    static func searchFor(_ predicate: NSPredicate?,
+                          sortBy: [NSSortDescriptor]? = nil,
+                          limit: Int? = nil,
+                          context: NSManagedObjectContext) -> [Self] {
         let request = NSFetchRequest<Self>()
         request.predicate = predicate
         request.entity = entity()
 
+        if let limit = limit {
+            request.fetchLimit = limit
+        }
+
         if let sortBy = sortBy {
             request.sortDescriptors = sortBy
         }