sway-utils/src/performance.rs - master - githubmirror/sway

sway-utils/src/performance.rs
92 строки3 KB
Igor Rončević
Extend compiler metrics with `DeclEngine` metrics (#7664)
20 июн 2026, 20:09
20 июн 2026, 20:098ab7c59
use serde::{Deserialize, Serialize};

#[derive(Clone, Debug, Deserialize, Serialize)]
pub struct CompilationPhaseMetrics {
    pub phase: String,
    pub elapsed: f64,
    pub memory_usage: Option<u64>,
}

#[derive(Clone, Debug, Deserialize, Serialize)]
pub struct ConcurrentSlabMetrics {
    pub slab: String,
    pub length: usize,
    pub capacity: usize,
    /// Approximate memory usage of the content of the slab [Vec].
    pub memory_usage: usize,
    /// Approximate memory usage of the total content of elements of type `T`
    /// the slab slots point to via `Arc<T>`. **This does not
    /// include any additional memory allocated by individual `T` elements,
    /// just the raw size of each `T`.**
    pub content_memory_usage: usize,
    pub free_slots_length: usize,
    pub free_slots_capacity: usize,
}

#[derive(Clone, Debug, Default, Deserialize, Serialize)]
pub struct DeclEngineMetrics {
    pub slabs: Vec<ConcurrentSlabMetrics>,
}

#[derive(Clone, Debug, Default, Deserialize, Serialize)]
pub struct PerformanceMetrics {
    pub bytecode_size: usize,
    pub reused_programs: u64,
    pub compilation_phases: Vec<CompilationPhaseMetrics>,
    pub decl_engine: DeclEngineMetrics,
}

#[derive(serde::Serialize, Clone)]
pub struct FunctionEntryPoint {
    /// The original entry point function name.
    pub fn_name: String,
    /// The immediate instruction offset at which the entry function begins.
    pub imm: u64,
    /// The function selector (only `Some` for contract ABI methods).
    pub selector: Option<[u8; 4]>,
}

#[macro_export]
// Time the given expression and print/save the result.
macro_rules! time_expr {
    ($pkg_name:expr, $description:expr, $key:expr, $expression:expr, $build_config:expr, $data:expr) => {{
        use std::io::{BufRead, Read, Write};
        if let Some(cfg) = $build_config {
            if cfg.profile {
                println!("/dyno start {} {}", $pkg_name, $description);
                let output = { $expression };
                println!("/dyno stop {} {}", $pkg_name, $description);
                output
            } else if cfg.time_phases || cfg.metrics_outfile.is_some() {
                let expr_start = std::time::Instant::now();
                let output = { $expression };
                let elapsed = expr_start.elapsed();
                if cfg.time_phases {
                    println!("  Time elapsed to {}: {:?}", $description, elapsed);
                }
                if cfg.metrics_outfile.is_some() {
                    #[cfg(not(target_os = "macos"))]
                    let memory_usage = {
                        use sysinfo::{System, SystemExt};
                        let mut sys = System::new();
                        sys.refresh_system();
                        Some(sys.used_memory())
                    };
                    #[cfg(target_os = "macos")]
                    let memory_usage = None;

                    $data.compilation_phases.push(CompilationPhaseMetrics {
                        phase: $key.to_string(),
                        elapsed: elapsed.as_secs_f64(),
                        memory_usage,
                    });
                }
                output
            } else {
                $expression
            }
        } else {
            $expression
        }
    }};
}