Added websocket connection ratelimiter, updated CryptoExchange.Net ve…

…rsion to fix some reconnect issues
JKorf · Aug 25, 2024 · 7d02624 · 7d02624
1 parent 5ec9fab
commit 7d02624
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/BitMart.Net/BitMart.Net.csproj b/BitMart.Net/BitMart.Net.csproj
@@ -1,4 +1,4 @@
-<Project Sdk="Microsoft.NET.Sdk">
+<Project Sdk="Microsoft.NET.Sdk">
   <PropertyGroup>
     <TargetFrameworks>netstandard2.0;netstandard2.1</TargetFrameworks>
     <LangVersion>10.0</LangVersion>
@@ -48,10 +48,10 @@
       <PrivateAssets>all</PrivateAssets>
       <IncludeAssets>runtime; build; native; contentfiles; analyzers; buildtransitive</IncludeAssets>
     </PackageReference>
+    <PackageReference Include="CryptoExchange.Net" Version="7.11.1" />
     <PackageReference Include="Microsoft.CodeAnalysis.NetAnalyzers" Version="8.0.0">
       <PrivateAssets>all</PrivateAssets>
       <IncludeAssets>runtime; build; native; contentfiles; analyzers; buildtransitive</IncludeAssets>
     </PackageReference>
-    <PackageReference Include="CryptoExchange.Net" Version="7.11.0" />
   </ItemGroup>
 </Project>
diff --git a/BitMart.Net/BitMartExchange.cs b/BitMart.Net/BitMartExchange.cs
@@ -58,10 +58,14 @@ private void Initialize()
         {
             BitMart = new RateLimitGate("BitMart IP");
             BitMart.RateLimitTriggered += (x) => RateLimitTriggered?.Invoke(x);
+            SocketLimits = new RateLimitGate("Socket limits")
+                .AddGuard(new RateLimitGuard(RateLimitGuard.PerHost, new LimitItemTypeFilter(RateLimitItemType.Connection), 100, TimeSpan.FromSeconds(60), RateLimitWindowType.Sliding, connectionWeight: 1));
+            SocketLimits.RateLimitTriggered += (x) => RateLimitTriggered?.Invoke(x);
         }
 
 
         internal IRateLimitGate BitMart { get; private set; }
+        internal IRateLimitGate SocketLimits { get; private set; }
 
     }
 }
diff --git a/BitMart.Net/Clients/SpotApi/BitMartSocketClientSpotApi.cs b/BitMart.Net/Clients/SpotApi/BitMartSocketClientSpotApi.cs
@@ -47,8 +47,8 @@ internal BitMartSocketClientSpotApi(ILogger logger, BitMartSocketOptions options
             base(logger, options.Environment.SocketClientSpotAddress!, options, options.SpotOptions)
         {
             KeepAliveInterval = TimeSpan.Zero;
-
             RegisterPeriodicQuery("ping", TimeSpan.FromSeconds(15), x => new PingQuery(), null);
+            RateLimiter = BitMartExchange.RateLimiter.SocketLimits;
         }
         #endregion