Refactored inboundGovernorLoop error handling

IntersectMBO · Jan 14, 2022 · 654e9cf · 654e9cf
1 parent c13bdae
commit 654e9cf
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/ouroboros-network-framework/src/Ouroboros/Network/InboundGovernor.hs b/ouroboros-network-framework/src/Ouroboros/Network/InboundGovernor.hs
@@ -112,9 +112,10 @@ inboundGovernor trTracer tracer serverControlChannel inboundIdleTimeout
        (\(e :: SomeAsyncException) -> do
          state <- atomically $ readTVar st
          _ <- Map.traverseWithKey
-               (\connId _ ->
+               (\connId _ -> do
+                 let state' = unregisterConnection connId state
                  traceWith trTracer
-                           (mkRemoteTransitionTrace connId state emptyState)
+                           (mkRemoteTransitionTrace connId state state')
                )
                (igsConnections state)