MNTR fix from akkadotnet#4139

IgorFedchenko · Jan 9, 2020 · f1c5386 · f1c5386
1 parent f1a044d
commit f1c5386
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/src/core/Akka.NodeTestRunner/Program.cs b/src/core/Akka.NodeTestRunner/Program.cs
@@ -161,16 +161,32 @@ private static Assembly DefaultOnResolving(AssemblyLoadContext assemblyLoadConte
 
     class RunnerTcpClient : ReceiveActor, IWithUnboundedStash
     {
+        private IActorRef _connection;
+
         public RunnerTcpClient()
         {
             Become(WaitingForConnection);
         }
 
+        /// <inheritdoc />
+        protected override void PostStop()
+        {
+            // Close connection property to avoid exception logged at TcpConnection actor once this actor is terminated
+            try
+            {
+                _connection.Ask<Tcp.Closed>(Tcp.Close.Instance, TimeSpan.FromSeconds(1)).Wait();
+            }
+            catch { /* well... at least we have tried */ }
+
+            base.PostStop();
+        }
+
         private void WaitingForConnection()
         {
             Receive<Tcp.Connected>(connected =>
             {
                 Sender.Tell(new Tcp.Register(Self));
+                _connection = Sender;
                 Become(Connected(Sender));
             });
             Receive<string>(_ => Stash.Stash());