[Rust] CPU 성능 프로파일링: flamegraph

#Rust

flamegraph는 perf 기반의 성능 프로파일링을 위한 그래프 도구다.
어떤 함수가 CPU 리소스를 얼마나 점유했는지 등을 가시화해서 성능 최적화에 도움을 주는 용도다.
메모리 프로파일링은 이걸로 안되는거같다.

사전 조건

먼저 linux kernel 측정 도구인 perf가 깔려있어야 한다. 아래 포스트를 참고한다.
https://blog.naver.com/sssang97/223140670513

그리고 몇가지 보안 수준 옵션을 좀 조절해줘야 한다.
아래 옵션을 -1로 깔아서 다 열어주고

sudo sysctl kernel.perf_event_paranoid=-1

이것도 풀어준다.

echo 0 | sudo tee /proc/sys/kernel/kptr_restrict

CLI 설치

cargo install을 통해 설치한다.

cargo install flamegraph

이거면 된다.

기본 사용법

먼저 성능 측정을 위한 간단한 서버를 작성해봤다.

[dependencies]
axum = "0.7.4"
tokio = { versopm = "1", features = ["full"] }

[profile.release]
debug = true

use axum::{routing::get, Router};

#[tokio::main]
async fn main() {
    // build our application with a route
    let app = Router::new()
        .route("/", get(root))
        .route("/eat-cpu", get(eat_cpu));

    // run our app with hyper, listening globally on port 3000
    let listener = tokio::net::TcpListener::bind("0.0.0.0:3000").await.unwrap();
    axum::serve(listener, app).await.unwrap();
}

// basic handler that responds with a static string
async fn root() -> &'static str {
    "Hello, World!"
}

async fn eat_cpu() -> &'static str {
    let mut hash_set = std::collections::HashSet::new();

    for i in 0..1000000 {
        hash_set.insert(i);
    }

    "cpu"
}