更快的速度查找record的引用

shuying2244 · Oct 24, 2024 · b4743cf · b4743cf
1 parent 43206ff
commit b4743cf
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/apps/els_lsp/src/els_dt_document.erl b/apps/els_lsp/src/els_dt_document.erl
@@ -364,6 +364,8 @@ tokens_to_words([{string, _Location, String} | Tokens], Words) ->
         _ ->
             tokens_to_words(Tokens, Words)
     end;
+tokens_to_words([{var, _Location, Var} | Tokens], Words) ->
+    tokens_to_words(Tokens, sets:add_element(Var, Words));
 tokens_to_words([{'?', _}, {var, _, Macro} | Tokens], Words) ->
     tokens_to_words(Tokens, sets:add_element(Macro, Words));
 tokens_to_words([{'-', _}, {atom, _, define}, {'(', _}, {var, _, Macro} | Tokens], Words) ->

diff --git a/apps/els_lsp/src/els_references_provider.erl b/apps/els_lsp/src/els_references_provider.erl
@@ -98,12 +98,12 @@ find_references(Uri, #{
     {F, A, _Index} = Id,
     Key = {els_uri:module(Uri), F, A},
     find_references_for_id(Kind, Key);
-find_references(Uri, POI = #{kind := Kind}) when
+find_references(_Uri, #{kind := Kind, id := Key}) when
     Kind =:= record;
     Kind =:= record_def_field;
     Kind =:= define
 ->
-    find_scoped_references_for_def(Uri, POI);
+    find_references_for_id(Kind, Key);
 find_references(Uri, POI = #{kind := Kind, id := Id}) when
     Kind =:= type_definition
 ->

diff --git a/apps/els_lsp/src/els_text_search.erl b/apps/els_lsp/src/els_text_search.erl
@@ -44,8 +44,8 @@ extract_pattern({behaviour, Name}) ->
     Name;
 extract_pattern({atom, Name}) ->
     Name;
-extract_pattern({record_field, {_Record, Name}}) ->
-    Name;
+extract_pattern({record_field, {Record, _Name}}) ->
+    Record;
 extract_pattern({record, Name}) ->
     Name.