From d2572c9f3836f8734ec37c5e5f59bb466a713882 Mon Sep 17 00:00:00 2001
From: smathot <s.mathot@cogsci.nl>
Date: Wed, 23 Dec 2020 11:42:36 +0100
Subject: [PATCH] LspEndpoint: add timeout

---
 pylspclient/lsp_endpoint.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/pylspclient/lsp_endpoint.py b/pylspclient/lsp_endpoint.py
index 8364d6f..28b1a2c 100644
--- a/pylspclient/lsp_endpoint.py
+++ b/pylspclient/lsp_endpoint.py
@@ -5,7 +5,7 @@ from pylspclient import lsp_structs
 
 
 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)
         self.json_rpc_endpoint = json_rpc_endpoint
         self.notify_callbacks = notify_callbacks
@@ -13,6 +13,7 @@ class LspEndpoint(threading.Thread):
         self.event_dict = {}
         self.response_dict = {}
         self.next_id = 0
+        self._timeout = timeout
         self.shutdown_flag = False
 
 
@@ -93,7 +94,8 @@ class LspEndpoint(threading.Thread):
         if self.shutdown_flag:
             return None
 
-        cond.wait()
+        if not cond.wait(timeout=self._timeout):
+            raise TimeoutError()
         cond.release()
 
         self.event_dict.pop(current_id)