Prevent retaining the cross dissolve transition view longer than needed

kean · May 29, 2024 · 4131f39 · 4131f39
1 parent 206d1e5
commit 4131f39
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/Sources/NukeExtensions/ImageViewExtensions.swift b/Sources/NukeExtensions/ImageViewExtensions.swift
@@ -435,9 +435,10 @@ extension ImageViewController {
                 transitionView.alpha = 0
                 imageView.alpha = 1
             },
-            completion: { isCompleted in
-                if isCompleted {
+            completion: { [weak transitionView] isCompleted in
+                if isCompleted, let transitionView {
                     transitionView.removeFromSuperview()
+                    transitionView.image = nil
                 }
             }
         )