Merge pull request #47 from customerio/hotfix/local-store-fix

Added timestamp to Queue URL and Updated local store logic
customerio · Feb 29, 2024 · 2956de4 · 2956de4
2 parents e648691 + 62ce135
commit 2956de4
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 6 deletions.
diff --git a/src/managers/message-manager.js b/src/managers/message-manager.js
@@ -17,7 +17,7 @@ import {
 } from "./message-component-manager";
 import { resolveMessageProperties } from "./gist-properties-manager";
 import { positions, addPageElement } from "./page-component-manager";
-import { checkMessageQueue } from "./queue-manager";
+import { checkMessageQueue, updateQueueLocalStore, getMessagesFromLocalStore } from "./queue-manager";
 
 var shownMessages = [];
 
@@ -191,7 +191,7 @@ async function handleGistEvents(e) {
         log(`Engine render for message: ${currentMessage.messageId} timer elapsed in ${timeElapsed.toFixed(3)} seconds`);
         currentMessage.currentRoute = e.data.gist.parameters.route;
         if (currentMessage.firstLoad) {
-          shownMessages.push(currentMessage);
+          messageShown(currentMessage);
           if (currentMessage.overlay) {
             showOverlayComponent(currentMessage);
           } else {
@@ -286,6 +286,14 @@ async function handleGistEvents(e) {
   }
 }
 
+function messageShown(message) {
+  shownMessages.push(message);
+  var messagesInLocalStore = getMessagesFromLocalStore();
+  if (messagesInLocalStore != null && messagesInLocalStore.length > 0) {
+    updateQueueLocalStore(messagesInLocalStore.filter(item => item.queueId !== message.queueId));
+  }
+}
+
 export function hasMessageBeenShownBefore(message) {
   return shownMessages.find(msg => msg.queueId === message.queueId) !== undefined;
 }
diff --git a/src/managers/queue-manager.js b/src/managers/queue-manager.js
@@ -88,13 +88,13 @@ export async function pullMessagesFromQueue() {
         var responseData = [];
         if (response) {
           if (response.status === 200 || response.status === 204) {
-            var expiryDate = new Date(new Date().getTime() + MESSAGES_LOCAL_STORE_CACHE_IN_MINUTES);
-            setKeyWithExpiryToLocalStore(userQueueLocalStoreName, response.data, expiryDate);
+            log("200 response, updating local store.");
             responseData = response.data;
+            updateQueueLocalStore(responseData);
           }
           else if (response.status === 304) {
             log("304 response, using local store.");
-            responseData = getKeyFromLocalStore(userQueueLocalStoreName);
+            responseData = getMessagesFromLocalStore();
           }
           if (responseData && responseData.length > 0) {
             log(`Message queue checked for user ${getUserToken()}, ${responseData.length} messages found.`);
@@ -116,4 +116,13 @@ export async function pullMessagesFromQueue() {
   } else {
     log(`User token reset, skipping queue check.`);
   }
+}
+
+export function updateQueueLocalStore(messages) {
+  var expiryDate = new Date(new Date().getTime() + MESSAGES_LOCAL_STORE_CACHE_IN_MINUTES);
+  setKeyWithExpiryToLocalStore(userQueueLocalStoreName, messages, expiryDate);
+}
+
+export function getMessagesFromLocalStore() {
+  return getKeyFromLocalStore(userQueueLocalStoreName);
 }
diff --git a/src/services/queue-service.js b/src/services/queue-service.js
@@ -11,8 +11,9 @@ export async function getUserQueue() {
   var response;
   try {
     if (!checkInProgress) {
+      var timestamp = new Date().getTime();
       checkInProgress = true;
-      response = await UserNetworkInstance().post(`/api/v1/users`, {});
+      response = await UserNetworkInstance().post(`/api/v1/users?timestamp=${timestamp}`, {});
     }
   } catch (error) {
     if (error.response) {