Merge pull request #17 from someyura/master

Fix retain cycle at TraitQueryLayoutSet
DeclarativeHub · Mar 26, 2021 · 50ae5ab · 50ae5ab
2 parents 63ff2af + e492e76
commit 50ae5ab
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/Sources/Layout/TraitQueryLayoutSet.swift b/Sources/Layout/TraitQueryLayoutSet.swift
@@ -45,7 +45,8 @@ public class TraitQueryLayoutSet: LayoutNode {
         updateLayout()
         NotificationCenter.default.addObserver(self, selector: #selector(didChangeTraitCollection), name: .windowTraitCollectionDidChange, object: nil)
         let revertable = Revertable()
-        revertable.appendBlock {
+        revertable.appendBlock { [weak self] in
+            guard let self = self else { return }
             self.container = nil
             self.revertable?.revert()
             NotificationCenter.default.removeObserver(self)