fix(real time read): added check to assert clinet is still connected …

…and reconnect if not when fetching new data resolves PRODSUP-4052
naveego · May 24, 2023 · c9f7947 · c9f7947
1 parent 3f13556
commit c9f7947
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/PluginSalesforce/API/Factory/PushTopicConnection.cs b/PluginSalesforce/API/Factory/PushTopicConnection.cs
@@ -34,6 +34,10 @@ public void Disconnect()
 
         public async IAsyncEnumerable<string> GetCurrentMessages()
         {
+            if (!_bayeuxClient.Connected) {
+                Connect();
+            }
+
             var messages = _listener.GetMessages();
 
             foreach (var message in messages)

diff --git a/PluginSalesforce/API/Read/ReadRecordsRealTime.cs b/PluginSalesforce/API/Read/ReadRecordsRealTime.cs
@@ -49,6 +49,7 @@ public static async Task<int> ReadRecordsRealTimeAsync(RequestHelper client, Rea
                 ? JsonConvert.DeserializeObject<RealTimeState>(request.RealTimeStateJson)
                 : new RealTimeState();
 
+            // build cometd client
             var conn = connectionFactory.GetPushTopicConnection(client, @"/topic/" + realTimeSettings.ChannelName);
 
             try
@@ -61,11 +62,10 @@ public static async Task<int> ReadRecordsRealTimeAsync(RequestHelper client, Rea
 
                 using (var db = new LiteDatabase(Path.Join(path, $"{jobId}_RealTimeReadRecords.db")))
                 {
-                    var realtimeRecordsCollection = db.GetCollection<RealTimeRecord>(CollectionName);
-
-                    // build cometd client
                     conn.Connect();
-
+
+                    var realtimeRecordsCollection = db.GetCollection<RealTimeRecord>(CollectionName);
+
                     // a full init needs to happen
                     if (jobVersion > realTimeState.JobVersion || shapeVersion > realTimeState.ShapeVersion)
                     {