crates/rust-analyzer/src/request_metrics.rs

   1 //! Records stats about requests
   2 use std::time::Duration;
   3
   4 use lsp_server::RequestId;
   5
   6 #[derive(Debug)]
   7 pub(crate) struct RequestMetrics {
   8     pub(crate) id: RequestId,
   9     pub(crate) method: String,
  10     pub(crate) duration: Duration,
  11 }
  12
  13 const N_COMPLETED_REQUESTS: usize = 10;
  14
  15 #[derive(Debug, Default)]
  16 pub(crate) struct LatestRequests {
  17     // hand-rolling VecDeque here to print things in a nicer way
  18     buf: [Option<RequestMetrics>; N_COMPLETED_REQUESTS],
  19     idx: usize,
  20 }
  21
  22 impl LatestRequests {
  23     pub(crate) fn record(&mut self, request: RequestMetrics) {
  24         // special case: don't track status request itself
  25         if request.method == "rust-analyzer/analyzerStatus" {
  26             return;
  27         }
  28         let idx = self.idx;
  29         self.buf[idx] = Some(request);
  30         self.idx = (idx + 1) % N_COMPLETED_REQUESTS;
  31     }
  32
  33     pub(crate) fn iter(&self) -> impl Iterator<Item = (bool, &RequestMetrics)> {
  34         let idx = self.idx;
  35         self.buf.iter().enumerate().filter_map(move |(i, req)| Some((i == idx, req.as_ref()?)))
  36     }
  37 }