rust/neon
1
0
Fork 0
mirror of https://github.com/neondatabase/neon.git synced 2026-06-04 22:10:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

utils: add ?force=true hint for CPU profiler (#10368)

Browse Source 
This makes it less annoying to try to take a CPU profile when a
continuous profile is already running.
This commit is contained in:
Erik Grinaker
2025-01-13 15:23:42 +01:00
committed by GitHub
parent b2d0e1a519
commit b31ed0acd1
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
libs/utils/src/http/endpoint.rs
View File

@@ -372,7 +372,11 @@ pub async fn profile_cpu_handler(req: Request<Body>) -> Result<Response<Body>, A
match PROFILE_LOCK.try_lock() {
Ok(lock) => break lock,
Err(_) if force => PROFILE_CANCEL.notify_waiters(),
Err(_) => return Err(ApiError::Conflict("profiler already running".into())),
Err(_) => {
return Err(ApiError::Conflict(
"profiler already running (use ?force=true to cancel it)".into(),
))
}
}
tokio::time::sleep(Duration::from_millis(1)).await; // don't busy-wait
};