From 8930110efdcd62b818d035a588fe48c336d6686c Mon Sep 17 00:00:00 2001
From: Matthew Chen <charlesmchen@gmail.com>
Date: Wed, 14 Feb 2018 14:09:27 -0500
Subject: [PATCH] Fix glitch in batch processing of incoming messages.

---
 .../src/Messages/OWSMessageManager.m             | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/SignalServiceKit/src/Messages/OWSMessageManager.m b/SignalServiceKit/src/Messages/OWSMessageManager.m
index 342a59dc7..1be57bed7 100644
--- a/SignalServiceKit/src/Messages/OWSMessageManager.m
+++ b/SignalServiceKit/src/Messages/OWSMessageManager.m
@@ -137,12 +137,16 @@ NS_ASSUME_NONNULL_BEGIN
 
 - (void)yapDatabaseModified:(NSNotification *)notification
 {
-    static dispatch_once_t onceToken;
-    dispatch_once(&onceToken, ^{
-        [AppReadiness runNowOrWhenAppIsReady:^{
-            [OWSMessageUtils.sharedManager updateApplicationBadgeCount];
-        }];
-    });
+    if (AppReadiness.isAppReady) {
+        [OWSMessageUtils.sharedManager updateApplicationBadgeCount];
+    } else {
+        static dispatch_once_t onceToken;
+        dispatch_once(&onceToken, ^{
+            [AppReadiness runNowOrWhenAppIsReady:^{
+                [OWSMessageUtils.sharedManager updateApplicationBadgeCount];
+            }];
+        });
+    }
 }
 
 #pragma mark - Blocking