itdominator/PyLSPClient
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

LspEndpoint: add timeout

Browse Source
This commit is contained in:
smathot 2020-12-23 11:42:36 +01:00
parent a10ed5327b
commit d2572c9f38
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
pylspclient/lsp_endpoint.py
View File

@ -5,7 +5,7 @@ from pylspclient import lsp_structs
class LspEndpoint(threading.Thread): class LspEndpoint(threading.Thread):
def __init__(self, json_rpc_endpoint, method_callbacks={}, notify_callbacks={}): def __init__(self, json_rpc_endpoint, method_callbacks={}, notify_callbacks={}, timeout=2):
threading.Thread.__init__(self) threading.Thread.__init__(self)
self.json_rpc_endpoint = json_rpc_endpoint self.json_rpc_endpoint = json_rpc_endpoint
self.notify_callbacks = notify_callbacks self.notify_callbacks = notify_callbacks
@ -13,6 +13,7 @@ class LspEndpoint(threading.Thread):
self.event_dict = {} self.event_dict = {}
self.response_dict = {} self.response_dict = {}
self.next_id = 0 self.next_id = 0
self._timeout = timeout
self.shutdown_flag = False self.shutdown_flag = False
@ -93,7 +94,8 @@ class LspEndpoint(threading.Thread):
if self.shutdown_flag: if self.shutdown_flag:
return None return None
cond.wait() if not cond.wait(timeout=self._timeout):
raise TimeoutError()
cond.release() cond.release()
self.event_dict.pop(current_id) self.event_dict.pop(current_id)