refactor core modules, integrate network stats
This commit is contained in:
@@ -19,6 +19,7 @@
|
||||
# https://www.gnu.org/licenses/old-licenses/lgpl-2.1.en.html
|
||||
|
||||
import socket, ssl, select, copy, errno
|
||||
import network_stats
|
||||
|
||||
# rs_proxyType
|
||||
RS_PT_NONE = 0
|
||||
@@ -310,6 +311,7 @@ class Rocksock():
|
||||
if self.sock is None:
|
||||
raise(RocksockException(RS_E_NO_SOCKET, failedproxy=self._failed_proxy(pnum)))
|
||||
try:
|
||||
network_stats.add_tx(len(buf))
|
||||
return self.sock.sendall(buf)
|
||||
except socket.error as e:
|
||||
raise(self._translate_socket_error(e, pnum))
|
||||
@@ -340,6 +342,7 @@ class Rocksock():
|
||||
raise(RocksockException(RS_E_SSL_GENERIC, failedproxy=s, errortype=RS_ET_SSL))
|
||||
if len(chunk) == 0:
|
||||
raise(RocksockException(RS_E_REMOTE_DISCONNECTED, failedproxy=self._failed_proxy(pnum)))
|
||||
network_stats.add_rx(len(chunk))
|
||||
data += chunk
|
||||
if count == -1: break
|
||||
else: count -= len(chunk)
|
||||
|
||||
Reference in New Issue
Block a user