Flush message counter on shutdown

signalapp · Sep 8, 2021 · 27bdf6e · 27bdf6e
1 parent 96bf1b9
commit 27bdf6e
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 4 deletions.
diff --git a/ts/background.ts b/ts/background.ts
@@ -618,6 +618,9 @@ export async function startApp(): Promise<void> {
     window.Events = createIPCEvents({
       shutdown: async () => {
         window.log.info('background/shutdown');
+
+        window.Signal.Util.flushMessageCounter();
+
         // Stop background processing
         AttachmentDownloads.stop();
         if (idleDetector) {

diff --git a/ts/util/incrementMessageCounter.ts b/ts/util/incrementMessageCounter.ts
@@ -17,6 +17,16 @@ export function incrementMessageCounter(): number {
   return receivedAtCounter;
 }
 
-const debouncedUpdateLastReceivedAt = debounce(() => {
-  localStorage.setItem('lastReceivedAtCounter', String(receivedAtCounter));
-}, 500);
+export function flushMessageCounter(): void {
+  debouncedUpdateLastReceivedAt.flush();
+}
+
+const debouncedUpdateLastReceivedAt = debounce(
+  () => {
+    localStorage.setItem('lastReceivedAtCounter', String(receivedAtCounter));
+  },
+  25,
+  {
+    maxWait: 25,
+  }
+);
diff --git a/ts/util/index.ts b/ts/util/index.ts
@@ -14,7 +14,10 @@ import { getStringForProfileChange } from './getStringForProfileChange';
 import { getTextWithMentions } from './getTextWithMentions';
 import { getUserAgent } from './getUserAgent';
 import { hasExpired } from './hasExpired';
-import { incrementMessageCounter } from './incrementMessageCounter';
+import {
+  incrementMessageCounter,
+  flushMessageCounter,
+} from './incrementMessageCounter';
 import { isFileDangerous } from './isFileDangerous';
 import { makeLookup } from './makeLookup';
 import {
@@ -50,6 +53,7 @@ export {
   createWaitBatcher,
   deleteForEveryone,
   downloadAttachment,
+  flushMessageCounter,
   fromWebSafeBase64,
   generateSecurityNumber,
   getStringForProfileChange,