Attempt to fix continuation misuse

finnvoor · Feb 12, 2024 · 1213922 · 1213922
1 parent 4cf0f65
commit 1213922
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/Sources/Upscaling/UpscalingExportSession.swift b/Sources/Upscaling/UpscalingExportSession.swift
@@ -348,6 +348,7 @@ public class UpscalingExportSession {
                     } else {
                         assetWriterInput.markAsFinished()
                         continuation.resume()
+                        return
                     }
                 }
             }
@@ -392,10 +393,12 @@ public class UpscalingExportSession {
                         } else {
                             assetWriterInput.markAsFinished()
                             continuation.resume(throwing: Error.missingImageBuffer)
+                            return
                         }
                     } else {
                         assetWriterInput.markAsFinished()
                         continuation.resume()
+                        return
                     }
                 }
             }
@@ -466,10 +469,12 @@ public class UpscalingExportSession {
                         } else {
                             assetWriterInput.markAsFinished()
                             continuation.resume(throwing: Error.missingTaggedBuffers)
+                            return
                         }
                     } else {
                         assetWriterInput.markAsFinished()
                         continuation.resume()
+                        return
                     }
                 }
             }