BLURG

src/Messages/Interactions/TSOutgoingMessage.h

@@ -164,6 +164,8 @@ typedef NS_ENUM(NSInteger, TSGroupMetaMessage) {
 // This isn't a perfect arrangement, but in practice this will prevent
 // data loss and will resolve all known issues.
 - (void)updateWithMessageState:(TSOutgoingMessageState)messageState;
+- (void)updateWithMessageState:(TSOutgoingMessageState)messageState
+                   transaction:(YapDatabaseReadWriteTransaction *)transaction;
 - (void)updateWithSendingError:(NSError *)error;
 - (void)updateWithHasSyncedTranscript:(BOOL)hasSyncedTranscript;
 - (void)updateWithCustomMessage:(NSString *)customMessage transaction:(YapDatabaseReadWriteTransaction *)transaction;

src/Messages/Interactions/TSOutgoingMessage.m

@@ -256,16 +256,34 @@ NSString *const kTSOutgoingMessageSentRecipientAll = @"kTSOutgoingMessageSentRec
     }];
 }
 
+//- (void)updateWithMessageState:(TSOutgoingMessageState)messageState
+//{
+//    [self.dbConnection readWriteWithBlock:^(YapDatabaseReadWriteTransaction *transaction) {
+//        [self applyChangeToSelfAndLatestOutgoingMessage:transaction
+//                                            changeBlock:^(TSOutgoingMessage *message) {
+//                                                [message setMessageState:messageState];
+//                                            }];
+//    }];
+//}
+
 - (void)updateWithMessageState:(TSOutgoingMessageState)messageState
 {
     [self.dbConnection readWriteWithBlock:^(YapDatabaseReadWriteTransaction *transaction) {
-        [self applyChangeToSelfAndLatestOutgoingMessage:transaction
-                                            changeBlock:^(TSOutgoingMessage *message) {
-                                                [message setMessageState:messageState];
-                                            }];
+        [self updateWithMessageState:messageState transaction:transaction];
     }];
 }
 
+- (void)updateWithMessageState:(TSOutgoingMessageState)messageState
+                   transaction:(YapDatabaseReadWriteTransaction *)transaction
+{
+    OWSAssert(transaction);
+    
+    [self applyChangeToSelfAndLatestOutgoingMessage:transaction
+                                        changeBlock:^(TSOutgoingMessage *message) {
+                                            [message setMessageState:messageState];
+                                        }];
+}
+
 - (void)updateWithHasSyncedTranscript:(BOOL)hasSyncedTranscript
 {
     [self.dbConnection readWriteWithBlock:^(YapDatabaseReadWriteTransaction *transaction) {