From f6adb85b681c23b64cc33197eb67d5d7fcf920f0 Mon Sep 17 00:00:00 2001
From: Aleksey Kladov <aleksey.kladov@gmail.com>
Date: Fri, 25 Jan 2019 19:11:58 +0300
Subject: add gc request

---
 crates/ra_ide_api/src/imp.rs             |  5 +++--
 crates/ra_ide_api/src/lib.rs             |  4 ++++
 crates/ra_lsp_server/src/main_loop.rs    | 27 ++++++++++++++++++---------
 crates/ra_lsp_server/src/req.rs          |  8 ++++++++
 crates/ra_lsp_server/src/server_world.rs |  4 ++++
 5 files changed, 37 insertions(+), 11 deletions(-)

(limited to 'crates')

diff --git a/crates/ra_ide_api/src/imp.rs b/crates/ra_ide_api/src/imp.rs
index 8ecb8b17c..961f7b230 100644
--- a/crates/ra_ide_api/src/imp.rs
+++ b/crates/ra_ide_api/src/imp.rs
@@ -72,13 +72,14 @@ impl db::RootDatabase {
         self.set_source_root(root_id, Arc::new(source_root));
     }
 
-    #[allow(unused)]
     /// Ideally, we should call this function from time to time to collect heavy
     /// syntax trees. However, if we actually do that, everything is recomputed
     /// for some reason. Needs investigation.
-    fn gc_syntax_trees(&mut self) {
+    pub(crate) fn collect_garbage(&mut self) {
         self.query(ra_db::SourceFileQuery)
             .sweep(salsa::SweepStrategy::default().discard_values());
+        self.query(hir::db::HirSourceFileQuery)
+            .sweep(salsa::SweepStrategy::default().discard_values());
         self.query(hir::db::FileItemsQuery)
             .sweep(salsa::SweepStrategy::default().discard_values());
         self.query(hir::db::FileItemQuery)
diff --git a/crates/ra_ide_api/src/lib.rs b/crates/ra_ide_api/src/lib.rs
index 3502bfd2e..ffd026b04 100644
--- a/crates/ra_ide_api/src/lib.rs
+++ b/crates/ra_ide_api/src/lib.rs
@@ -285,6 +285,10 @@ impl AnalysisHost {
     pub fn apply_change(&mut self, change: AnalysisChange) {
         self.db.apply_change(change)
     }
+
+    pub fn collect_garbage(&mut self) {
+        self.db.collect_garbage();
+    }
 }
 
 /// Analysis is a snapshot of a world state at a moment in time. It is the main
diff --git a/crates/ra_lsp_server/src/main_loop.rs b/crates/ra_lsp_server/src/main_loop.rs
index f51576521..ddd20a41f 100644
--- a/crates/ra_lsp_server/src/main_loop.rs
+++ b/crates/ra_lsp_server/src/main_loop.rs
@@ -205,17 +205,26 @@ fn main_loop_inner(
                         Some(req) => req,
                         None => return Ok(()),
                     };
-                    match on_request(state, pending_requests, pool, &task_sender, req)? {
-                        None => (),
-                        Some(req) => {
-                            log::error!("unknown request: {:?}", req);
-                            let resp = RawResponse::err(
-                                req.id,
-                                ErrorCode::MethodNotFound as i32,
-                                "unknown request".to_string(),
-                            );
+                    match req.cast::<req::CollectGarbage>() {
+                        Ok((id, ())) => {
+                            state.collect_garbadge();
+                            let resp = RawResponse::ok::<req::CollectGarbage>(id, &());
                             msg_sender.send(RawMessage::Response(resp)).unwrap()
                         }
+                        Err(req) => {
+                            match on_request(state, pending_requests, pool, &task_sender, req)? {
+                                None => (),
+                                Some(req) => {
+                                    log::error!("unknown request: {:?}", req);
+                                    let resp = RawResponse::err(
+                                        req.id,
+                                        ErrorCode::MethodNotFound as i32,
+                                        "unknown request".to_string(),
+                                    );
+                                    msg_sender.send(RawMessage::Response(resp)).unwrap()
+                                }
+                            }
+                        }
                     }
                 }
                 RawMessage::Notification(not) => {
diff --git a/crates/ra_lsp_server/src/req.rs b/crates/ra_lsp_server/src/req.rs
index ec6b6d905..5968e592b 100644
--- a/crates/ra_lsp_server/src/req.rs
+++ b/crates/ra_lsp_server/src/req.rs
@@ -19,6 +19,14 @@ impl Request for AnalyzerStatus {
     const METHOD: &'static str = "ra/analyzerStatus";
 }
 
+pub enum CollectGarbage {}
+
+impl Request for CollectGarbage {
+    type Params = ();
+    type Result = ();
+    const METHOD: &'static str = "ra/collectGarbage";
+}
+
 pub enum SyntaxTree {}
 
 impl Request for SyntaxTree {
diff --git a/crates/ra_lsp_server/src/server_world.rs b/crates/ra_lsp_server/src/server_world.rs
index 5cb97b29b..bf04f1125 100644
--- a/crates/ra_lsp_server/src/server_world.rs
+++ b/crates/ra_lsp_server/src/server_world.rs
@@ -231,6 +231,10 @@ impl ServerWorldState {
             vfs: Arc::clone(&self.vfs),
         }
     }
+
+    pub fn collect_garbadge(&mut self) {
+        self.analysis_host.collect_garbage()
+    }
 }
 
 impl ServerWorld {
-- 
cgit v1.2.3