Workaround for ReadAllAsync()

planetarium · Nov 25, 2024 · 7c26f68 · 7c26f68
1 parent dd93cf4
commit 7c26f68
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/src/Libplanet.Net/Transports/NetMQTransport.cs b/src/Libplanet.Net/Transports/NetMQTransport.cs
@@ -662,10 +662,26 @@ await ProcessMessageHandler.InvokeAsync(
                                         channel.Writer.Complete();
                                     }
 
+#if NETCOREAPP3_0 || NETCOREAPP3_1 || NET
                                     await foreach (
                                         var messageContent in channel.Reader.ReadAllAsync(
                                             _runtimeCancellationTokenSource.Token))
                                     {
+#else
+                                    while (true)
+                                    {
+                                        MessageContent messageContent;
+                                        try
+                                        {
+                                            messageContent = await channel.Reader.ReadAsync(
+                                                _runtimeCancellationTokenSource.Token);
+                                        }
+                                        catch (ChannelClosedException)
+                                        {
+                                            break;
+                                        }
+
+#endif
                                         await ReplyMessageAsync(
                                             messageContent,
                                             message.Identity ?? Array.Empty<byte>(),