feat: move bench() method and bench test proc macro to toools/benches

kclvm/tools/benches/benchmark.rs

@@ -1,6 +1,10 @@
 use criterion::{criterion_group, criterion_main, Criterion};
 use kclvm_query::override_file;
 use kclvm_tools::format::{format, FormatOptions};
+use std::{
+    fmt,
+    time::{Duration, Instant},
+};
 
 pub fn criterion_benchmark_override(c: &mut Criterion) {
     c.bench_function("override", |b| {
@@ -29,3 +33,77 @@ criterion_group!(
     criterion_benchmark_format
 );
 criterion_main!(benches);
+
+pub struct StopWatch {
+    time: Instant,
+}
+
+pub struct StopWatchSpan {
+    pub time: Duration,
+}
+
+impl StopWatch {
+    pub fn start() -> StopWatch {
+        let time = Instant::now();
+        StopWatch { time }
+    }
+
+    pub fn elapsed(&mut self) -> StopWatchSpan {
+        let time = self.time.elapsed();
+
+        StopWatchSpan { time }
+    }
+}
+
+impl fmt::Display for StopWatchSpan {
+    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+        write!(f, "{:.2?}", self.time)?;
+        Ok(())
+    }
+}
+
+/// Utility for writing benchmark tests.
+///
+/// If you need to benchmark the entire test, you can directly add the macro `#[bench_test]` like this:
+/// ```
+/// #[test]
+/// #[bench_test]
+/// fn benchmark_foo() {
+///     actual_work(analysis)
+/// }
+/// ```
+///
+/// If you need to skip some preparation stages and only test some parts of test, you can use the `bench()` method.
+/// A benchmark test looks like this:
+///
+/// ```
+/// #[test]
+/// fn benchmark_foo() {
+///     let data = bench_fixture::some_fixture();
+///     let analysis = some_setup();
+///
+///     {
+///         let _b = bench("foo");
+///         actual_work(analysis)
+///     };
+/// }
+/// ```
+///
+///
+pub fn bench(label: &'static str) -> impl Drop {
+    struct Bencher {
+        sw: StopWatch,
+        label: &'static str,
+    }
+
+    impl Drop for Bencher {
+        fn drop(&mut self) {
+            eprintln!("{}: {}", self.label, self.sw.elapsed());
+        }
+    }
+
+    Bencher {
+        sw: StopWatch::start(),
+        label,
+    }
+}

kclvm/tools/src/LSP/Cargo.toml

@@ -41,4 +41,4 @@ salsa = { version = "0.16.1", default-features = false }
 serde_json = { version = "1.0", default-features = false }
 parking_lot = { version = "0.12.0", default-features = false }
 rustc-hash = { version = "1.1.0", default-features = false }
-proc_macro_crate = {path = "./proc_macro_crate"}
+proc_macro_crate = {path = "../../benches/proc_macro_crate"}

kclvm/tools/src/LSP/src/lib.rs

@@ -16,4 +16,3 @@ mod goto_def;
 mod hover;
 mod quick_fix;
 mod request;
-mod test_utils;

kclvm/tools/src/LSP/src/main.rs

@@ -16,7 +16,6 @@ mod notification;
 mod quick_fix;
 mod request;
 mod state;
-mod test_utils;
 mod to_lsp;
 mod util;
 

kclvm/tools/src/LSP/src/tests.rs

@@ -1234,7 +1234,6 @@ fn file_path_from_url_test() {
 }
 
 #[test]
-#[bench_test]
 fn goto_import_external_file_test() {
     let path = PathBuf::from(".")
         .join("src")
@@ -1283,7 +1282,6 @@ fn goto_import_external_file_test() {
 }
 
 #[test]
-#[bench_test]
 fn format_signle_file_test() {
     const FILE_INPUT_SUFFIX: &str = ".input";
     const FILE_OUTPUT_SUFFIX: &str = ".golden";