22
33from ..port_manager import PortManager
44from ..exceptions import PortForException
5+ from ..import consts
56
7+ import os
68import threading
79import random
810import typing
@@ -15,6 +17,8 @@ class PortManager__Generic(PortManager):
1517_available_ports :typing .Set [int ]
1618_reserved_ports :typing .Set [int ]
1719
20+ _lock_dir :str
21+
1822def __init__ (self ,os_ops :OsOperations ):
1923assert os_ops is not None
2024assert isinstance (os_ops ,OsOperations )
@@ -23,6 +27,12 @@ def __init__(self, os_ops: OsOperations):
2327self ._available_ports :typing .Set [int ]= set (range (1024 ,65535 ))
2428self ._reserved_ports :typing .Set [int ]= set ()
2529
30+ temp_dir = os_ops .get_tempdir ()
31+ assert type (temp_dir )== str # noqa: E721
32+ self ._lock_dir = os .path .join (temp_dir ,consts .TMP_TESTGRES_PORTS )
33+ assert type (self ._lock_dir )== str # noqa: E721
34+ os_ops .makedirs (self ._lock_dir )
35+
2636def reserve_port (self )-> int :
2737assert self ._guard is not None
2838assert type (self ._available_ports )== set # noqa: E721t
@@ -41,9 +51,23 @@ def reserve_port(self) -> int:
4151if not self ._os_ops .is_port_free (port ):
4252continue
4353
44- self ._reserved_ports .add (port )
45- self ._available_ports .discard (port )
54+ try :
55+ lock_path = self .helper__make_lock_path (port )
56+ self ._os_ops .makedir (lock_path )
57+ except :# noqa: 722
58+ continue
59+
60+ assert self ._os_ops .path_exists (lock_path )
61+
62+ try :
63+ self ._reserved_ports .add (port )
64+ except :# noqa: 722
65+ assert not (port in self ._reserved_ports )
66+ self ._os_ops .rmdir (lock_path )
67+ raise
68+
4669assert port in self ._reserved_ports
70+ self ._available_ports .discard (port )
4771assert not (port in self ._available_ports )
4872return port
4973
@@ -55,10 +79,26 @@ def release_port(self, number: int) -> None:
5579assert self ._guard is not None
5680assert type (self ._reserved_ports )== set # noqa: E721
5781
82+ lock_path = self .helper__make_lock_path (number )
83+
5884with self ._guard :
5985assert number in self ._reserved_ports
6086assert not (number in self ._available_ports )
6187self ._available_ports .add (number )
6288self ._reserved_ports .discard (number )
6389assert not (number in self ._reserved_ports )
6490assert number in self ._available_ports
91+
92+ assert isinstance (self ._os_ops ,OsOperations )
93+ assert self ._os_ops .path_exists (lock_path )
94+ self ._os_ops .rmdir (lock_path )
95+
96+ return
97+
98+ def helper__make_lock_path (self ,port_number :int )-> str :
99+ assert type (port_number )== int # noqa: E721
100+ assert type (self ._lock_dir )== str # noqa: E721
101+
102+ result = os .path .join (self ._lock_dir ,str (port_number )+ ".lock" )
103+ assert type (result )== str # noqa: E721
104+ return result