Bump dedi interface version

BeatTogether · Apr 5, 2024 · 6e2d6c7 · 6e2d6c7
1 parent fa8cfdf
commit 6e2d6c7
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 15 deletions.
diff --git a/BeatTogether.DedicatedServer.Interface/BeatTogether.DedicatedServer.Interface.csproj b/BeatTogether.DedicatedServer.Interface/BeatTogether.DedicatedServer.Interface.csproj
@@ -7,7 +7,7 @@
     <Authors>BeatTogether Team</Authors>
     <Company>BeatTogether</Company>
     <RepositoryUrl>https://github.com/beattogether/BeatTogether.DedicatedServer</RepositoryUrl>
-    <Version>1.7.1</Version>
+    <Version>1.7.2</Version>
     <Nullable>enable</Nullable>
   </PropertyGroup>
 

diff --git a/BeatTogether.DedicatedServer.Node/PortAllocator.cs b/BeatTogether.DedicatedServer.Node/PortAllocator.cs
@@ -9,6 +9,8 @@ public sealed class PortAllocator : IPortAllocator
     {
         private readonly NodeConfiguration _configuration;
 
+        private readonly object _lock = new object();
+
         private readonly HashSet<int> _acquiredPorts = new();
         private readonly HashSet<int> _releasedPorts = new();
 
@@ -24,27 +26,34 @@ public PortAllocator(
 
         public int? AcquirePort()
         {
-            if (_acquiredPorts.Count >= _configuration.MaximumSlots)
-                return null;
-            int port;
-            if (_releasedPorts.Any())
+            lock (_lock)
             {
-                port = _releasedPorts.First();
-                _releasedPorts.Remove(port);
+                if (_acquiredPorts.Count >= _configuration.MaximumSlots)
+                    return null;
+                int port;
+                if (_releasedPorts.Any())
+                {
+                    port = _releasedPorts.First();
+                    _releasedPorts.Remove(port);
+                }
+                else
+                    port = ++_lastPort;
+                _acquiredPorts.Add(port);
+                return port;
             }
-            else
-                port = ++_lastPort;
-            _acquiredPorts.Add(port);
-            return port;
+
         }
 
         public bool ReleasePort(int port)
         {
+            lock (_lock)
+            {
+                if (!_acquiredPorts.Remove(port))
+                    return false;
+                _releasedPorts.Add(port);
+                return true;
+            }
 
-            if (!_acquiredPorts.Remove(port))
-                return false;
-            _releasedPorts.Add(port);
-            return true;
         }
     }
 }