feat: ✨ 通用 profile 方式 (#532)

* feat: ✨ 通用 profile 方式

* feat: ✨ compile after starting capture frame

---------

Co-authored-by: pshu <[email protected]>

Author: stormslowly

.cargo/config.toml

@@ -3,6 +3,7 @@ codecov = "llvm-cov nextest --workspace"
 lint = "clippy --workspace --all-targets --all-features -- -D warnings -Aclippy::module_inception"
 lintfix = "clippy --workspace --fix --allow-dirty --allow-staged  --all-targets --all-features -- -D warnings -Aclippy::module_inception"
 mako = "run --bin mako"
+profile_mako = "run --features profile --bin mako"
 
 # macos
 [target.x86_64-apple-darwin]

crates/mako/src/main.rs

@@ -5,6 +5,8 @@ use std::sync::Arc;
 use mako_core::anyhow::Result;
 use mako_core::clap::Parser;
 use mako_core::tokio;
+#[cfg(feature = "profile")]
+use mako_core::tokio::sync::Notify;
 use mako_core::tracing::debug;
 
 use crate::compiler::Args;
@@ -91,23 +93,39 @@ async fn main() -> Result<()> {
 
     #[cfg(feature = "profile")]
     {
+        let notify = Arc::new(Notify::new());
+        let to_be_notify = notify.clone();
+
+        tokio::spawn(async move {
+            let compiler = compiler.clone();
+
+            to_be_notify.notified().await;
+
+            compiler.compile().unwrap();
+
+            if cli.watch {
+                let d = crate::dev::DevServer::new(root.clone(), compiler.clone());
+                d.serve().await;
+            }
+        });
+
         mako_core::puffin::set_scopes_on(true);
         let native_options = Default::default();
-        let compiler = compiler.clone();
         let _ = mako_core::eframe::run_native(
             "puffin egui eframe",
             native_options,
-            Box::new(move |_cc| Box::new(ProfileApp::new(compiler))),
+            Box::new(move |_cc| Box::new(ProfileApp::new(notify))),
         );
     }
 
     #[cfg(not(feature = "profile"))]
-    compiler.compile()?;
-
-    if cli.watch {
-        let d = crate::dev::DevServer::new(root.clone(), compiler);
-        // TODO: when in Dev Mode, Dev Server should start asap, and provider a loading  while in first compiling
-        d.serve().await;
+    {
+        compiler.compile()?;
+        if cli.watch {
+            let d = crate::dev::DevServer::new(root.clone(), compiler);
+            // TODO: when in Dev Mode, Dev Server should start asap, and provider a loading  while in first compiling
+            d.serve().await;
+        }
     }
     Ok(())
 }

crates/mako/src/profile_gui.rs

@@ -1,23 +1,24 @@
+#[cfg(feature = "profile")]
 use std::sync::Arc;
 
 #[cfg(feature = "profile")]
 use mako_core::eframe::egui;
+#[cfg(feature = "profile")]
+use mako_core::tokio::sync::Notify;
 
-use crate::compiler::Compiler;
-
+#[cfg(feature = "profile")]
 pub struct ProfileApp {
-    #[allow(dead_code)]
-    frame_counter: u64,
-    #[allow(dead_code)]
-    compiler: Arc<Compiler>,
+    notified: bool,
+    notify: Arc<Notify>,
 }
 
+#[cfg(feature = "profile")]
 impl ProfileApp {
     #[allow(dead_code)]
-    pub fn new(compiler: Arc<Compiler>) -> Self {
+    pub fn new(notify: Arc<Notify>) -> Self {
         Self {
-            frame_counter: 0,
-            compiler,
+            notified: false,
+            notify,
         }
     }
 }
@@ -27,12 +28,10 @@ impl mako_core::eframe::App for ProfileApp {
     fn update(&mut self, ctx: &egui::Context, _frame: &mut mako_core::eframe::Frame) {
         mako_core::puffin::GlobalProfiler::lock().new_frame(); // call once per frame!
 
-        mako_core::puffin_egui::profiler_window(ctx);
-
-        if self.frame_counter == 0 {
-            self.compiler.compile().unwrap();
+        if !self.notified {
+            self.notified = true;
+            self.notify.notify_one();
         }
-
-        self.frame_counter += 1;
+        mako_core::puffin_egui::profiler_window(ctx);
     }
 }