CERBERUS/kitty
1
0
Fork 0
mirror of https://github.com/kovidgoyal/kitty.git synced 2026-07-11 02:03:22 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

Also prevent long running diff workers from causing a hang on exit

Browse source
This commit is contained in:
Kovid Goyal 2018-08-04 17:51:33 +05:30
parent 927bfcd37a
commit e05d48a574
No known key found for this signature in database
GPG key ID: 06BC317B515ACE7C
2 changed files with 11 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

14
kittens/diff/main.py
View file

@ -25,7 +25,7 @@ from .collect import (
set_highlight_data
)
from .config import init_config
from .patch import Differ, set_diff_command
from .patch import Differ, set_diff_command, worker_processes
from .render import ImageSupportWarning, LineRef, render_diff
from .search import BadRegex, Search
@ -511,9 +511,10 @@ usage = 'file_or_directory_left file_or_directory_right'
def terminate_processes(processes):
for pid in processes:
os.kill(pid, signal.SIGTERM)
for pid in processes:
os.kill(pid, signal.SIGKILL)
try:
os.kill(pid, signal.SIGKILL)
except Exception:
pass
def main(args):
@ -538,8 +539,9 @@ def main(args):
for message in showwarning.warnings:
from kitty.utils import safe_print
safe_print(message, file=sys.stderr)
processes = getattr(highlight_collection, 'processes', ())
terminate_processes(tuple(processes))
highlight_processes = getattr(highlight_collection, 'processes', ())
terminate_processes(tuple(highlight_processes))
terminate_processes(tuple(worker_processes))
if loop.return_code != 0:
if handler.report_traceback_on_exit:
print(handler.report_traceback_on_exit, file=sys.stderr)

3
kittens/diff/patch.py
View file

@ -14,6 +14,7 @@ from .diff_speedup import changed_center
left_lines = right_lines = None
GIT_DIFF = 'git diff --no-color --no-ext-diff --exit-code -U_CONTEXT_ --no-index --'
DIFF_DIFF = 'diff -p -U _CONTEXT_ --'
worker_processes = []
def find_differ():
@ -43,8 +44,10 @@ def run_diff(file1, file2, context=3):
p = subprocess.Popen(
cmd + [path1, path2],
stdout=subprocess.PIPE, stderr=subprocess.PIPE, stdin=subprocess.DEVNULL)
worker_processes.append(p.pid)
stdout, stderr = p.communicate()
returncode = p.wait()
worker_processes.remove(p.pid)
if returncode in (0, 1):
return True, returncode == 1, stdout.decode('utf-8')
return False, returncode, stderr.decode('utf-8')