feat: Added PipeServer.CreatePipeStreamForConnectionFunc.

HavenDV · Dec 8, 2023 · bd85946 · bd85946
1 parent 5048d67
commit bd85946
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/libs/H.Pipes/PipeServer.cs b/src/libs/H.Pipes/PipeServer.cs
@@ -26,6 +26,11 @@ public sealed class PipeServer<T> : IPipeServer<T>
     /// </summary>
     public Func<string, NamedPipeServerStream>? CreatePipeStreamFunc { get; set; }
 
+    /// <summary>
+    /// If set, used instead of CreatePipeStreamFunc for connections
+    /// </summary>
+    public Func<string, NamedPipeServerStream>? CreatePipeStreamForConnectionFunc { get; set; }
+
     /// <summary>
     /// PipeStreamInitializeAction
     /// </summary>
@@ -196,7 +201,7 @@ await handshakeWrapper.WriteAsync(Encoding.UTF8.GetBytes(connectionPipeName), to
                     }
 
                     // Wait for the client to connect to the data pipe
-                    var connectionStream = CreatePipeStreamFunc?.Invoke(connectionPipeName) ?? PipeServerFactory.Create(connectionPipeName);
+                    var connectionStream = (CreatePipeStreamForConnectionFunc ?? CreatePipeStreamFunc)?.Invoke(connectionPipeName) ?? PipeServerFactory.Create(connectionPipeName);
 
                     PipeStreamInitializeAction?.Invoke(connectionStream);