DXR is a code search and navigation tool aimed at making sense of large projects. It supports full-text and regex searches as well as structural queries.

Git (74cd81b8ce)

VCS Links

Line Code
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77
/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at https://mozilla.org/MPL/2.0/. */

//! A module for writing time profiler traces out to a self contained HTML file.

use profile_traits::time::{ProfilerCategory, TimerMetadata};
use std::fs;
use std::io::{self, Write};
use std::path;

/// An RAII class for writing the HTML trace dump.
#[derive(Debug)]
pub struct TraceDump {
    file: fs::File,
}

#[derive(Debug, Serialize)]
struct TraceEntry {
    category: ProfilerCategory,
    metadata: Option<TimerMetadata>,

    #[serde(rename = "startTime")]
    start_time: u64,

    #[serde(rename = "endTime")]
    end_time: u64,
}

impl TraceDump {
    /// Create a new TraceDump and write the prologue of the HTML file out to
    /// disk.
    pub fn new<P>(trace_file_path: P) -> io::Result<TraceDump>
    where
        P: AsRef<path::Path>,
    {
        let mut file = fs::File::create(trace_file_path)?;
        write_prologue(&mut file)?;
        Ok(TraceDump { file: file })
    }

    /// Write one trace to the trace dump file.
    pub fn write_one(
        &mut self,
        category: &(ProfilerCategory, Option<TimerMetadata>),
        time: (u64, u64),
    ) {
        let entry = TraceEntry {
            category: category.0,
            metadata: category.1.clone(),
            start_time: time.0,
            end_time: time.1,
        };
        serde_json::to_writer(&mut self.file, &entry).unwrap();
        writeln!(&mut self.file, ",").unwrap();
    }
}

impl Drop for TraceDump {
    /// Write the epilogue of the trace dump HTML file out to disk on
    /// destruction.
    fn drop(&mut self) {
        write_epilogue(&mut self.file).unwrap();
    }
}

fn write_prologue(file: &mut fs::File) -> io::Result<()> {
    writeln!(file, "{}", include_str!("./trace-dump-prologue-1.html"))?;
    writeln!(file, "{}", include_str!("./trace-dump.css"))?;
    writeln!(file, "{}", include_str!("./trace-dump-prologue-2.html"))
}

fn write_epilogue(file: &mut fs::File) -> io::Result<()> {
    writeln!(file, "{}", include_str!("./trace-dump-epilogue-1.html"))?;
    writeln!(file, "{}", include_str!("./trace-dump.js"))?;
    writeln!(file, "{}", include_str!("./trace-dump-epilogue-2.html"))
}