diff --git a/lib/next_ls.ex b/lib/next_ls.ex index 4b9f1cf0..ee7044e3 100644 --- a/lib/next_ls.ex +++ b/lib/next_ls.ex @@ -454,21 +454,12 @@ defmodule NextLS do refresh_refs = dispatch(lsp.assigns.registry, :runtimes, fn entries -> - for {pid, %{name: name, uri: wuri, db: db}} <- entries, String.starts_with?(uri, wuri), into: %{} do + for {pid, %{name: name, uri: wuri}} <- entries, String.starts_with?(uri, wuri), into: %{} do token = Progress.token() Progress.start(lsp, token, "Compiling #{name}...") task = Task.Supervisor.async_nolink(lsp.assigns.task_supervisor, fn -> - DB.query( - db, - ~Q""" - DELETE FROM 'references' - WHERE file = ?; - """, - [URI.parse(uri).path] - ) - {name, Runtime.compile(pid)} end) diff --git a/lib/next_ls/db.ex b/lib/next_ls/db.ex index 749069b6..293e9a00 100644 --- a/lib/next_ls/db.ex +++ b/lib/next_ls/db.ex @@ -19,6 +19,9 @@ defmodule NextLS.DB do @spec insert_reference(pid(), map()) :: :ok def insert_reference(server, payload), do: GenServer.cast(server, {:insert_reference, payload}) + @spec clean_references(pid(), String.t()) :: :ok + def clean_references(server, filename), do: GenServer.cast(server, {:clean_references, filename}) + def init(args) do file = Keyword.fetch!(args, :file) registry = Keyword.fetch!(args, :registry) @@ -137,6 +140,22 @@ defmodule NextLS.DB do {:noreply, s} end + def handle_cast({:clean_references, filename}, %{conn: conn} = s) do + {:message_queue_len, count} = Process.info(self(), :message_queue_len) + NextLS.DB.Activity.update(s.activity, count) + + __query__( + {conn, s.logger}, + ~Q""" + DELETE FROM 'references' + WHERE file = ?; + """, + [filename] + ) + + {:noreply, s} + end + def __query__({conn, logger}, query, args) do args = Enum.map(args, &cast/1) diff --git a/lib/next_ls/runtime/sidecar.ex b/lib/next_ls/runtime/sidecar.ex index c0b0d062..64ec9626 100644 --- a/lib/next_ls/runtime/sidecar.ex +++ b/lib/next_ls/runtime/sidecar.ex @@ -25,4 +25,10 @@ defmodule NextLS.Runtime.Sidecar do {:noreply, state} end + + def handle_info({{:tracer, :start}, filename}, state) do + DB.clean_references(state.db, filename) + + {:noreply, state} + end end diff --git a/priv/monkey/_next_ls_private_compiler.ex b/priv/monkey/_next_ls_private_compiler.ex index 63e59164..e382a96e 100644 --- a/priv/monkey/_next_ls_private_compiler.ex +++ b/priv/monkey/_next_ls_private_compiler.ex @@ -52,7 +52,13 @@ defmodule NextLSPrivate.Tracer do @source "user" - def trace(:start, _env) do + def trace(:start, env) do + Process.send( + parent_pid(), + {{:tracer, :start}, env.file}, + [] + ) + :ok end