Perfilando código Rust: encontrando gargalos na prática

Perfilando código Rust: encontrando gargalos na prática

Rust é famoso por seu desempenho, mas até o código mais rápido pode conter gargalos inesperados. A perfilagem é a arte e a ciência de identificar esses problemas. Neste artigo, vamos explorar na prática como encontrar e eliminar "gargalos" em aplicações Rust, usando ferramentas e técnicas modernas.

Por que a perfilagem é importante para Rust?

Embora Rust garanta a ausência de um coletor de lixo e ofereça controle de baixo nível sobre a memória, isso não protege contra algoritmos ineficientes, alocações desnecessárias ou uso incorreto do cache. Sem perfilagem, você corre o risco de otimizar o que não precisa. A perfilagem responde a perguntas como:

\n
• Quais funções consomem mais CPU?
\n
• Onde ocorrem alocações frequentes de memória?
\n
• Quais partes do código bloqueiam a execução (por exemplo, devido a mutexes)?
• Quão eficientemente o cache do processador está sendo usado?

Ferramentas para perfilagem em Rust

Vamos considerar três abordagens principais que cobrem 90% das tarefas de perfilagem.

1. Benchmarking integrado com cargo bench

A maneira mais simples é usar cargo bench do ecossistema Rust. Ele permite medir o tempo de execução de funções individuais.

Exemplo de benchmark para uma função que ordena um vetor:

// benches/my_bench.rs\nuse criterion::{black_box, criterion_group, criterion_main, Criterion};\n\nfn sort_benchmark(c: &mut Criterion) {\n    let mut data: Vec<i32> = (0..10000).rev().collect();\n\n    c.bench_function("sort 10k", |b| {\n        b.iter(|| {\n            // black_box previne a otimização do compilador\n            let mut clone = black_box(data.clone());\n            clone.sort();\n            black_box(clone)\n        })\n    });\n}\n\ncriterion_group!(benches, sort_benchmark);\ncriterion_main!(benches);

Execução: cargo bench

Você receberá um relatório com tempo, desvio padrão e comparação com execuções anteriores. Isso é ideal para testes de regressão de desempenho.

2. Perfilador de sistema perf (Linux)

perf é uma ferramenta poderosa do kernel Linux. Ele coleta amostras de contadores de desempenho (ciclos de CPU, cache misses, instruções) sem modificar o código.

Uso básico:

# Coleta de dados\nperf record -F 99 --call-graph dwarf ./my_rust_program\n\n# Visualizar relatório\nperf report --stdio\n\n# Gerar flamegraph\nperf script | stackcollapse-perf.pl | flamegraph.pl > flame.svg

A flag -F 99 define a frequência de amostragem para 99 Hz (geralmente suficiente). --call-graph dwarf ativa o registro da pilha de chamadas. A saída é um flamegraph interativo (SVG), onde o eixo X representa o tempo e o eixo Y a pilha de chamadas. A largura do retângulo mostra quanto tempo a função ocupou.

3. Ferramenta flamegraph para Rust

Existe um utilitário especializado cargo-flamegraph que simplifica a criação de flamegraphs para projetos Rust:

cargo install flamegraph\ncargo flamegraph --bin my_binary

Após a execução, um arquivo flamegraph.svg será criado na raiz do projeto. Abra-o no navegador — você verá imediatamente as partes "quentes" do código.

Exemplo prático: otimizando um parser

Imagine que estamos escrevendo um parser JSON em Rust e ele está lento. Vamos usar a perfilagem para encontrar o problema.

Passo 1: Benchmark

Criamos um benchmark para analisar um arquivo JSON grande (por exemplo, 10 MB). Executamos cargo bench e vemos que a análise leva 2,5 segundos. Isso é muito.

Passo 2: Flamegraph

Executamos cargo flamegraph no binário que analisa este arquivo. Abrimos o SVG e vemos que 60% do tempo