From d31af734df4beeae83dda7d6083c5febf4ba27e5 Mon Sep 17 00:00:00 2001
From: Garrett Moon <garrett@excitedpixel.com>
Date: Fri, 20 Oct 2017 15:01:38 -0700
Subject: [PATCH] Dispatch batch update to main #trivial (#626)

* Dispatch batch update to main

* TableView too
---
 Source/ASCollectionView.mm | 4 +++-
 Source/ASTableView.mm      | 7 ++++++-
 2 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/Source/ASCollectionView.mm b/Source/ASCollectionView.mm
index 70dff45aab..ef9df71a5c 100644
--- a/Source/ASCollectionView.mm
+++ b/Source/ASCollectionView.mm
@@ -1587,7 +1587,9 @@ static NSString * const kReuseIdentifier = @"_ASCollectionReuseIdentifier";
 {
   if (_leadingScreensForBatching != leadingScreensForBatching) {
     _leadingScreensForBatching = leadingScreensForBatching;
-    [self _checkForBatchFetching];
+    ASPerformBlockOnMainThread(^{
+      [self _checkForBatchFetching];
+    });
   }
 }
 
diff --git a/Source/ASTableView.mm b/Source/ASTableView.mm
index 4606726f23..1458a47810 100644
--- a/Source/ASTableView.mm
+++ b/Source/ASTableView.mm
@@ -1318,7 +1318,12 @@ static NSString * const kCellReuseIdentifier = @"_ASTableViewCell";
 
 - (void)setLeadingScreensForBatching:(CGFloat)leadingScreensForBatching
 {
-  _leadingScreensForBatching = leadingScreensForBatching;
+  if (_leadingScreensForBatching != leadingScreensForBatching) {
+    _leadingScreensForBatching = leadingScreensForBatching;
+    ASPerformBlockOnMainThread(^{
+      [self _checkForBatchFetching];
+    });
+  }
 }
 
 - (BOOL)automaticallyAdjustsContentOffset