Record number of events dropped at the end for retry mechanims

gojek · Apr 17, 2024 · 9dc23bc · 9dc23bc
1 parent 599acae
commit 9dc23bc
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/Sources/Clickstream/NetworkManager/Core/NetworkBuilder.swift b/Sources/Clickstream/NetworkManager/Core/NetworkBuilder.swift
@@ -71,7 +71,7 @@ extension DefaultNetworkBuilder {
                     checkedSelf.trackHealthEvents(eventBatch: eventBatch,
                                                           eventBatchData: data)
                 }
-
+                eventRequest.eventCount = eventBatch.events.count
                 checkedSelf.retryMech.trackBatch(with: eventRequest)
                 #if EVENT_VISUALIZER_ENABLED
                 /// Update status of the event batch to sent to network

diff --git a/Sources/Clickstream/NetworkManager/Domain/Entities/EventRequest.swift b/Sources/Clickstream/NetworkManager/Domain/Entities/EventRequest.swift
@@ -19,6 +19,7 @@ struct EventRequest: Codable, Equatable {
     var createdTimestamp: Date?
     var eventType: Constants.EventType?
     var isInternal: Bool?
+    var eventCount: Int
 
     init(guid: String,
          data: Data? = nil) {
@@ -29,6 +30,7 @@ struct EventRequest: Codable, Equatable {
         self.createdTimestamp = Date()
         self.isInternal = false
         self.eventType = .realTime
+        self.eventCount = 0
     }
 
     static func == (lhs: Self, rhs: Self) -> Bool {
@@ -86,6 +88,10 @@ extension EventRequest: DatabasePersistable {
             t.add(column: "eventType", .text)
         }
 
-        return [("addsIsInternalToEventRequest", addsIsInternal), ("addsEventTypeToEventRequest", addsEventType)]
+        let addsEventCount: (TableAlteration) -> Void = { t in
+            t.add(column: "eventCount", .integer)
+        }
+
+        return [("addsIsInternalToEventRequest", addsIsInternal), ("addsEventTypeToEventRequest", addsEventType), ("addsEventCountToEventRequest", addsEventCount)]
     }
 }