make event retry count configurable

rudderlabs · Nov 3, 2020 · 66e9648 · 66e9648
1 parent 9777852
commit 66e9648
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 5 deletions.
diff --git a/analytics.js b/analytics.js
@@ -50,7 +50,7 @@ const queryDefaults = {
  */
 function enqueue(rudderElement, type) {
   if (!this.eventRepository) {
-    this.eventRepository = EventRepository;
+    this.eventRepository = EventRepository();
   }
   this.eventRepository.enqueue(rudderElement, type);
 }
@@ -78,7 +78,6 @@ class Analytics {
     this.toBeProcessedArray = [];
     this.toBeProcessedByIntegrationArray = [];
     this.storage = Storage;
-    this.eventRepository = EventRepository;
     this.sendAdblockPage = false;
     this.sendAdblockPageOptions = {};
     this.clientSuppliedCallbacks = {};
@@ -887,6 +886,18 @@ class Analytics {
       this.registerCallbacks(true);
     }
 
+    const queueOptions = {};
+    if (options) {
+      if (options.maxRetries) {
+        queueOptions.maxAttempts = options.maxRetries;
+      }
+      if (options.maxQueueSize) {
+        queueOptions.maxItems = options.maxQueueSize;
+      }
+    }
+
+    this.eventRepository = new EventRepository(queueOptions);
+
     this.eventRepository.writeKey = writeKey;
     if (serverUrl) {
       this.eventRepository.url = serverUrl;

diff --git a/utils/EventRepository.js b/utils/EventRepository.js
@@ -41,7 +41,7 @@ class EventRepository {
    *Creates an instance of EventRepository.
    * @memberof EventRepository
    */
-  constructor() {
+  constructor(options) {
     this.eventsBuffer = [];
     this.writeKey = "";
     this.url = "";
@@ -51,6 +51,10 @@ class EventRepository {
     // previous implementation
     // setInterval(this.preaparePayloadAndFlush, FLUSH_INTERVAL_DEFAULT, this);
 
+    if (options) {
+      // TODO: add checks for value - has to be +ve?
+      Object.assign(queueOptions, options);
+    }
     this.payloadQueue = new Queue("rudder", queueOptions, function (
       item,
       done
@@ -229,5 +233,4 @@ class EventRepository {
     });
   }
 }
-let eventRepository = new EventRepository();
-export { eventRepository as EventRepository };
+export { EventRepository };