Update EventsListViewModel.swift

gojek · Apr 18, 2024 · c3361be · c3361be
1 parent 326c628
commit c3361be
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/Sources/EventVisualizer/src/EventsList/EventsListViewModel.swift b/Sources/EventVisualizer/src/EventsList/EventsListViewModel.swift
@@ -67,9 +67,14 @@ final class EventsListViewModel: EventsListViewModelInput {
                       let timestamp = message.asDictionary["storage.\(Constants.EventVisualizer.eventTimestamp)"] as? SwiftProtobuf.Google_Protobuf_Timestamp {
                 eventTimeStamp = "\(timestamp.date)"
                 state = EventsHelper.shared.getState(of: eventGuid)
-            } else if let eventGuid = message.asDictionary["storage.meta.storage.\(Constants.EventVisualizer.eventGID)"] as? String,
-                      let timestamp = message.asDictionary["storage.\(Constants.EventVisualizer.eventTimestamp)"] as? SwiftProtobuf.Google_Protobuf_Timestamp {
-                eventTimeStamp = "\(timestamp.date)"
+            } else if let eventGuid = message.asDictionary["storage.meta.storage.\(Constants.EventVisualizer.eventGID)"] as? String {
+                if let timestamp = message.asDictionary["storage.\(Constants.EventVisualizer.eventTimestamp)"] as? SwiftProtobuf.Google_Protobuf_Timestamp {
+                    eventTimeStamp = "\(timestamp.date)"
+                } else if let nanos = message.asDictionary["storage.eventTimestamp.nanos"] as? Int32,
+                            let seconds = message.asDictionary["storage.eventTimestamp.seconds"] as? Int64 {
+                    let timestamp = SwiftProtobuf.Google_Protobuf_Timestamp(seconds: seconds, nanos: nanos)
+                    eventTimeStamp = "\(timestamp.date)"
+                }
                 state = EventsHelper.shared.getState(of: eventGuid)
             }
         }