Add atomic batch operations

3.0-preview

Author: roanutil

Sources/CoreDataRepository/CoreDataRepository+Batch.swift

@@ -67,6 +67,28 @@ extension CoreDataRepository {
         return (success: successes, failed: failures)
     }
 
+    /// Create a batch of unmanaged models.
+    public func createAtomiclly<Model: UnmanagedModel>(
+        _ items: [Model],
+        transactionAuthor: String? = nil
+    ) async -> Result<[Model], CoreDataError> {
+        await context.performInScratchPad(schedule: .enqueued) { [context] scratchPad in
+            let objects = try items.map { item in
+                let object = Model.ManagedModel(context: scratchPad)
+                try item.updating(managed: object)
+                return object
+            }
+            try scratchPad.save()
+            try context.performAndWait {
+                context.transactionAuthor = transactionAuthor
+                try context.save()
+                context.transactionAuthor = nil
+            }
+            try scratchPad.obtainPermanentIDs(for: objects)
+            return try objects.map(Model.init(managed:))
+        }
+    }
+
     /// Read a batch of unmanaged models.
     ///
     /// This operation is non-atomic. Each instance may succeed or fail individually.
@@ -107,6 +129,21 @@ extension CoreDataRepository {
         return (success: successes, failed: failures)
     }
 
+    /// Read a batch of unmanaged models.
+    public func readAtomically<Model: UnmanagedModel>(
+        _ urls: [URL],
+        of _: Model.Type
+    ) async -> Result<[Model], CoreDataError> {
+        await context.performInChild(schedule: .enqueued) { readContext in
+            try urls.map { url in
+                let id = try readContext.objectId(from: url).get()
+                let object = try readContext.notDeletedObject(for: id)
+                let managed: Model.ManagedModel = try object.asManagedModel()
+                return try Model(managed: managed)
+            }
+        }
+    }
+
     /// Execute a NSBatchUpdateRequest against the store.
     public func update(
         _ request: NSBatchUpdateRequest,
@@ -123,6 +160,33 @@ extension CoreDataRepository {
         }
     }
 
+    /// Update the store with a batch of unmanaged models.
+    public func updateAtomically<Model: UnmanagedModel>(
+        _ items: [Model],
+        transactionAuthor: String? = nil
+    ) async -> Result<[Model], CoreDataError> {
+        await context.performInScratchPad(schedule: .enqueued) { [context] scratchPad in
+            scratchPad.transactionAuthor = transactionAuthor
+            let objects = try items.map { item in
+                guard let url = item.managedIdUrl else {
+                    throw CoreDataError.noUrlOnItemToMapToObjectId
+                }
+                let id = try scratchPad.objectId(from: url).get()
+                let object = try scratchPad.notDeletedObject(for: id)
+                let managed: Model.ManagedModel = try object.asManagedModel()
+                try item.updating(managed: managed)
+                return managed
+            }
+            try scratchPad.save()
+            try context.performAndWait {
+                context.transactionAuthor = transactionAuthor
+                try context.save()
+                context.transactionAuthor = nil
+            }
+            return try objects.map(Model.init(managed:))
+        }
+    }
+
     /// Update the store with a batch of unmanaged models.
     ///
     /// This operation is non-atomic. Each instance may succeed or fail individually.
@@ -223,4 +287,27 @@ extension CoreDataRepository {
         })
         return (success: successes, failed: failures)
     }
+
+    /// Delete from the store with a batch of unmanaged models.
+    public func deleteAtomically(
+        _ urls: [URL],
+        transactionAuthor: String? = nil
+    ) async -> Result<Void, CoreDataError> {
+        await context.performInScratchPad(schedule: .enqueued) { [context] scratchPad in
+            scratchPad.transactionAuthor = transactionAuthor
+            for url in urls {
+                let id = try scratchPad.objectId(from: url).get()
+                let object = try scratchPad.notDeletedObject(for: id)
+                object.prepareForDeletion()
+                scratchPad.delete(object)
+            }
+            try scratchPad.save()
+            try context.performAndWait {
+                context.transactionAuthor = transactionAuthor
+                try context.save()
+                context.transactionAuthor = nil
+            }
+            return ()
+        }
+    }
 }