magmaus3/continuwuity
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

instrument tokio before/after poll hooks

Browse source 
Signed-off-by: Jason Volk <jason@zemos.net>
This commit is contained in:
Jason Volk 2025-03-30 01:54:55 +00:00
parent 095734a8e7
commit 5bf5afaec8
1 changed files with 24 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

24
src/main/runtime.rs
View file

@ -61,6 +61,8 @@ pub(super) fn new(args: &Args) -> Result<tokio::runtime::Runtime> {
#[cfg(tokio_unstable)] #[cfg(tokio_unstable)]
builder builder
.on_task_spawn(task_spawn) .on_task_spawn(task_spawn)
.on_before_task_poll(task_enter)
.on_after_task_poll(task_leave)
.on_task_terminate(task_terminate); .on_task_terminate(task_terminate);
#[cfg(tokio_unstable)] #[cfg(tokio_unstable)]
@ -215,3 +217,25 @@ fn task_spawn(meta: &tokio::runtime::TaskMeta<'_>) {}
), ),
)] )]
fn task_terminate(meta: &tokio::runtime::TaskMeta<'_>) {} fn task_terminate(meta: &tokio::runtime::TaskMeta<'_>) {}
#[cfg(tokio_unstable)]
#[tracing::instrument(
name = "enter",
level = "trace",
skip_all,
fields(
id = %meta.id()
),
)]
fn task_enter(meta: &tokio::runtime::TaskMeta<'_>) {}
#[cfg(tokio_unstable)]
#[tracing::instrument(
name = "leave",
level = "trace",
skip_all,
fields(
id = %meta.id()
),
)]
fn task_leave(meta: &tokio::runtime::TaskMeta<'_>) {}