Use AS::Mutex in ASCollectionElement (#1445)

TextureGroup · Apr 5, 2019 · b9c8b32 · b9c8b32
1 parent a66f048
commit b9c8b32
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/Source/Details/ASCollectionElement.mm b/Source/Details/ASCollectionElement.mm
@@ -9,7 +9,7 @@
 
 #import <AsyncDisplayKit/ASCollectionElement.h>
 #import <AsyncDisplayKit/ASCellNode+Internal.h>
-#import <mutex>
+#import <AsyncDisplayKit/ASThread.h>
 
 @interface ASCollectionElement ()
 
@@ -19,7 +19,7 @@ @interface ASCollectionElement ()
 @end
 
 @implementation ASCollectionElement {
-  std::mutex _lock;
+  AS::Mutex _lock;
   ASCellNode *_node;
 }
 
@@ -45,7 +45,7 @@ - (instancetype)initWithNodeModel:(id)nodeModel
 
 - (ASCellNode *)node
 {
-  std::lock_guard<std::mutex> l(_lock);
+  AS::MutexLocker l(_lock);
   if (_nodeBlock != nil) {
     ASCellNode *node = _nodeBlock();
     _nodeBlock = nil;
@@ -64,7 +64,7 @@ - (ASCellNode *)node
 
 - (ASCellNode *)nodeIfAllocated
 {
-  std::lock_guard<std::mutex> l(_lock);
+  AS::MutexLocker l(_lock);
   return _node;
 }
 
@@ -73,7 +73,7 @@ - (void)setTraitCollection:(ASPrimitiveTraitCollection)traitCollection
   ASCellNode *nodeIfNeedsPropagation;
 
   {
-    std::lock_guard<std::mutex> l(_lock);
+    AS::MutexLocker l(_lock);
     if (! ASPrimitiveTraitCollectionIsEqualToASPrimitiveTraitCollection(_traitCollection, traitCollection)) {
       _traitCollection = traitCollection;
       nodeIfNeedsPropagation = _node;