add server option EnableProxyProtocol

kerryjiang · Aug 10, 2024 · 4a924e6 · 4a924e6
1 parent 2b191a2
commit 4a924e6
Show file tree

Hide file tree

Showing 6 changed files with 11 additions and 12 deletions.
diff --git a/src/SuperSocket.ProtoBase/Extensions.cs b/src/SuperSocket.ProtoBase/Extensions.cs
@@ -148,11 +148,5 @@ public static int Write(this IBufferWriter<byte> writer, ReadOnlySpan<char> text
 
             return totalBytes;
         }
-
-        public static IPipelineFilterFactory<TPackageInfo> UseProxyProtocol<TPackageInfo>(this IPipelineFilterFactory<TPackageInfo> pipelineFilterFactory)
-            where TPackageInfo : class
-        {
-            return new ProxyProtocolPipelineFilterFactory<TPackageInfo>(pipelineFilterFactory);
-        }
     }
 }
diff --git a/src/SuperSocket.ProtoBase/PackagePartsPipelineFilter.cs b/src/SuperSocket.ProtoBase/PackagePartsPipelineFilter.cs
@@ -3,7 +3,6 @@
 namespace SuperSocket.ProtoBase
 {
     public abstract class PackagePartsPipelineFilter<TPackageInfo> : IPipelineFilter<TPackageInfo>
-        where TPackageInfo : class
     {
         private IPackagePartReader<TPackageInfo> _currentPartReader;
 
@@ -40,7 +39,7 @@ public virtual TPackageInfo Filter(SequenceReader<byte> reader)
                 }
 
                 if (needMoreData || reader.Remaining <= 0)
-                    return null;
+                    return default;
             }
         }
 
@@ -51,7 +50,7 @@ protected virtual void OnPartReaderSwitched(IPackagePartReader<TPackageInfo> cur
 
         public virtual void Reset()
         {
-            CurrentPackage = null;
+            CurrentPackage = default;
             _currentPartReader = null;
         }
 

diff --git a/src/SuperSocket.ProtoBase/ProxyProtocol/ProxyProtocolPipelineFilter.cs b/src/SuperSocket.ProtoBase/ProxyProtocol/ProxyProtocolPipelineFilter.cs
@@ -4,7 +4,6 @@
 namespace SuperSocket.ProtoBase.ProxyProtocol
 {
     public class ProxyProtocolPipelineFilter<TPackageInfo> : PackagePartsPipelineFilter<TPackageInfo>
-        where TPackageInfo : class
     {
         private readonly IPipelineFilter<TPackageInfo> _applicationPipelineFilter;
 

diff --git a/src/SuperSocket.ProtoBase/ProxyProtocol/ProxyProtocolPipelineFilterFactory.cs b/src/SuperSocket.ProtoBase/ProxyProtocol/ProxyProtocolPipelineFilterFactory.cs
@@ -4,7 +4,6 @@
 namespace SuperSocket.ProtoBase.ProxyProtocol
 {
     public class ProxyProtocolPipelineFilterFactory<TPackageInfo> : IPipelineFilterFactory<TPackageInfo>
-        where TPackageInfo : class
     {
         private readonly IPipelineFilterFactory<TPackageInfo> _pipelineFilterFactory;
 

diff --git a/src/SuperSocket.Server.Abstractions/ServerOptions.cs b/src/SuperSocket.Server.Abstractions/ServerOptions.cs
@@ -20,5 +20,7 @@ public class ServerOptions : ConnectionOptions
         /// In seconds.
         /// </summary>
         public int PackageHandlingTimeOut { get; set; } = 30;
+
+        public bool EnableProxyProtocol { get; set; }
     }
 }
diff --git a/src/SuperSocket.Server/SuperSocketService.cs b/src/SuperSocket.Server/SuperSocketService.cs
@@ -15,6 +15,7 @@
 using SuperSocket.Server.Abstractions.Session;
 using SuperSocket.Server.Abstractions.Middleware;
 using SuperSocket.Server.Connection;
+using SuperSocket.ProtoBase.ProxyProtocol;
 
 namespace SuperSocket.Server
 {
@@ -113,7 +114,12 @@ public SuperSocketService(IServiceProvider serviceProvider, IOptions<ServerOptio
 
         protected virtual IPipelineFilterFactory<TReceivePackageInfo> GetPipelineFilterFactory()
         {
-            return _serviceProvider.GetRequiredService<IPipelineFilterFactory<TReceivePackageInfo>>();
+            var filterFactory = _serviceProvider.GetRequiredService<IPipelineFilterFactory<TReceivePackageInfo>>();
+
+            if (Options.EnableProxyProtocol)
+                filterFactory = new ProxyProtocolPipelineFilterFactory<TReceivePackageInfo>(filterFactory);
+
+            return filterFactory;
         }
 
         private bool AddConnectionListener(ListenOptions listenOptions, ServerOptions serverOptions)