Add option to enable main-thread mode #3

FluidGroup · Feb 19, 2023 · 68df47f · 68df47f
1 parent 835f48d
commit 68df47f
Show file tree

Hide file tree

Showing 4 changed files with 18 additions and 1 deletion.
diff --git a/Source/ASCollectionNode.h b/Source/ASCollectionNode.h
@@ -75,6 +75,11 @@ NS_ASSUME_NONNULL_BEGIN
  */
 @property (nullable, weak) id <ASCollectionDataSource> dataSource;
 
+/**
+ * A Boolean valut that indicates whether performs collection node as main thread only.
+ */
+@property (nonatomic) BOOL isMainThreadModeEnabled;
+
 /**
  * The number of screens left to scroll before the delegate -collectionNode:beginBatchFetchingWithContext: is called.
  *

diff --git a/Source/ASCollectionNode.mm b/Source/ASCollectionNode.mm
@@ -432,6 +432,16 @@ - (_ASCollectionPendingState *)pendingState
   return _pendingState;
 }
 
+- (void)setIsMainThreadModeEnabled:(BOOL)isMainThreadModeEnabled
+{
+  self.dataController.isMainThreadModeEnabled = isMainThreadModeEnabled;
+}
+
+- (BOOL)isMainThreadModeEnabled
+{
+  return self.dataController.isMainThreadModeEnabled;
+}
+
 - (void)setInverted:(BOOL)inverted
 {
   self.transform = inverted ? CATransform3DMakeScale(1, -1, 1)  : CATransform3DIdentity;

diff --git a/Source/Details/ASDataController.h b/Source/Details/ASDataController.h
@@ -160,6 +160,8 @@ ASDK_EXTERN NSString * const ASCollectionInvalidUpdateException;
 
 - (instancetype)init NS_UNAVAILABLE;
 
+@property (nonatomic) BOOL isMainThreadModeEnabled;
+
 /**
  * The node that owns this data controller, if any.
  *

diff --git a/Source/Details/ASDataController.mm b/Source/Details/ASDataController.mm
@@ -662,7 +662,7 @@ - (void)updateWithChangeSet:(_ASHierarchyChangeSet *)changeSet
 
   // Step 3 can be done on the main thread or on _editingTransactionQueue
   // depending on an experiment.
-  BOOL mainThreadOnly = ASActivateExperimentalFeature(ASExperimentalMainThreadOnlyDataController);
+  BOOL mainThreadOnly = self.isMainThreadModeEnabled || ASActivateExperimentalFeature(ASExperimentalMainThreadOnlyDataController);
   if (mainThreadOnly) {
     // In main-thread-only mode allocate and layout all nodes serially on the main thread.
     //
-Original file line number
+Diff line change
@@ Expand Up @@
     - (instancetype)init NS_UNAVAILABLE;
+    @property (nonatomic) BOOL isMainThreadModeEnabled;
     /**
      * The node that owns this data controller, if any.
      *
@@ Expand Down @@