fixed close reason and loca/remote endpoints for KestrelPipeConnection

kerryjiang · Mar 30, 2024 · 9f8ae75 · 9f8ae75
1 parent 7e45414
commit 9f8ae75
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/SuperSocket.Kestrel/KestrelPipeConnection.cs b/src/SuperSocket.Kestrel/KestrelPipeConnection.cs
@@ -14,6 +14,9 @@ public KestrelPipeConnection(ConnectionContext context, ConnectionOptions option
         : base(context.Transport.Input, context.Transport.Output, options)
     {
         _context = context;
+        context.ConnectionClosed.Register(() => OnClosed());
+        LocalEndPoint = context.LocalEndPoint;
+        RemoteEndPoint = context.RemoteEndPoint;
     }
 
     protected override async void Close()
@@ -28,4 +31,12 @@ protected override async void Close()
             await context.DisposeAsync();
         }
     }
+
+    protected override void OnClosed()
+    {
+        if (!CloseReason.HasValue)
+            CloseReason = Connection.CloseReason.RemoteClosing;
+
+        base.OnClosed();
+    }
 }