Rename folder

crates/rust-analyzer/src/main_loop/pending_requests.rs

@@ -0,0 +1,75 @@
+//! Data structures that keep track of inflight requests.
+
+use std::time::{Duration, Instant};
+
+use lsp_server::RequestId;
+use rustc_hash::FxHashMap;
+
+#[derive(Debug)]
+pub struct CompletedRequest {
+    pub id: RequestId,
+    pub method: String,
+    pub duration: Duration,
+}
+
+#[derive(Debug)]
+pub(crate) struct PendingRequest {
+    pub(crate) id: RequestId,
+    pub(crate) method: String,
+    pub(crate) received: Instant,
+}
+
+impl From<PendingRequest> for CompletedRequest {
+    fn from(pending: PendingRequest) -> CompletedRequest {
+        CompletedRequest {
+            id: pending.id,
+            method: pending.method,
+            duration: pending.received.elapsed(),
+        }
+    }
+}
+
+#[derive(Debug, Default)]
+pub(crate) struct PendingRequests {
+    map: FxHashMap<RequestId, PendingRequest>,
+}
+
+impl PendingRequests {
+    pub(crate) fn start(&mut self, request: PendingRequest) {
+        let id = request.id.clone();
+        let prev = self.map.insert(id.clone(), request);
+        assert!(prev.is_none(), "duplicate request with id {}", id);
+    }
+    pub(crate) fn cancel(&mut self, id: &RequestId) -> bool {
+        self.map.remove(id).is_some()
+    }
+    pub(crate) fn finish(&mut self, id: &RequestId) -> Option<CompletedRequest> {
+        self.map.remove(id).map(CompletedRequest::from)
+    }
+}
+
+const N_COMPLETED_REQUESTS: usize = 10;
+
+#[derive(Debug, Default)]
+pub struct LatestRequests {
+    // hand-rolling VecDeque here to print things in a nicer way
+    buf: [Option<CompletedRequest>; N_COMPLETED_REQUESTS],
+    idx: usize,
+}
+
+impl LatestRequests {
+    pub(crate) fn record(&mut self, request: CompletedRequest) {
+        // special case: don't track status request itself
+        if request.method == "rust-analyzer/analyzerStatus" {
+            return;
+        }
+        let idx = self.idx;
+        self.buf[idx] = Some(request);
+        self.idx = (idx + 1) % N_COMPLETED_REQUESTS;
+    }
+
+    pub(crate) fn iter(&self) -> impl Iterator<Item = (bool, &CompletedRequest)> {
+        let idx = self.idx;
+        self.buf.iter().enumerate().filter_map(move |(i, req)| Some((i == idx, req.as_ref()?)))
+    }
+}

crates/rust-analyzer/src/main_loop/subscriptions.rs

@@ -0,0 +1,22 @@
+//! Keeps track of file subscriptions -- the set of currently opened files for
+//! which we want to publish diagnostics, syntax highlighting, etc.
+
+use ra_ide::FileId;
+use rustc_hash::FxHashSet;
+
+#[derive(Default, Debug)]
+pub(crate) struct Subscriptions {
+    subs: FxHashSet<FileId>,
+}
+
+impl Subscriptions {
+    pub(crate) fn add_sub(&mut self, file_id: FileId) {
+        self.subs.insert(file_id);
+    }
+    pub(crate) fn remove_sub(&mut self, file_id: FileId) {
+        self.subs.remove(&file_id);
+    }
+    pub(crate) fn subscriptions(&self) -> Vec<FileId> {
+        self.subs.iter().cloned().collect()
+    }
+}