Merge pull request #28 from roanutil/feature/make-internal-aggregate-…

…endpoint-public Add a unified aggregate endpoint
roanutil · Jul 18, 2024 · 742aa43 · 742aa43
2 parents ca4d404 + 06df31f
commit 742aa43
Showing 1 changed file with 32 additions and 9 deletions.
diff --git a/Sources/CoreDataRepository/CoreDataRepository+Aggregate.swift b/Sources/CoreDataRepository/CoreDataRepository+Aggregate.swift
@@ -12,6 +12,38 @@ import Foundation
 // swiftlint:disable file_length
 
 extension CoreDataRepository {
+    public enum AggregateFunction: String {
+        case count
+        case sum
+        case average
+        case min
+        case max
+    }
+
+    @inlinable
+    public func aggregate<Value: Numeric>(
+        function: AggregateFunction,
+        predicate: NSPredicate,
+        entityDesc: NSEntityDescription,
+        attributeDesc: NSAttributeDescription,
+        groupBy: NSAttributeDescription? = nil,
+        as valueType: Value.Type
+    ) async -> Result<Value, CoreDataError> {
+        switch function {
+        case .count:
+            return await count(predicate: predicate, entityDesc: entityDesc, as: valueType)
+        default:
+            return await Self.send(
+                function: .sum,
+                context: context,
+                predicate: predicate,
+                entityDesc: entityDesc,
+                attributeDesc: attributeDesc,
+                groupBy: groupBy
+            )
+        }
+    }
+
     // MARK: Count
 
     /// Get the count or quantity of managed object instances that satisfy the predicate.
@@ -375,15 +407,6 @@ extension CoreDataRepository {
 
     // MARK: Internals
 
-    @usableFromInline
-    enum AggregateFunction: String {
-        case count
-        case sum
-        case average
-        case min
-        case max
-    }
-
     private static func aggregate<Value: Numeric>(
         context: NSManagedObjectContext,
         request: NSFetchRequest<NSDictionary>