username/ppf
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

rocksock: skip shutdown on never-connected sockets

Browse Source 
Track connection state with _connected flag. Only call
socket.shutdown() on successfully connected sockets.
Saves ~39s/session on workers (974k disconnect calls).
This commit is contained in:
Username
2026-02-22 13:58:20 +01:00
parent 2ea7eb41b7
commit 0a53e4457f
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
rocksock.py
View File

@@ -242,6 +242,7 @@ class Rocksock():
target = RocksockProxy(host, port, RS_PT_NONE) target = RocksockProxy(host, port, RS_PT_NONE)
self.proxychain.append(target) self.proxychain.append(target)
self.sock = None self.sock = None
self._connected = False
self.timeout = timeout self.timeout = timeout
def _translate_socket_error(self, e, pnum): def _translate_socket_error(self, e, pnum):
@@ -302,15 +303,18 @@ class Rocksock():
select.select([], [self.sock], []) select.select([], [self.sock], [])
""" """
self._connected = True
def disconnect(self): def disconnect(self):
if self.sock is None: return if self.sock is None: return
try: if self._connected:
self.sock.shutdown(socket.SHUT_RDWR) try:
except socket.error: self.sock.shutdown(socket.SHUT_RDWR)
pass except socket.error:
pass
self.sock.close() self.sock.close()
self.sock = None self.sock = None
self._connected = False
def canread(self): def canread(self):
return select.select([self.sock], [], [], 0)[0] return select.select([self.sock], [], [], 0)[0]