KittyPatch/kitty
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Increase the timeout when talking to prewarm process

Browse Source
This commit is contained in:
Kovid Goyal 2022-06-08 19:12:04 +05:30
parent 7001d609fa
commit e338cdab31
No known key found for this signature in database
GPG Key ID: 06BC317B515ACE7C
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
kitty/prewarm.py
View File

@ -29,6 +29,7 @@ if TYPE_CHECKING:
error_events = select.POLLERR | select.POLLNVAL | select.POLLHUP
TIMEOUT = 4.0
class PrewarmProcessFailed(Exception):
@ -121,7 +122,8 @@ class PrewarmProcess:
argv: List[str],
cwd: str = '',
env: Optional[Dict[str, str]] = None,
stdin_data: Optional[Union[str, bytes]] = None
stdin_data: Optional[Union[str, bytes]] = None,
timeout: float = TIMEOUT,
) -> Child:
self.ensure_worker()
tty_name = os.ttyname(tty_fd)
@ -146,7 +148,7 @@ class PrewarmProcess:
self.send_to_prewarm_process(f'fork:{shm.name}\n')
input_buf = b''
st = time.monotonic()
while time.monotonic() - st < 2:
while time.monotonic() - st < timeout:
for (fd, event) in self.poll.poll(0.2):
if event & error_events:
raise PrewarmProcessFailed('Failed doing I/O with prewarm process')
@ -169,7 +171,7 @@ class PrewarmProcess:
self.children[child_id] = c = Child(child_id, pid)
return c
def send_to_prewarm_process(self, output_buf: Union[str, bytes] = b'', timeout: float = 2) -> None:
def send_to_prewarm_process(self, output_buf: Union[str, bytes] = b'', timeout: float = TIMEOUT) -> None:
if isinstance(output_buf, str):
output_buf = output_buf.encode()
st = time.monotonic()