extern/nushell
1
0
Fork 1
mirror of https://github.com/nushell/nushell.git synced 2025-04-23 20:58:19 +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, PrepareRenameResponse, ProgressToken, Range, ReferenceParams, RenameParams,
TextDocumentPositionParams, TextEdit, Uri, WorkspaceEdit, WorkspaceFolder, TextDocumentPositionParams, TextEdit, Uri, WorkspaceEdit, WorkspaceFolder,
}; };
use memchr::memmem;
use miette::{miette, IntoDiagnostic, Result}; use miette::{miette, IntoDiagnostic, Result};
use nu_glob::Uninterruptible; use nu_glob::Uninterruptible;
use nu_protocol::{ use nu_protocol::{
@ -391,6 +390,8 @@ impl LanguageServer {
.collect(); .collect();
let len = scripts.len(); let len = scripts.len();
let definition_span = Self::find_definition_span_by_id(&working_set, &id_tracker.id); 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() { for (i, fp) in scripts.iter().enumerate() {
#[cfg(test)] #[cfg(test)]
@ -411,7 +412,7 @@ impl LanguageServer {
let file = if let Some(file) = docs.get_document(&uri) { let file = if let Some(file) = docs.get_document(&uri) {
file file
} else { } else {
let file_content_byte = match fs::read(fp) { let file_bytes = match fs::read(fp) {
Ok(it) => it, Ok(it) => it,
Err(_) => { Err(_) => {
// continue on fs error // continue on fs error
@ -419,7 +420,7 @@ impl LanguageServer {
} }
}; };
// skip if the file does not contain what we're looking for // 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 // progress without any data
data_sender data_sender
.send(InternalMessage::OnGoing(token.clone(), percentage)) .send(InternalMessage::OnGoing(token.clone(), percentage))
@ -429,7 +430,7 @@ impl LanguageServer {
&FullTextDocument::new( &FullTextDocument::new(
"nu".to_string(), "nu".to_string(),
0, 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) let _ = Self::find_reference_in_file(&mut working_set, file, fp, &mut id_tracker)