Add cusotm method for performing arbitrary operations against a scratchpad context

feature/add-method-for-custom-operations-in-context

Author: roanutil

Sources/CoreDataRepository/CoreDataRepository+Custom.swift

@@ -0,0 +1,24 @@
+// CoreDataRepository+Custom.swift
+// CoreDataRepository
+//
+//
+// MIT License
+//
+// Copyright © 2024 Andrew Roan
+
+import CoreData
+import Foundation
+
+extension CoreDataRepository {
+    /// Escape hatch method for performing arbitrary operations inside a 'scratchpad' `NSManagedObjectContext` where
+    /// changes will be discarded if not saved.
+    public func custom<T>(
+        schedule: NSManagedObjectContext.ScheduledTaskType = .enqueued,
+        block: @escaping (
+            _ parentContext: NSManagedObjectContext,
+            _ scratchPadContext: NSManagedObjectContext
+        ) throws -> T
+    ) async -> Result<T, CoreDataError> {
+        await context.performInScratchPad(schedule: schedule) { [context] scratchPad in try block(context, scratchPad) }
+    }
+}