From b726141031ab14b10aa60d0920508cc952a68b26 Mon Sep 17 00:00:00 2001 From: sowmyasudhasingh Date: Wed, 19 Jan 2022 14:57:25 +0530 Subject: [PATCH] Fix broken `--local-executor` logic for windows ever since it was made default --- proxy/core/acceptor/acceptor.py | 7 ++++--- proxy/core/work/threadless.py | 2 +- proxy/proxy.py | 4 ++-- 3 files changed, 7 insertions(+), 6 deletions(-) diff --git a/proxy/core/acceptor/acceptor.py b/proxy/core/acceptor/acceptor.py index d83c5e00ae..8fa34afdf8 100644 --- a/proxy/core/acceptor/acceptor.py +++ b/proxy/core/acceptor/acceptor.py @@ -148,7 +148,8 @@ def run_once(self) -> None: if locked: self.lock.release() for work in works: - if self.flags.local_executor == int(DEFAULT_LOCAL_EXECUTOR): + if self.flags.threadless and \ + self.flags.local_executor: assert self._local_work_queue self._local_work_queue.put(work) else: @@ -171,7 +172,7 @@ def run(self) -> None: type=socket.SOCK_STREAM, ) try: - if self.flags.local_executor == int(DEFAULT_LOCAL_EXECUTOR): + if self.flags.threadless and self.flags.local_executor: self._start_local() self.selector.register(self.sock, selectors.EVENT_READ) while not self.running.is_set(): @@ -180,7 +181,7 @@ def run(self) -> None: pass finally: self.selector.unregister(self.sock) - if self.flags.local_executor == int(DEFAULT_LOCAL_EXECUTOR): + if self.flags.threadless and self.flags.local_executor: self._stop_local() self.sock.close() logger.debug('Acceptor#%d shutdown', self.idd) diff --git a/proxy/core/work/threadless.py b/proxy/core/work/threadless.py index 5ff7713766..e5c69a39c6 100644 --- a/proxy/core/work/threadless.py +++ b/proxy/core/work/threadless.py @@ -429,7 +429,7 @@ def run(self) -> None: data=wqfileno, ) assert self.loop - # logger.debug('Working on {0} works'.format(len(self.works))) + logger.debug('Working on {0} works'.format(len(self.works))) self.loop.create_task(self._run_forever()) self.loop.run_forever() except KeyboardInterrupt: diff --git a/proxy/proxy.py b/proxy/proxy.py index 9aed833f69..e2e2af1525 100644 --- a/proxy/proxy.py +++ b/proxy/proxy.py @@ -25,7 +25,7 @@ from .common.utils import bytes_ from .common.flag import FlagParser, flags -from .common.constants import DEFAULT_ENABLE_SSH_TUNNEL, DEFAULT_LOCAL_EXECUTOR, DEFAULT_LOG_FILE +from .common.constants import DEFAULT_ENABLE_SSH_TUNNEL, DEFAULT_LOG_FILE from .common.constants import DEFAULT_OPEN_FILE_LIMIT, DEFAULT_PLUGINS, DEFAULT_VERSION from .common.constants import DEFAULT_ENABLE_DASHBOARD, DEFAULT_WORK_KLASS, DEFAULT_PID_FILE from .common.constants import DEFAULT_LOG_FORMAT, DEFAULT_LOG_LEVEL, IS_WINDOWS @@ -242,7 +242,7 @@ def shutdown(self) -> None: @property def remote_executors_enabled(self) -> bool: return self.flags.threadless and \ - not (self.flags.local_executor == int(DEFAULT_LOCAL_EXECUTOR)) + not self.flags.local_executor def _write_pid_file(self) -> None: if self.flags.pid_file: