extern/nushell
1
0
Fork 1
mirror of https://github.com/nushell/nushell.git synced 2025-04-23 12:48:22 +02:00
Code Issues Packages Projects Releases Wiki Activity

perf: memmem::Finder

Browse Source
This commit is contained in:
blindfs 2025-04-09 14:22:26 +08:00
parent 3f9e6ab985
commit 457cb3652b
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
crates/nu-lsp/src/workspace.rs
View File

@ -8,7 +8,6 @@ use lsp_types::{
PrepareRenameResponse, ProgressToken, Range, ReferenceParams, RenameParams,
TextDocumentPositionParams, TextEdit, Uri, WorkspaceEdit, WorkspaceFolder,
};
use memchr::memmem;
use miette::{miette, IntoDiagnostic, Result};
use nu_glob::Uninterruptible;
use nu_protocol::{
@ -391,6 +390,8 @@ impl LanguageServer {
.collect();
let len = scripts.len();
let definition_span = Self::find_definition_span_by_id(&working_set, &id_tracker.id);
let bytes_to_search = id_tracker.name.to_owned();
let finder = memchr::memmem::Finder::new(&bytes_to_search);
for (i, fp) in scripts.iter().enumerate() {
#[cfg(test)]
@ -411,7 +412,7 @@ impl LanguageServer {
let file = if let Some(file) = docs.get_document(&uri) {
file
} else {
let file_content_byte = match fs::read(fp) {
let file_bytes = match fs::read(fp) {
Ok(it) => it,
Err(_) => {
// continue on fs error
@ -419,7 +420,7 @@ impl LanguageServer {
}
};
// skip if the file does not contain what we're looking for
if memmem::find(&file_content_byte, id_tracker.name.as_ref()).is_none() {
if finder.find(&file_bytes).is_none() {
// progress without any data
data_sender
.send(InternalMessage::OnGoing(token.clone(), percentage))
@ -429,7 +430,7 @@ impl LanguageServer {
&FullTextDocument::new(
"nu".to_string(),
0,
String::from_utf8_lossy(&file_content_byte).into(),
String::from_utf8_lossy(&file_bytes).into(),
)
};
let _ = Self::find_reference_in_file(&mut working_set, file, fp, &mut id_tracker)