Merge branch 'charlesmchen/skipRedundantSyncMessages'

Signal/src/ViewControllers/OWSLinkDeviceViewController.m

@@ -1,5 +1,5 @@
 //
-//  Copyright (c) 2017 Open Whisper Systems. All rights reserved.
+//  Copyright (c) 2018 Open Whisper Systems. All rights reserved.
 //
 
 #import "OWSLinkDeviceViewController.h"
@@ -147,7 +147,7 @@ NS_ASSUME_NONNULL_BEGIN
 - (void)provisionWithParser:(OWSDeviceProvisioningURLParser *)parser
 {
     // Optimistically set this flag.
-    [OWSDeviceManager.sharedManager setMayHaveLinkedDevices:YES dbConnection:self.dbConnection];
+    [OWSDeviceManager.sharedManager setMayHaveLinkedDevices];
 
     ECKeyPair *_Nullable identityKeyPair = [[OWSIdentityManager sharedManager] identityKeyPair];
     OWSAssert(identityKeyPair);

Signal/src/ViewControllers/OWSLinkedDevicesTableViewController.m

@@ -146,9 +146,7 @@ int const OWSLinkedDevicesTableViewControllerSectionAddDevice = 1;
             if (devices.count > 1) {
                 // Setting this flag here shouldn't be necessary, but we do so
                 // because the "cost" is low and it will improve robustness.
-                [OWSDeviceManager.sharedManager
-                    setMayHaveLinkedDevices:YES
-                               dbConnection:[[TSStorageManager sharedManager] newDatabaseConnection]];
+                [OWSDeviceManager.sharedManager setMayHaveLinkedDevices];
             }
 
             if (devices.count > [OWSDevice numberOfKeysInCollection]) {

SignalServiceKit/src/Devices/OWSDevice.h

@@ -1,5 +1,5 @@
 //
-//  Copyright (c) 2017 Open Whisper Systems. All rights reserved.
+//  Copyright (c) 2018 Open Whisper Systems. All rights reserved.
 //
 
 #import "TSYapDatabaseObject.h"
@@ -16,7 +16,7 @@ extern uint32_t const OWSDevicePrimaryDeviceId;
 + (instancetype)sharedManager;
 
 - (BOOL)mayHaveLinkedDevices:(YapDatabaseConnection *)dbConnection;
-- (void)setMayHaveLinkedDevices:(BOOL)value dbConnection:(YapDatabaseConnection *)dbConnection;
+- (void)setMayHaveLinkedDevices;
 
 - (BOOL)hasReceivedSyncMessageInLastSeconds:(NSTimeInterval)intervalSeconds;
 - (void)setHasReceivedSyncMessage;

SignalServiceKit/src/Devices/OWSDevice.m

@@ -1,5 +1,5 @@
 //
-//  Copyright (c) 2017 Open Whisper Systems. All rights reserved.
+//  Copyright (c) 2018 Open Whisper Systems. All rights reserved.
 //
 
 #import "OWSDevice.h"
@@ -19,7 +19,9 @@ NSString *const kTSStorageManager_MayHaveLinkedDevices = @"kTSStorageManager_May
 
 @interface OWSDeviceManager ()
 
+// This property should only be accessed while synchronized on self.
 @property (atomic, nullable) NSNumber *mayHaveLinkedDevicesCached;
+
 @property (atomic) NSDate *lastReceivedSyncMessage;
 
 @end
@@ -59,20 +61,25 @@ NSString *const kTSStorageManager_MayHaveLinkedDevices = @"kTSStorageManager_May
     }
 }
 
-- (void)setMayHaveLinkedDevices:(BOOL)value dbConnection:(YapDatabaseConnection *)dbConnection
+- (void)setMayHaveLinkedDevices
 {
-    OWSAssert(dbConnection);
-
     @synchronized(self)
     {
-        self.mayHaveLinkedDevicesCached = @(value);
+        if (self.mayHaveLinkedDevicesCached != nil && self.mayHaveLinkedDevicesCached.boolValue) {
+            // Skip redundant writes.
+            return;
+        }
 
-        [dbConnection asyncReadWriteWithBlock:^(YapDatabaseReadWriteTransaction *_Nonnull transaction) {
-            [transaction setObject:@(value)
+        self.mayHaveLinkedDevicesCached = @(YES);
+    }
+
+    // Note that we write async to avoid opening transactions within transactions.
+    [TSStorageManager.sharedManager.newDatabaseConnection
+        asyncReadWriteWithBlock:^(YapDatabaseReadWriteTransaction *_Nonnull transaction) {
+            [transaction setObject:@(YES)
                             forKey:kTSStorageManager_MayHaveLinkedDevices
                       inCollection:kTSStorageManager_OWSDeviceCollection];
         }];
-    }
 }
 
 - (BOOL)hasReceivedSyncMessageInLastSeconds:(NSTimeInterval)intervalSeconds
@@ -83,6 +90,8 @@ NSString *const kTSStorageManager_MayHaveLinkedDevices = @"kTSStorageManager_May
 - (void)setHasReceivedSyncMessage
 {
     self.lastReceivedSyncMessage = [NSDate new];
+
+    [self setMayHaveLinkedDevices];
 }
 
 @end

SignalServiceKit/src/Messages/OWSMessageSender.m

@@ -1,5 +1,5 @@
 //
-//  Copyright (c) 2017 Open Whisper Systems. All rights reserved.
+//  Copyright (c) 2018 Open Whisper Systems. All rights reserved.
 //
 
 #import "OWSMessageSender.h"
@@ -1090,7 +1090,7 @@ NSString *const OWSMessageSenderRateLimitedException = @"RateLimitedException";
     if (missingDevices.count > 0) {
         NSString *localNumber = [TSAccountManager localNumber];
         if ([localNumber isEqualToString:recipient.uniqueId]) {
-            [OWSDeviceManager.sharedManager setMayHaveLinkedDevices:YES dbConnection:self.dbConnection];
+            [OWSDeviceManager.sharedManager setMayHaveLinkedDevices];
         }
     }