Filtered rusi rpc status codes for host restart (#190)

osstotalsoft · Nov 9, 2021 · a2876ec · a2876ec
1 parent f36b431
commit a2876ec
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/Messaging/NBB.Messaging.Rusi/RusiMessagingTransport.cs b/src/Messaging/NBB.Messaging.Rusi/RusiMessagingTransport.cs
@@ -98,7 +98,7 @@ public async Task<IDisposable> SubscribeAsync(string topic, Func<TransportReceiv
             {
                 try
                 {
-                    await foreach (var msg in subscription.ResponseStream.ReadAllAsync())
+                    await foreach (var msg in subscription.ResponseStream.ReadAllAsync(cancellationToken))
                     {
                         var receiveContext = new TransportReceiveContext(
                             new TransportReceivedData.PayloadBytesAndHeaders(msg.Data.ToByteArray(), msg.Metadata));
@@ -118,7 +118,7 @@ public async Task<IDisposable> SubscribeAsync(string topic, Func<TransportReceiv
                         });
                     }
                 }
-                catch (RpcException ex)
+                catch (RpcException ex) when (new[] { StatusCode.Unavailable, StatusCode.Aborted }.Contains(ex.StatusCode))
                 {
                     _logger.LogError(ex, "Rusi transport unrecoverable exception");