LBRYCommunity/lbry-sdk
1
0
Fork 1
Code Issues Pull requests Projects Releases 282 Packages Wiki Activity

take load into consideration when picking fastest

Browse source
This commit is contained in:
Victor Shyba 2019-08-19 13:30:45 -03:00
parent 20af1396b9
commit c826c7da0d
1 changed files with 8 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
torba/torba/client/basenetwork.py
View file

@ -20,7 +20,8 @@ class ClientSession(BaseClientSession):
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)
self._on_disconnect_controller = StreamController() self._on_disconnect_controller = StreamController()
self.on_disconnected = self._on_disconnect_controller.stream self.on_disconnected = self._on_disconnect_controller.stream
self.bw_limit = self.framer.max_size = self.max_errors = 1 << 32 self.framer.max_size = self.max_errors = 1 << 32
self.bw_limit = -1
self.timeout = timeout self.timeout = timeout
self.max_seconds_idle = timeout * 2 self.max_seconds_idle = timeout * 2
self.response_time: Optional[float] = None self.response_time: Optional[float] = None
@ -31,6 +32,10 @@ class ClientSession(BaseClientSession):
def available(self): def available(self):
return not self.is_closing() and self._can_send.is_set() and self.response_time is not None return not self.is_closing() and self._can_send.is_set() and self.response_time is not None
@property
def pending_size(self):
return len(self.connection.pending_requests())
async def send_request(self, method, args=()): async def send_request(self, method, args=()):
try: try:
start = perf_counter() start = perf_counter()
@ -215,7 +220,8 @@ class SessionPool:
if not self.available_sessions: if not self.available_sessions:
return None return None
return min( return min(
[(session.response_time, session) for session in self.available_sessions], key=itemgetter(0) [(session.response_time * session.pending_size, session) for session in self.available_sessions],
key=itemgetter(0)
)[1] )[1]
def start(self, default_servers): def start(self, default_servers):