Merge pull request #26 from harlanhaskins/master

Adds a completion handler to UICollectionView.animateItemChanges
wokalski · Dec 8, 2016 · f564aa4 · f564aa4
2 parents b4a2b9e + 36fa2a4
commit f564aa4
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/Sources/Diff+UIKit.swift b/Sources/Diff+UIKit.swift
@@ -63,14 +63,15 @@ public extension UICollectionView {
     /// - parameter newData:            Data which reflects the current state of UITableView
     public func animateItemChanges<T: Collection>(
         oldData: T,
-        newData: T
+        newData: T,
+        completion: ((Bool) -> Void)? = nil
     ) where T.Iterator.Element: Equatable {
         performBatchUpdates({
             let update = BatchUpdate(diff: oldData.extendedDiff(newData))
             self.deleteItems(at: update.deletions)
             self.insertItems(at: update.insertions)
             update.moves.forEach { self.moveItem(at: $0.from, to: $0.to) }
-        }, completion: nil)
+        }, completion: completion)
     }
 }