risingwave_rt/
prof.rs

1// Copyright 2025 RisingWave Labs
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7//     http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
15use std::path::PathBuf;
16use std::time::Duration;
17
18use thiserror_ext::AsReport;
19
20pub(crate) fn spawn_prof_thread(profile_path: String) -> std::thread::JoinHandle<()> {
21    tracing::info!("writing prof data to directory {}", profile_path);
22    let profile_path = PathBuf::from(profile_path);
23
24    std::fs::create_dir_all(&profile_path).unwrap();
25
26    std::thread::spawn(move || {
27        let mut cnt = 0;
28        loop {
29            let guard = pprof::ProfilerGuardBuilder::default()
30                .blocklist(&["libc", "libgcc", "pthread", "vdso"])
31                .build()
32                .unwrap();
33            std::thread::sleep(Duration::from_secs(60));
34            match guard.report().build() {
35                Ok(report) => {
36                    let profile_svg = profile_path.join(format!("{}.svg", cnt));
37                    let file = std::fs::File::create(&profile_svg).unwrap();
38                    report.flamegraph(file).unwrap();
39                    tracing::info!("produced {:?}", profile_svg);
40                }
41                Err(err) => {
42                    tracing::warn!(error = %err.as_report(), "failed to generate flamegraph");
43                }
44            }
45            cnt += 1;
46        }
47    })
48}
risingwave_rt/prof.rs

risingwave_rt/
prof.rs
