diff --git a/AsyncDisplayKit/ASCollectionNode+Beta.h b/AsyncDisplayKit/ASCollectionNode+Beta.h index e05e740b..a2073d44 100644 --- a/AsyncDisplayKit/ASCollectionNode+Beta.h +++ b/AsyncDisplayKit/ASCollectionNode+Beta.h @@ -14,9 +14,13 @@ NS_ASSUME_NONNULL_BEGIN @interface ASCollectionNode (Beta) - (instancetype)initWithFrame:(CGRect)frame collectionViewLayout:(UICollectionViewLayout *)layout layoutFacilitator:(nullable id)layoutFacilitator; + - (void)beginUpdates; + - (void)endUpdatesAnimated:(BOOL)animated; +- (void)endUpdatesAnimated:(BOOL)animated completion:(void (^)(BOOL))completion; + @end NS_ASSUME_NONNULL_END diff --git a/AsyncDisplayKit/ASCollectionNode.mm b/AsyncDisplayKit/ASCollectionNode.mm index c7ddc168..2331a38e 100644 --- a/AsyncDisplayKit/ASCollectionNode.mm +++ b/AsyncDisplayKit/ASCollectionNode.mm @@ -194,7 +194,12 @@ - (void)endUpdatesAnimated:(BOOL)animated { - [self.view.dataController endUpdatesAnimated:animated completion:nil]; + [self endUpdatesAnimated:animated completion:nil]; +} + +- (void)endUpdatesAnimated:(BOOL)animated completion:(void (^)(BOOL))completion +{ + [self.view.dataController endUpdatesAnimated:animated completion:completion]; } #pragma mark - ASCollectionView Forwards