Attempt to fix crashes and race conditions in User Properties manager

qonversion · Oct 17, 2024 · 234b847 · 234b847
1 parent 34ba186
commit 234b847
Showing 1 changed file with 33 additions and 19 deletions.
diff --git a/Sources/Qonversion/Qonversion/Main/QNUserPropertiesManager/QNUserPropertiesManager.m b/Sources/Qonversion/Qonversion/Main/QNUserPropertiesManager/QNUserPropertiesManager.m
@@ -26,10 +26,10 @@ @interface QNUserPropertiesManager()
 
 @property (nonatomic) QNInMemoryStorage *inMemoryStorage;
 
-@property (nonatomic, strong) NSMutableArray<QONUserPropertiesEmptyCompletionHandler> *completionBlocks;
+@property (atomic, strong) NSMutableArray<QONUserPropertiesEmptyCompletionHandler> *completionBlocks;
 
-@property (nonatomic, assign, readwrite) BOOL sendingScheduled;
-@property (nonatomic, assign, readwrite) BOOL updatingCurrently;
+@property (atomic, assign, readwrite) BOOL sendingScheduled;
+@property (atomic, assign, readwrite) BOOL updatingCurrently;
 @property (nonatomic, assign, readwrite) NSUInteger retryDelay;
 @property (nonatomic, assign, readwrite) NSUInteger retriesCounter;
 
@@ -94,25 +94,31 @@ - (void)forceSendProperties:(QONUserPropertiesEmptyCompletionHandler)completion
     return;
   }
 
-  [self.completionBlocks addObject:completion];
+  @synchronized (self) {
+    [self.completionBlocks addObject:completion];
+  }
 
   [self sendProperties:YES];
 }
 
 - (void)sendPropertiesWithDelay:(NSUInteger)delay {
-  if (!_sendingScheduled) {
-    _sendingScheduled = YES;
-    __block __weak QNUserPropertiesManager *weakSelf = self;
-    [_backgroundQueue addOperationWithBlock:^{
-      dispatch_async(dispatch_get_main_queue(), ^{
-        [weakSelf performSelector:@selector(sendPropertiesInBackground) withObject:nil afterDelay:delay];
-      });
-    }];
+  @synchronized (self) {
+    if (!_sendingScheduled) {
+      _sendingScheduled = YES;
+      __block __weak QNUserPropertiesManager *weakSelf = self;
+      [_backgroundQueue addOperationWithBlock:^{
+        dispatch_async(dispatch_get_main_queue(), ^{
+          [weakSelf performSelector:@selector(sendPropertiesInBackground) withObject:nil afterDelay:delay];
+        });
+      }];
+    }
   }
 }
 
 - (void)sendPropertiesInBackground {
-  _sendingScheduled = NO;
+  @synchronized (self) {
+    _sendingScheduled = NO;
+  }
   [self sendProperties];
 }
 
@@ -137,28 +143,36 @@ - (void)sendProperties:(BOOL)force {
     NSDictionary *properties = [self->_inMemoryStorage.storageDictionary copy];
 
     if (!properties || ![properties respondsToSelector:@selector(valueForKey:)]) {
-      self->_updatingCurrently = NO;
+      @synchronized (self) {
+        self->_updatingCurrently = NO;
+      }
       return;
     }
 
     if (properties.count == 0) {
-      self->_updatingCurrently = NO;
+      @synchronized (self) {
+        self->_updatingCurrently = NO;
+      }
       return;
     }
 
     self.inMemoryStorage.storageDictionary = @{};
     __block __weak QNUserPropertiesManager *weakSelf = self;
     [self.apiClient sendProperties:properties
                         completion:^(NSDictionary * _Nullable dict, NSError * _Nullable error) {
-      weakSelf.updatingCurrently = NO;
 
-      NSArray *completions = [weakSelf.completionBlocks copy];
+      NSArray *completions = @[];
+      @synchronized (self) {
+        weakSelf.updatingCurrently = NO;
+
+        completions = [weakSelf.completionBlocks copy];
+        [weakSelf.completionBlocks removeAllObjects];
+      }
+
       for (QONUserPropertiesEmptyCompletionHandler storedCompletion in completions) {
         storedCompletion();
       }
 
-      [weakSelf.completionBlocks removeAllObjects];
-
       if (error) {
         // copy of an existing array to prevent erasing properties set while the current request is in progress
         NSMutableDictionary *allProperties = [self.inMemoryStorage.storageDictionary mutableCopy];