archive/rsnext
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

improve vizualization (vercel/turbo#225)

Browse source
This commit is contained in:
Tobias Koppers 2022-08-30 18:31:35 +02:00 committed by GitHub
parent 1e8d63d10d
commit 42dda2f181
1 changed files with 59 additions and 31 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

54
packages/next-swc/crates/next-dev/src/turbo_tasks_viz.rs
View file

@ -1,10 +1,13 @@
use std::{str::FromStr, sync::Arc}; use std::{str::FromStr, sync::Arc, time::Duration};
use anyhow::Result; use anyhow::Result;
use mime::Mime; use mime::Mime;
use turbo_tasks::TurboTasks; use turbo_tasks::{get_invalidator, TurboTasks};
use turbo_tasks_fs::{File, FileContent}; use turbo_tasks_fs::{File, FileContent};
use turbo_tasks_memory::{stats::Stats, viz, MemoryBackend}; use turbo_tasks_memory::{
stats::{ReferenceType, Stats},
viz, MemoryBackend,
};
use turbopack_core::version::VersionedContentVc; use turbopack_core::version::VersionedContentVc;
use turbopack_dev_server::source::{ContentSource, ContentSourceVc}; use turbopack_dev_server::source::{ContentSource, ContentSourceVc};
@ -20,40 +23,65 @@ impl TurboTasksSourceVc {
} }
} }
const INVALIDATION_INTERVAL: Duration = Duration::from_secs(3);
#[turbo_tasks::value_impl] #[turbo_tasks::value_impl]
impl ContentSource for TurboTasksSource { impl ContentSource for TurboTasksSource {
#[turbo_tasks::function] #[turbo_tasks::function]
fn get(&self, path: &str) -> Result<VersionedContentVc> { fn get(&self, path: &str) -> Result<VersionedContentVc> {
let tt = &self.turbo_tasks; let tt = &self.turbo_tasks;
if path == "graph" { let invalidator = get_invalidator();
tokio::spawn({
async move {
tokio::time::sleep(INVALIDATION_INTERVAL).await;
invalidator.invalidate();
}
});
match path {
"graph" => {
let mut stats = Stats::new(); let mut stats = Stats::new();
let b = tt.backend(); let b = tt.backend();
b.with_all_cached_tasks(|task| { b.with_all_cached_tasks(|task| {
stats.add_id(b, task); stats.add_id(b, task);
}); });
let tree = stats.treeify(); let tree = stats.treeify(ReferenceType::Dependency);
let graph = viz::graph::visualize_stats_tree(tree); let graph = viz::graph::visualize_stats_tree(tree, ReferenceType::Dependency);
return Ok(FileContent::Content( Ok(FileContent::Content(
File::from_source(viz::graph::wrap_html(&graph)) File::from_source(viz::graph::wrap_html(&graph))
.with_content_type(Mime::from_str("text/html")?), .with_content_type(Mime::from_str("text/html")?),
) )
.into()); .into())
} }
if path == "table" { "call-graph" => {
let mut stats = Stats::new(); let mut stats = Stats::new();
let b = tt.backend(); let b = tt.backend();
b.with_all_cached_tasks(|task| { b.with_all_cached_tasks(|task| {
stats.add_id(b, task); stats.add_id(b, task);
}); });
let tree = stats.treeify(); let tree = stats.treeify(ReferenceType::Child);
let graph = viz::graph::visualize_stats_tree(tree, ReferenceType::Child);
Ok(FileContent::Content(
File::from_source(viz::graph::wrap_html(&graph))
.with_content_type(Mime::from_str("text/html")?),
)
.into())
}
"table" => {
let mut stats = Stats::new();
let b = tt.backend();
b.with_all_cached_tasks(|task| {
stats.add_id(b, task);
});
let tree = stats.treeify(ReferenceType::Dependency);
let table = viz::table::create_table(tree); let table = viz::table::create_table(tree);
return Ok(FileContent::Content( Ok(FileContent::Content(
File::from_source(viz::table::wrap_html(&table)) File::from_source(viz::table::wrap_html(&table))
.with_content_type(Mime::from_str("text/html")?), .with_content_type(Mime::from_str("text/html")?),
) )
.into()); .into())
}
_ => Ok(FileContent::NotFound.into()),
} }
Ok(FileContent::NotFound.into())
} }
#[turbo_tasks::function] #[turbo_tasks::function]