Remove duplicate entries (#338)

nokia · Dec 9, 2021 · 2e4a473 · 2e4a473
1 parent c1dda4a
commit 2e4a473
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/internal/lsp/definition.go b/internal/lsp/definition.go
@@ -32,7 +32,7 @@ func (s *Server) definition(ctx context.Context, params *protocol.DefinitionPara
 		}
 	}
 
-	return locs, nil
+	return unifyLocs(locs), nil
 }
 
 func cTags(suite *ntt.Suite, file string, line int, col int) []protocol.Location {
@@ -70,3 +70,16 @@ func cTags(suite *ntt.Suite, file string, line int, col int) []protocol.Location
 	}
 	return ret
 }
+
+func unifyLocs(locs []protocol.Location) []protocol.Location {
+	m := make(map[protocol.Location]bool)
+	for _, loc := range locs {
+		m[loc] = true
+	}
+
+	ret := make([]protocol.Location, len(m))
+	for loc := range m {
+		ret = append(ret, loc)
+	}
+	return ret
+}
diff --git a/internal/lsp/references.go b/internal/lsp/references.go
@@ -69,5 +69,5 @@ func (s *Server) references(ctx context.Context, params *protocol.ReferenceParam
 		elapsed := time.Since(start)
 		log.Debug(fmt.Sprintf("References took %s. IdentifierInfo: %#v", elapsed, id.String()))
 	}
-	return locs, nil
+	return unifyLocs(locs), nil
 }