add some profile info

Author: CppCXY

crates/code_analysis/src/compilation/analyzer/decl/mod.rs

@@ -2,7 +2,10 @@ mod docs;
 mod exprs;
 mod stats;
 
-use crate::db_index::{DbIndex, LuaScopeKind};
+use crate::{
+    db_index::{DbIndex, LuaScopeKind},
+    profile::Profile,
+};
 
 use super::AnalyzeContext;
 use emmylua_parser::{LuaAst, LuaAstNode, LuaChunk, LuaSyntaxKind};
@@ -14,6 +17,7 @@ use crate::{
 };
 
 pub(crate) fn analyze(db: &mut DbIndex, context: &mut AnalyzeContext) {
+    let _p = Profile::cond_new("decl analyze", context.tree_list.len() > 1);
     for in_filed_tree in context.tree_list.iter() {
         db.get_reference_index_mut()
             .create_local_reference(in_filed_tree.file_id);

crates/code_analysis/src/compilation/analyzer/doc/mod.rs

@@ -9,14 +9,14 @@ mod type_ref_tags;
 
 use super::AnalyzeContext;
 use crate::{
-    db_index::{DbIndex, LuaTypeDeclId},
-    FileId,
+    db_index::{DbIndex, LuaTypeDeclId}, profile::Profile, FileId
 };
 use emmylua_parser::{LuaAstNode, LuaComment, LuaDocDescriptionOwner, LuaSyntaxNode};
 use file_generic_index::FileGenericIndex;
 use tags::get_owner_id;
 
 pub(crate) fn analyze(db: &mut DbIndex, context: &mut AnalyzeContext) {
+    let _p = Profile::cond_new("doc analyze", context.tree_list.len() > 1);
     let tree_list = context.tree_list.clone();
     for in_filed_tree in tree_list.iter() {
         let root = &in_filed_tree.value;

crates/code_analysis/src/compilation/analyzer/flow/mod.rs

@@ -1,11 +1,12 @@
 mod reference_flow;
 
-use crate::{db_index::DbIndex, FileId};
+use crate::{db_index::DbIndex, profile::Profile, FileId};
 use emmylua_parser::LuaChunk;
 
 use super::AnalyzeContext;
 
 pub(crate) fn analyze(db: &mut DbIndex, context: &mut AnalyzeContext) {
+    let _p = Profile::cond_new("flow analyze", context.tree_list.len() > 1);
     let tree_list = context.tree_list.clone();
     // build decl and ref flow chain
     for in_filed_tree in &tree_list {

crates/code_analysis/src/compilation/analyzer/lua/mod.rs

@@ -13,14 +13,13 @@ use stats::{
 };
 
 use crate::{
-    db_index::{DbIndex, LuaType},
-    semantic::{infer_expr, LuaInferConfig},
-    FileId,
+    db_index::{DbIndex, LuaType}, profile::Profile, semantic::{infer_expr, LuaInferConfig}, FileId
 };
 
 use super::{unresolve::UnResolve, AnalyzeContext};
 
 pub(crate) fn analyze(db: &mut DbIndex, context: &mut AnalyzeContext) {
+    let _p = Profile::cond_new("lua analyze", context.tree_list.len() > 1);
     let tree_list = context.tree_list.clone();
     for in_filed_tree in &tree_list {
         let root = &in_filed_tree.value;

crates/code_analysis/src/compilation/analyzer/mod.rs

@@ -6,7 +6,7 @@ mod unresolve;
 
 use std::sync::Arc;
 
-use crate::{db_index::DbIndex, Emmyrc, InFiled};
+use crate::{db_index::DbIndex, profile::Profile, Emmyrc, InFiled};
 use emmylua_parser::LuaChunk;
 use unresolve::UnResolve;
 
@@ -21,6 +21,7 @@ pub fn analyze(db: &mut DbIndex, context: AnalyzeContext) {
 }
 
 fn module_analyze(db: &mut DbIndex, context: &mut AnalyzeContext) {
+    let _p = Profile::cond_new("module_analyze", context.tree_list.len() > 1);
     for in_filed_tree in &context.tree_list {
         let file_id = in_filed_tree.file_id;
         if let Some(path) = db.get_vfs().get_file_path(&file_id).cloned() {

crates/code_analysis/src/compilation/analyzer/unresolve/mod.rs

@@ -4,8 +4,7 @@ mod resolve;
 mod resolve_closure_param;
 
 use crate::{
-    db_index::{DbIndex, LuaDeclId, LuaMemberId, LuaSignatureId},
-    FileId,
+    db_index::{DbIndex, LuaDeclId, LuaMemberId, LuaSignatureId}, profile::Profile, FileId
 };
 use emmylua_parser::{LuaCallExpr, LuaExpr};
 use infer_manager::InferManager;
@@ -19,6 +18,7 @@ use resolve_closure_param::try_resolve_closure_params;
 use super::{lua::LuaReturnPoint, AnalyzeContext};
 
 pub fn analyze(db: &mut DbIndex, context: &mut AnalyzeContext) {
+    let _p = Profile::cond_new("resolve analyze", context.tree_list.len() > 1);
     let mut unresolves = std::mem::take(&mut context.unresolves);
     let mut infer_manager = InferManager::new(context.config.clone());
     while try_resolve(db, &mut infer_manager, &mut unresolves) {

crates/code_analysis/src/lib.rs

@@ -4,6 +4,7 @@ mod db_index;
 mod diagnostic;
 mod semantic;
 mod vfs;
+mod profile;
 
 pub use compilation::*;
 pub use config::*;
@@ -15,6 +16,7 @@ pub use semantic::*;
 use std::{collections::HashSet, env, path::PathBuf, sync::Arc};
 use tokio_util::sync::CancellationToken;
 pub use vfs::*;
+pub use profile::Profile;
 
 #[macro_use]
 extern crate rust_i18n;

crates/code_analysis/src/profile/mod.rs

@@ -0,0 +1,32 @@
+use std::time::Instant;
+use log::info;
+
+pub struct Profile<'a> {
+    name: &'a str,
+    start: Instant,
+}
+
+#[allow(unused)]
+impl<'a> Profile<'a> {
+    pub fn new(name: &'a str) -> Self {
+        Self {
+            name,
+            start: Instant::now(),
+        }
+    }
+
+    pub fn cond_new(name: &'a str, cond: bool) -> Option<Self> {
+        if cond {
+            Some(Self::new(name))
+        } else {
+            None
+        }
+    }
+}
+
+impl<'a> Drop for Profile<'a> {
+    fn drop(&mut self) {
+        let duration = self.start.elapsed();
+        info!("{}: cost {:?}", self.name, duration);
+    }
+}

crates/emmylua_doc_cli/src/markdown_generator/mod.rs

@@ -1,6 +1,6 @@
 use code_analysis::EmmyLuaAnalysis;
 
-
+#[allow(unused)]
 pub fn generate_markdown(analysis: &mut EmmyLuaAnalysis, output: &str) {
 
 }

crates/emmylua_ls/src/handlers/initialized/mod.rs

@@ -13,7 +13,7 @@ use crate::{
 };
 use client_config::get_client_config;
 pub use client_config::ClientConfig;
-use code_analysis::{uri_to_file_path, EmmyLuaAnalysis, Emmyrc, FileId};
+use code_analysis::{uri_to_file_path, EmmyLuaAnalysis, Emmyrc, FileId, Profile};
 use codestyle::load_editorconfig;
 use collect_files::collect_files;
 use log::info;
@@ -147,17 +147,21 @@ pub async fn init_analysis(
     }
 
     let mut count = 0;
-    while let Some(_) = rx.recv().await {
-        count += 1;
-
-        if client_id.is_vscode() {
-            status_bar.report_progress(
-                format!("diagnostic {}/{}", count, file_count).as_str(),
-                0.75,
-            );
-        }
-        if count == file_count {
-            break;
+    if file_count != 0 {
+        let text= format!("diagnose {} files", file_count);
+        let _p = Profile::new(text.as_str());
+        while let Some(_) = rx.recv().await {
+            count += 1;
+
+            if client_id.is_vscode() {
+                status_bar.report_progress(
+                    format!("diagnostic {}/{}", count, file_count).as_str(),
+                    0.75,
+                );
+            }
+            if count == file_count {
+                break;
+            }
         }
     }