Clear cached layout attributes if the data source counts change (#125)

airbnb · Mar 8, 2024 · 281c390 · 281c390
1 parent 63315fe
commit 281c390
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/MagazineLayout/Public/MagazineLayout.swift b/MagazineLayout/Public/MagazineLayout.swift
@@ -808,6 +808,13 @@ public final class MagazineLayout: UICollectionViewLayout {
     hasDataSourceCountInvalidationBeforeReceivingUpdateItems = context.invalidateDataSourceCounts &&
       !context.invalidateEverything
 
+    if context.invalidateDataSourceCounts {
+      itemLayoutAttributes.removeAll()
+      headerLayoutAttributes.removeAll()
+      footerLayoutAttributes.removeAll()
+      backgroundLayoutAttributes.removeAll()
+    }
+
     super.invalidateLayout(with: context)
   }