fix a bug of delete transaction

fatbobman · Aug 12, 2022 · fd6f3ed · fd6f3ed
1 parent 269b2a6
commit fd6f3ed
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/Sources/PersistentHistoryTrackingKit/Cleaner.swift b/Sources/PersistentHistoryTrackingKit/Cleaner.swift
@@ -29,19 +29,23 @@ struct Cleaner: TransactionCleanerProtocol {
     func cleanTransaction(before timestamp: Date?) throws {
         guard let timestamp = timestamp else { return }
         try backgroundContext.performAndWait {
-            let request = getPersistentStoreRequest(before: timestamp, for: authors)
-            try backgroundContext.execute(request)
+            if let request = getPersistentStoreRequest(before: timestamp, for: authors) {
+                try backgroundContext.execute(request)
+            }
         }
     }
 
     // make a request for delete transactions before timestamp
-    func getPersistentStoreRequest(before timestamp: Date, for allAuthors: [String]) -> NSPersistentStoreRequest {
+    func getPersistentStoreRequest(before timestamp: Date, for allAuthors: [String]) -> NSPersistentStoreRequest? {
         let historyStoreRequest = NSPersistentHistoryChangeRequest.deleteHistory(before: timestamp)
         if let fetchRequest = NSPersistentHistoryTransaction.fetchRequest {
             fetchRequest.predicate = createPredicateForAllAuthors(allAuthors: authors)
             historyStoreRequest.fetchRequest = fetchRequest
+            return historyStoreRequest
+        }
+        else {
+            return nil
         }
-        return historyStoreRequest
     }
 
     /// create predicate for all authors