From dbbb0beb3ec9f11a635f43e60f3b3a42ba61338a Mon Sep 17 00:00:00 2001 From: Kirill Bulatov Date: Thu, 25 Jul 2019 20:22:41 +0300 Subject: Make Analysis api cancellable --- crates/ra_lsp_server/src/conv.rs | 14 +++--- crates/ra_lsp_server/src/main_loop/handlers.rs | 64 ++++++++++++++------------ crates/ra_lsp_server/src/world.rs | 7 ++- 3 files changed, 48 insertions(+), 37 deletions(-) (limited to 'crates/ra_lsp_server/src') diff --git a/crates/ra_lsp_server/src/conv.rs b/crates/ra_lsp_server/src/conv.rs index 82c7e757f..6b3be444f 100644 --- a/crates/ra_lsp_server/src/conv.rs +++ b/crates/ra_lsp_server/src/conv.rs @@ -272,7 +272,7 @@ impl<'a> TryConvWith for &'a TextDocumentPositionParams { type Output = FilePosition; fn try_conv_with(self, world: &WorldSnapshot) -> Result { let file_id = self.text_document.try_conv_with(world)?; - let line_index = world.analysis().file_line_index(file_id); + let line_index = world.analysis().file_line_index(file_id)?; let offset = self.position.conv_with(&line_index); Ok(FilePosition { file_id, offset }) } @@ -283,7 +283,7 @@ impl<'a> TryConvWith for (&'a TextDocumentIdentifier, Range) { type Output = FileRange; fn try_conv_with(self, world: &WorldSnapshot) -> Result { let file_id = self.0.try_conv_with(world)?; - let line_index = world.analysis().file_line_index(file_id); + let line_index = world.analysis().file_line_index(file_id)?; let range = self.1.conv_with(&line_index); Ok(FileRange { file_id, range }) } @@ -308,7 +308,7 @@ impl TryConvWith for SourceChange { let cursor_position = match self.cursor_position { None => None, Some(pos) => { - let line_index = world.analysis().file_line_index(pos.file_id); + let line_index = world.analysis().file_line_index(pos.file_id)?; let edit = self .source_file_edits .iter() @@ -349,7 +349,7 @@ impl TryConvWith for SourceFileEdit { uri: self.file_id.try_conv_with(world)?, version: None, }; - let line_index = world.analysis().file_line_index(self.file_id); + let line_index = world.analysis().file_line_index(self.file_id)?; let edits = self.edit.as_atoms().iter().map_conv_with(&line_index).collect(); Ok(TextDocumentEdit { text_document, edits }) } @@ -378,7 +378,7 @@ impl TryConvWith for &NavigationTarget { type Ctx = WorldSnapshot; type Output = Location; fn try_conv_with(self, world: &WorldSnapshot) -> Result { - let line_index = world.analysis().file_line_index(self.file_id()); + let line_index = world.analysis().file_line_index(self.file_id())?; let range = self.range(); to_location(self.file_id(), range, &world, &line_index) } @@ -391,8 +391,8 @@ impl TryConvWith for (FileId, RangeInfo) { let (src_file_id, target) = self; let target_uri = target.info.file_id().try_conv_with(world)?; - let src_line_index = world.analysis().file_line_index(src_file_id); - let tgt_line_index = world.analysis().file_line_index(target.info.file_id()); + let src_line_index = world.analysis().file_line_index(src_file_id)?; + let tgt_line_index = world.analysis().file_line_index(target.info.file_id())?; let target_range = target.info.full_range().conv_with(&tgt_line_index); diff --git a/crates/ra_lsp_server/src/main_loop/handlers.rs b/crates/ra_lsp_server/src/main_loop/handlers.rs index 5bf950a53..14619ede2 100644 --- a/crates/ra_lsp_server/src/main_loop/handlers.rs +++ b/crates/ra_lsp_server/src/main_loop/handlers.rs @@ -39,9 +39,9 @@ pub fn handle_analyzer_status(world: WorldSnapshot, _: ()) -> Result { pub fn handle_syntax_tree(world: WorldSnapshot, params: req::SyntaxTreeParams) -> Result { let id = params.text_document.try_conv_with(&world)?; - let line_index = world.analysis().file_line_index(id); + let line_index = world.analysis().file_line_index(id)?; let text_range = params.range.map(|p| p.conv_with(&line_index)); - let res = world.analysis().syntax_tree(id, text_range); + let res = world.analysis().syntax_tree(id, text_range)?; Ok(res) } @@ -55,7 +55,7 @@ pub fn handle_extend_selection( use the new selection range API in LSP", ); let file_id = params.text_document.try_conv_with(&world)?; - let line_index = world.analysis().file_line_index(file_id); + let line_index = world.analysis().file_line_index(file_id)?; let selections = params .selections .into_iter() @@ -72,7 +72,7 @@ pub fn handle_selection_range( ) -> Result> { let _p = profile("handle_selection_range"); let file_id = params.text_document.try_conv_with(&world)?; - let line_index = world.analysis().file_line_index(file_id); + let line_index = world.analysis().file_line_index(file_id)?; params .positions .into_iter() @@ -113,13 +113,19 @@ pub fn handle_find_matching_brace( ) -> Result> { let _p = profile("handle_find_matching_brace"); let file_id = params.text_document.try_conv_with(&world)?; - let line_index = world.analysis().file_line_index(file_id); + let line_index = world.analysis().file_line_index(file_id)?; let res = params .offsets .into_iter() .map_conv_with(&line_index) .map(|offset| { - world.analysis().matching_brace(FilePosition { file_id, offset }).unwrap_or(offset) + if let Ok(Some(matching_brace_offset)) = + world.analysis().matching_brace(FilePosition { file_id, offset }) + { + matching_brace_offset + } else { + offset + } }) .map_conv_with(&line_index) .collect(); @@ -132,7 +138,7 @@ pub fn handle_join_lines( ) -> Result { let _p = profile("handle_join_lines"); let frange = (¶ms.text_document, params.range).try_conv_with(&world)?; - world.analysis().join_lines(frange).try_conv_with(&world) + world.analysis().join_lines(frange)?.try_conv_with(&world) } pub fn handle_on_enter( @@ -141,7 +147,7 @@ pub fn handle_on_enter( ) -> Result> { let _p = profile("handle_on_enter"); let position = params.try_conv_with(&world)?; - match world.analysis().on_enter(position) { + match world.analysis().on_enter(position)? { None => Ok(None), Some(edit) => Ok(Some(edit.try_conv_with(&world)?)), } @@ -153,7 +159,7 @@ pub fn handle_on_type_formatting( ) -> Result>> { let _p = profile("handle_on_type_formatting"); let mut position = params.text_document_position.try_conv_with(&world)?; - let line_index = world.analysis().file_line_index(position.file_id); + let line_index = world.analysis().file_line_index(position.file_id)?; // in `ra_ide_api`, the `on_type` invariant is that // `text.char_at(position) == typed_char`. @@ -163,7 +169,7 @@ pub fn handle_on_type_formatting( "=" => world.analysis().on_eq_typed(position), "." => world.analysis().on_dot_typed(position), _ => return Ok(None), - }; + }?; let mut edit = match edit { Some(it) => it, None => return Ok(None), @@ -181,11 +187,11 @@ pub fn handle_document_symbol( params: req::DocumentSymbolParams, ) -> Result> { let file_id = params.text_document.try_conv_with(&world)?; - let line_index = world.analysis().file_line_index(file_id); + let line_index = world.analysis().file_line_index(file_id)?; let mut parents: Vec<(DocumentSymbol, Option)> = Vec::new(); - for symbol in world.analysis().file_structure(file_id) { + for symbol in world.analysis().file_structure(file_id)? { let doc_symbol = DocumentSymbol { name: symbol.label, detail: symbol.detail, @@ -309,7 +315,7 @@ pub fn handle_runnables( params: req::RunnablesParams, ) -> Result> { let file_id = params.text_document.try_conv_with(&world)?; - let line_index = world.analysis().file_line_index(file_id); + let line_index = world.analysis().file_line_index(file_id)?; let offset = params.position.map(|it| it.conv_with(&line_index)); let mut res = Vec::new(); let workspace_root = world.workspace_root_for(file_id); @@ -383,7 +389,7 @@ pub fn handle_completion( let mut res = false; if let Some(ctx) = params.context { if ctx.trigger_character.unwrap_or_default() == ":" { - let source_file = world.analysis().parse(position.file_id); + let source_file = world.analysis().parse(position.file_id)?; let syntax = source_file.syntax(); let text = syntax.text(); if let Some(next_char) = text.char_at(position.offset) { @@ -405,7 +411,7 @@ pub fn handle_completion( None => return Ok(None), Some(items) => items, }; - let line_index = world.analysis().file_line_index(position.file_id); + let line_index = world.analysis().file_line_index(position.file_id)?; let items: Vec = items.into_iter().map(|item| item.conv_with(&line_index)).collect(); @@ -417,12 +423,12 @@ pub fn handle_folding_range( params: FoldingRangeParams, ) -> Result>> { let file_id = params.text_document.try_conv_with(&world)?; - let line_index = world.analysis().file_line_index(file_id); + let line_index = world.analysis().file_line_index(file_id)?; let res = Some( world .analysis() - .folding_ranges(file_id) + .folding_ranges(file_id)? .into_iter() .map(|fold| { let kind = match fold.kind { @@ -474,7 +480,7 @@ pub fn handle_hover( None => return Ok(None), Some(info) => info, }; - let line_index = world.analysis.file_line_index(position.file_id); + let line_index = world.analysis.file_line_index(position.file_id)?; let range = info.range.conv_with(&line_index); let res = Hover { contents: HoverContents::Markup(MarkupContent { @@ -503,7 +509,7 @@ pub fn handle_prepare_rename( // Refs should always have a declaration let r = refs.declaration(); let file_id = params.text_document.try_conv_with(&world)?; - let line_index = world.analysis().file_line_index(file_id); + let line_index = world.analysis().file_line_index(file_id)?; let loc = to_location(r.file_id(), r.range(), &world, &line_index)?; Ok(Some(PrepareRenameResponse::Range(loc.range))) @@ -536,7 +542,7 @@ pub fn handle_references( params: req::ReferenceParams, ) -> Result>> { let position = params.text_document_position.try_conv_with(&world)?; - let line_index = world.analysis().file_line_index(position.file_id); + let line_index = world.analysis().file_line_index(position.file_id)?; let refs = match world.analysis().find_all_refs(position)? { None => return Ok(None), @@ -563,9 +569,9 @@ pub fn handle_formatting( params: DocumentFormattingParams, ) -> Result>> { let file_id = params.text_document.try_conv_with(&world)?; - let file = world.analysis().file_text(file_id); + let file = world.analysis().file_text(file_id)?; - let file_line_index = world.analysis().file_line_index(file_id); + let file_line_index = world.analysis().file_line_index(file_id)?; let end_position = TextUnit::of_str(&file).conv_with(&file_line_index); use std::process; @@ -623,7 +629,7 @@ pub fn handle_code_action( ) -> Result> { let _p = profile("handle_code_action"); let file_id = params.text_document.try_conv_with(&world)?; - let line_index = world.analysis().file_line_index(file_id); + let line_index = world.analysis().file_line_index(file_id)?; let range = params.range.conv_with(&line_index); let assists = world.analysis().assists(FileRange { file_id, range })?.into_iter(); @@ -685,7 +691,7 @@ pub fn handle_code_lens( params: req::CodeLensParams, ) -> Result>> { let file_id = params.text_document.try_conv_with(&world)?; - let line_index = world.analysis().file_line_index(file_id); + let line_index = world.analysis().file_line_index(file_id)?; let mut lenses: Vec = Default::default(); let workspace_root = world.workspace_root_for(file_id); @@ -730,7 +736,7 @@ pub fn handle_code_lens( lenses.extend( world .analysis() - .file_structure(file_id) + .file_structure(file_id)? .into_iter() .filter(|it| match it.kind { SyntaxKind::TRAIT_DEF | SyntaxKind::STRUCT_DEF | SyntaxKind::ENUM_DEF => true, @@ -807,7 +813,7 @@ pub fn handle_document_highlight( params: req::TextDocumentPositionParams, ) -> Result>> { let file_id = params.text_document.try_conv_with(&world)?; - let line_index = world.analysis().file_line_index(file_id); + let line_index = world.analysis().file_line_index(file_id)?; let refs = match world.analysis().find_all_refs(params.try_conv_with(&world)?)? { None => return Ok(None), @@ -826,7 +832,7 @@ pub fn publish_diagnostics( file_id: FileId, ) -> Result { let uri = world.file_id_to_uri(file_id)?; - let line_index = world.analysis().file_line_index(file_id); + let line_index = world.analysis().file_line_index(file_id)?; let diagnostics = world .analysis() .diagnostics(file_id)? @@ -852,7 +858,7 @@ pub fn publish_decorations( } fn highlight(world: &WorldSnapshot, file_id: FileId) -> Result> { - let line_index = world.analysis().file_line_index(file_id); + let line_index = world.analysis().file_line_index(file_id)?; let res = world .analysis() .highlight(file_id)? @@ -881,7 +887,7 @@ pub fn handle_inlay_hints( ) -> Result> { let file_id = params.text_document.try_conv_with(&world)?; let analysis = world.analysis(); - let line_index = analysis.file_line_index(file_id); + let line_index = analysis.file_line_index(file_id)?; Ok(analysis .inlay_hints(file_id)? .into_iter() diff --git a/crates/ra_lsp_server/src/world.rs b/crates/ra_lsp_server/src/world.rs index 9fd654305..1d7755910 100644 --- a/crates/ra_lsp_server/src/world.rs +++ b/crates/ra_lsp_server/src/world.rs @@ -215,7 +215,12 @@ impl WorldSnapshot { } } res.push_str("\nanalysis:\n"); - res.push_str(&self.analysis.status()); + res.push_str( + &self + .analysis + .status() + .unwrap_or_else(|_| "Analysis retrieval was cancelled".to_owned()), + ); res } -- cgit v1.2.3