neovim

commit 816b56f878f0291c00a9018d5057b7b2b00f1891
parent bf1e098d971d455cba1bdc9f6b2e777572fd6b16
Author: Lewis Russell <lewis6991@gmail.com>
Date:   Wed,  7 Feb 2024 13:05:33 +0000

fix(lsp): cancel watchers when closing a client

Diffstat:
M
runtime/lua/vim/lsp/_watchfiles.lua
 | 
9
+++++++++
M
runtime/lua/vim/lsp/client.lua
 | 
1
+
M
test/functional/plugin/lsp_spec.lua
 | 
4
+++-

3 files changed, 13 insertions(+), 1 deletion(-)
diff --git a/runtime/lua/vim/lsp/_watchfiles.lua b/runtime/lua/vim/lsp/_watchfiles.lua
@@ -167,5 +167,14 @@ function M.unregister(unreg, ctx)
   end
 end
 
+--- @param client_id integer
+function M.cancel(client_id)
+  for _, reg_cancels in pairs(cancels[client_id]) do
+    for _, cancel in pairs(reg_cancels) do
+      cancel()
+    end
+  end
+end
+
 return M
 
diff --git a/runtime/lua/vim/lsp/client.lua b/runtime/lua/vim/lsp/client.lua
@@ -727,6 +727,7 @@ function Client:_stop(force)
       rpc.terminate()
       self._graceful_shutdown_failed = true
     end
+    vim.lsp._watchfiles.cancel(self.id)
   end)
 end
 
diff --git a/test/functional/plugin/lsp_spec.lua b/test/functional/plugin/lsp_spec.lua
@@ -4573,6 +4573,8 @@ describe('LSP', function()
 
           wait_for_message()
 
+          vim.lsp.stop_client(client_id)
+
           return server.messages
         ]],
             root_dir,
@@ -4581,7 +4583,7 @@ describe('LSP', function()
 
           local uri = vim.uri_from_fname(root_dir .. '/watch')
 
-          eq(4, #result)
+          eq(6, #result)
 
           eq({
             method = 'workspace/didChangeWatchedFiles',