From ed7ae78c6fd9e508f6e959c6a164cf8481f6b377 Mon Sep 17 00:00:00 2001 From: Aleksey Kladov Date: Fri, 17 Aug 2018 19:54:08 +0300 Subject: ServerWorld --- crates/server/src/main_loop/handlers.rs | 102 ++++++++++++-------------- crates/server/src/main_loop/mod.rs | 125 ++++++++++---------------------- 2 files changed, 86 insertions(+), 141 deletions(-) (limited to 'crates/server/src/main_loop') diff --git a/crates/server/src/main_loop/handlers.rs b/crates/server/src/main_loop/handlers.rs index f25c64c97..675f69bec 100644 --- a/crates/server/src/main_loop/handlers.rs +++ b/crates/server/src/main_loop/handlers.rs @@ -5,35 +5,33 @@ use languageserver_types::{ Command, TextDocumentIdentifier, WorkspaceEdit, SymbolInformation, Position, }; -use libanalysis::{World, Query}; +use libanalysis::{Query}; use libeditor::{self, CursorPosition}; use libsyntax2::TextUnit; use serde_json::{to_value, from_value}; use ::{ - PathMap, req::{self, Decoration}, Result, conv::{Conv, ConvWith, TryConvWith, MapConvWith, to_location}, + server_world::ServerWorld, }; pub fn handle_syntax_tree( - world: World, - path_map: PathMap, + world: ServerWorld, params: req::SyntaxTreeParams, ) -> Result { - let id = params.text_document.try_conv_with(&path_map)?; - let file = world.file_syntax(id)?; + let id = params.text_document.try_conv_with(&world)?; + let file = world.analysis().file_syntax(id)?; Ok(libeditor::syntax_tree(&file)) } pub fn handle_extend_selection( - world: World, - path_map: PathMap, + world: ServerWorld, params: req::ExtendSelectionParams, ) -> Result { - let file_id = params.text_document.try_conv_with(&path_map)?; - let file = world.file_syntax(file_id)?; - let line_index = world.file_line_index(file_id)?; + let file_id = params.text_document.try_conv_with(&world)?; + let file = world.analysis().file_syntax(file_id)?; + let line_index = world.analysis().file_line_index(file_id)?; let selections = params.selections.into_iter() .map_conv_with(&line_index) .map(|r| libeditor::extend_selection(&file, r).unwrap_or(r)) @@ -43,13 +41,12 @@ pub fn handle_extend_selection( } pub fn handle_find_matching_brace( - world: World, - path_map: PathMap, + world: ServerWorld, params: req::FindMatchingBraceParams, ) -> Result> { - let file_id = params.text_document.try_conv_with(&path_map)?; - let file = world.file_syntax(file_id)?; - let line_index = world.file_line_index(file_id)?; + let file_id = params.text_document.try_conv_with(&world)?; + let file = world.analysis().file_syntax(file_id)?; + let line_index = world.analysis().file_line_index(file_id)?; let res = params.offsets .into_iter() .map_conv_with(&line_index) @@ -62,13 +59,12 @@ pub fn handle_find_matching_brace( } pub fn handle_document_symbol( - world: World, - path_map: PathMap, + world: ServerWorld, params: req::DocumentSymbolParams, ) -> Result> { - let file_id = params.text_document.try_conv_with(&path_map)?; - let file = world.file_syntax(file_id)?; - let line_index = world.file_line_index(file_id)?; + let file_id = params.text_document.try_conv_with(&world)?; + let file = world.analysis().file_syntax(file_id)?; + let line_index = world.analysis().file_line_index(file_id)?; let mut parents: Vec<(DocumentSymbol, Option)> = Vec::new(); @@ -102,13 +98,12 @@ pub fn handle_document_symbol( } pub fn handle_code_action( - world: World, - path_map: PathMap, + world: ServerWorld, params: req::CodeActionParams, ) -> Result>> { - let file_id = params.text_document.try_conv_with(&path_map)?; - let file = world.file_syntax(file_id)?; - let line_index = world.file_line_index(file_id)?; + let file_id = params.text_document.try_conv_with(&world)?; + let file = world.analysis().file_syntax(file_id)?; + let line_index = world.analysis().file_line_index(file_id)?; let offset = params.range.conv_with(&line_index).start(); let mut ret = Vec::new(); @@ -127,8 +122,7 @@ pub fn handle_code_action( } pub fn handle_workspace_symbol( - world: World, - path_map: PathMap, + world: ServerWorld, params: req::WorkspaceSymbolParams, ) -> Result>> { let all_symbols = params.query.contains("#"); @@ -143,25 +137,25 @@ pub fn handle_workspace_symbol( q.limit(128); q }; - let mut res = exec_query(&world, &path_map, query)?; + let mut res = exec_query(&world, query)?; if res.is_empty() && !all_symbols { let mut query = Query::new(params.query); query.limit(128); - res = exec_query(&world, &path_map, query)?; + res = exec_query(&world, query)?; } return Ok(Some(res)); - fn exec_query(world: &World, path_map: &PathMap, query: Query) -> Result> { + fn exec_query(world: &ServerWorld, query: Query) -> Result> { let mut res = Vec::new(); - for (file_id, symbol) in world.world_symbols(query) { - let line_index = world.file_line_index(file_id)?; + for (file_id, symbol) in world.analysis().world_symbols(query) { + let line_index = world.analysis().file_line_index(file_id)?; let info = SymbolInformation { name: symbol.name.to_string(), kind: symbol.kind.conv(), location: to_location( file_id, symbol.node_range, - path_map, &line_index + world, &line_index )?, container_name: None, }; @@ -172,19 +166,18 @@ pub fn handle_workspace_symbol( } pub fn handle_goto_definition( - world: World, - path_map: PathMap, + world: ServerWorld, params: req::TextDocumentPositionParams, ) -> Result> { - let file_id = params.text_document.try_conv_with(&path_map)?; - let line_index = world.file_line_index(file_id)?; + let file_id = params.text_document.try_conv_with(&world)?; + let line_index = world.analysis().file_line_index(file_id)?; let offset = params.position.conv_with(&line_index); let mut res = Vec::new(); - for (file_id, symbol) in world.approximately_resolve_symbol(file_id, offset)? { - let line_index = world.file_line_index(file_id)?; + for (file_id, symbol) in world.analysis().approximately_resolve_symbol(file_id, offset)? { + let line_index = world.analysis().file_line_index(file_id)?; let location = to_location( file_id, symbol.node_range, - &path_map, &line_index, + &world, &line_index, )?; res.push(location) } @@ -192,8 +185,7 @@ pub fn handle_goto_definition( } pub fn handle_execute_command( - world: World, - path_map: PathMap, + world: ServerWorld, mut params: req::ExecuteCommandParams, ) -> Result<(req::ApplyWorkspaceEditParams, Option)> { if params.command.as_str() != "apply_code_action" { @@ -204,13 +196,13 @@ pub fn handle_execute_command( } let arg = params.arguments.pop().unwrap(); let arg: ActionRequest = from_value(arg)?; - let file_id = arg.text_document.try_conv_with(&path_map)?; - let file = world.file_syntax(file_id)?; + let file_id = arg.text_document.try_conv_with(&world)?; + let file = world.analysis().file_syntax(file_id)?; let action_result = match arg.id { ActionId::FlipComma => libeditor::flip_comma(&file, arg.offset).map(|f| f()), ActionId::AddDerive => libeditor::add_derive(&file, arg.offset).map(|f| f()), }.ok_or_else(|| format_err!("command not applicable"))?; - let line_index = world.file_line_index(file_id)?; + let line_index = world.analysis().file_line_index(file_id)?; let mut changes = HashMap::new(); changes.insert( arg.text_document.uri, @@ -265,13 +257,12 @@ impl ActionId { } pub fn publish_diagnostics( - world: World, - path_map: PathMap, + world: ServerWorld, uri: Url ) -> Result { - let file_id = uri.try_conv_with(&path_map)?; - let file = world.file_syntax(file_id)?; - let line_index = world.file_line_index(file_id)?; + let file_id = world.uri_to_file_id(&uri)?; + let file = world.analysis().file_syntax(file_id)?; + let line_index = world.analysis().file_line_index(file_id)?; let diagnostics = libeditor::diagnostics(&file) .into_iter() .map(|d| Diagnostic { @@ -286,13 +277,12 @@ pub fn publish_diagnostics( } pub fn publish_decorations( - world: World, - path_map: PathMap, + world: ServerWorld, uri: Url ) -> Result { - let file_id = uri.try_conv_with(&path_map)?; - let file = world.file_syntax(file_id)?; - let line_index = world.file_line_index(file_id)?; + let file_id = world.uri_to_file_id(&uri)?; + let file = world.analysis().file_syntax(file_id)?; + let line_index = world.analysis().file_line_index(file_id)?; let decorations = libeditor::highlight(&file) .into_iter() .map(|h| Decoration { diff --git a/crates/server/src/main_loop/mod.rs b/crates/server/src/main_loop/mod.rs index a8340df59..ad7c480dc 100644 --- a/crates/server/src/main_loop/mod.rs +++ b/crates/server/src/main_loop/mod.rs @@ -1,21 +1,20 @@ mod handlers; use std::{ - collections::{HashSet, HashMap}, + collections::{HashSet}, }; use threadpool::ThreadPool; use crossbeam_channel::{Sender, Receiver}; use languageserver_types::Url; -use libanalysis::{World, WorldState, FileId}; use serde_json::to_value; use { req, dispatch, - Task, Result, PathMap, + Task, Result, io::{Io, RawMsg, RawRequest, RawNotification}, - vfs::{FileEvent, FileEventKind}, - conv::TryConvWith, + vfs::FileEvent, + server_world::{ServerWorldState, ServerWorld}, main_loop::handlers::{ handle_syntax_tree, handle_extend_selection, @@ -32,17 +31,16 @@ use { pub(super) fn main_loop( io: &mut Io, - world: &mut WorldState, pool: &mut ThreadPool, task_sender: Sender, task_receiver: Receiver, fs_events_receiver: Receiver>, ) -> Result<()> { info!("server initialized, serving requests"); + let mut state = ServerWorldState::new(); + let mut next_request_id = 0; let mut pending_requests: HashSet = HashSet::new(); - let mut path_map = PathMap::new(); - let mut mem_map: HashMap> = HashMap::new(); let mut fs_events_receiver = Some(&fs_events_receiver); loop { enum Event { @@ -91,37 +89,17 @@ pub(super) fn main_loop( } Event::Fs(events) => { trace!("fs change, {} events", events.len()); - let changes = events.into_iter() - .map(|event| { - let text = match event.kind { - FileEventKind::Add(text) => Some(text), - FileEventKind::Remove => None, - }; - (event.path, text) - }) - .map(|(path, text)| { - (path_map.get_or_insert(path), text) - }) - .filter_map(|(id, text)| { - if mem_map.contains_key(&id) { - mem_map.insert(id, text); - None - } else { - Some((id, text)) - } - }); - - world.change_files(changes); + state.apply_fs_changes(events); } Event::Msg(msg) => { match msg { RawMsg::Request(req) => { - if !on_request(io, world, &path_map, pool, &task_sender, req)? { + if !on_request(io, &state, pool, &task_sender, req)? { return Ok(()); } } RawMsg::Notification(not) => { - on_notification(io, world, &mut path_map, pool, &task_sender, not, &mut mem_map)? + on_notification(io, &mut state, pool, &task_sender, not)? } RawMsg::Response(resp) => { if !pending_requests.remove(&resp.id) { @@ -136,45 +114,40 @@ pub(super) fn main_loop( fn on_request( io: &mut Io, - world: &WorldState, - path_map: &PathMap, + world: &ServerWorldState, pool: &ThreadPool, sender: &Sender, req: RawRequest, ) -> Result { let mut req = Some(req); handle_request_on_threadpool::( - &mut req, pool, path_map, world, sender, handle_syntax_tree, + &mut req, pool, world, sender, handle_syntax_tree, )?; handle_request_on_threadpool::( - &mut req, pool, path_map, world, sender, handle_extend_selection, + &mut req, pool, world, sender, handle_extend_selection, )?; handle_request_on_threadpool::( - &mut req, pool, path_map, world, sender, handle_find_matching_brace, + &mut req, pool, world, sender, handle_find_matching_brace, )?; handle_request_on_threadpool::( - &mut req, pool, path_map, world, sender, handle_document_symbol, + &mut req, pool, world, sender, handle_document_symbol, )?; handle_request_on_threadpool::( - &mut req, pool, path_map, world, sender, handle_code_action, + &mut req, pool, world, sender, handle_code_action, )?; handle_request_on_threadpool::( - &mut req, pool, path_map, world, sender, handle_workspace_symbol, + &mut req, pool, world, sender, handle_workspace_symbol, )?; handle_request_on_threadpool::( - &mut req, pool, path_map, world, sender, handle_goto_definition, + &mut req, pool, world, sender, handle_goto_definition, )?; dispatch::handle_request::(&mut req, |params, resp| { io.send(RawMsg::Response(resp.into_response(Ok(None))?)); - let world = world.snapshot({ - let pm = path_map.clone(); - move |id, path| pm.resolve(id, path) - }); - let path_map = path_map.clone(); + let world = world.snapshot(); let sender = sender.clone(); pool.execute(move || { - let (edit, cursor) = match handle_execute_command(world, path_map, params) { + let (edit, cursor) = match handle_execute_command(world, params) { Ok(res) => res, Err(e) => return sender.send(Task::Die(e)), }; @@ -221,60 +194,48 @@ fn on_request( fn on_notification( io: &mut Io, - world: &mut WorldState, - path_map: &mut PathMap, + state: &mut ServerWorldState, pool: &ThreadPool, sender: &Sender, not: RawNotification, - mem_map: &mut HashMap>, ) -> Result<()> { let mut not = Some(not); dispatch::handle_notification::(&mut not, |params| { let uri = params.text_document.uri; let path = uri.to_file_path() .map_err(|()| format_err!("invalid uri: {}", uri))?; - let file_id = path_map.get_or_insert(path); - mem_map.insert(file_id, None); - world.change_file(file_id, Some(params.text_document.text)); + state.add_mem_file(path, params.text_document.text); update_file_notifications_on_threadpool( pool, - world.snapshot({ - let pm = path_map.clone(); - move |id, path| pm.resolve(id, path) - }), - path_map.clone(), + state.snapshot(), sender.clone(), uri, ); Ok(()) })?; dispatch::handle_notification::(&mut not, |mut params| { - let file_id = params.text_document.try_conv_with(path_map)?; + let uri = params.text_document.uri; + let path = uri.to_file_path() + .map_err(|()| format_err!("invalid uri: {}", uri))?; let text = params.content_changes.pop() .ok_or_else(|| format_err!("empty changes"))? .text; - world.change_file(file_id, Some(text)); + state.change_mem_file(path.as_path(), text)?; update_file_notifications_on_threadpool( pool, - world.snapshot({ - let pm = path_map.clone(); - move |id, path| pm.resolve(id, path) - }), - path_map.clone(), + state.snapshot(), sender.clone(), - params.text_document.uri, + uri, ); Ok(()) })?; dispatch::handle_notification::(&mut not, |params| { - let file_id = params.text_document.try_conv_with(path_map)?; - let text = match mem_map.remove(&file_id) { - Some(text) => text, - None => bail!("unmatched close notification"), - }; - world.change_file(file_id, text); + let uri = params.text_document.uri; + let path = uri.to_file_path() + .map_err(|()| format_err!("invalid uri: {}", uri))?; + state.remove_mem_file(path.as_path())?; let not = req::PublishDiagnosticsParams { - uri: params.text_document.uri, + uri, diagnostics: Vec::new(), }; let not = dispatch::send_notification::(not); @@ -291,21 +252,16 @@ fn on_notification( fn handle_request_on_threadpool( req: &mut Option, pool: &ThreadPool, - path_map: &PathMap, - world: &WorldState, + world: &ServerWorldState, sender: &Sender, - f: fn(World, PathMap, R::Params) -> Result, + f: fn(ServerWorld, R::Params) -> Result, ) -> Result<()> { dispatch::handle_request::(req, |params, resp| { - let world = world.snapshot({ - let pm = path_map.clone(); - move |id, path| pm.resolve(id, path) - }); - let path_map = path_map.clone(); + let world = world.snapshot(); let sender = sender.clone(); pool.execute(move || { - let res = f(world, path_map, params); + let res = f(world, params); let task = match resp.into_response(res) { Ok(resp) => Task::Respond(resp), Err(e) => Task::Die(e), @@ -318,13 +274,12 @@ fn handle_request_on_threadpool( fn update_file_notifications_on_threadpool( pool: &ThreadPool, - world: World, - path_map: PathMap, + world: ServerWorld, sender: Sender, uri: Url, ) { pool.execute(move || { - match publish_diagnostics(world.clone(), path_map.clone(), uri.clone()) { + match publish_diagnostics(world.clone(), uri.clone()) { Err(e) => { error!("failed to compute diagnostics: {:?}", e) } @@ -333,7 +288,7 @@ fn update_file_notifications_on_threadpool( sender.send(Task::Notify(not)); } } - match publish_decorations(world, path_map.clone(), uri) { + match publish_decorations(world, uri) { Err(e) => { error!("failed to compute decorations: {:?}", e) } -- cgit v1.2.3