Add a metric containing the size of generated documentation

Author: Kobzol

Cargo.lock

@@ -33,6 +33,7 @@ filetime = "0.2.14"
 walkdir = "2"
 flate2 = { version = "1.0.22", features = ["rust_backend"] }
 rayon = "1.5.2"
+fs_extra = "1.2.0"
 
 [target.'cfg(windows)'.dependencies]
 miow = "0.3"

collector/Cargo.toml

@@ -848,6 +848,12 @@ impl<'a> Processor for BenchProcessor<'a> {
                 if let Some(ref profile) = res.1 {
                     store_artifact_sizes_into_stats(&mut res.0, profile);
                 }
+                if let Profile::Doc = data.profile {
+                    let doc_dir = data.cwd.join("target/doc");
+                    if doc_dir.is_dir() {
+                        store_documentation_size_into_stats(&mut res.0, &doc_dir);
+                    }
+                }
 
                 match data.scenario {
                     Scenario::Full => {
@@ -894,6 +900,17 @@ impl<'a> Processor for BenchProcessor<'a> {
     }
 }
 
+fn store_documentation_size_into_stats(stats: &mut Stats, doc_dir: &Path) {
+    match fs_extra::dir::get_size(doc_dir) {
+        Ok(size) => stats.insert("size:doc_bytes".to_string(), size as f64),
+        Err(error) => log::error!(
+            "Cannot get size of documentation directory {}: {:?}",
+            doc_dir.display(),
+            error
+        ),
+    }
+}
+
 fn store_artifact_sizes_into_stats(stats: &mut Stats, profile: &SelfProfile) {
     for artifact in profile.artifact_sizes.iter() {
         stats

collector/src/execute.rs

@@ -231,6 +231,9 @@ pub enum Metric {
     LlvmBitcodeSize,
     #[serde(rename = "size:llvm_ir")]
     LlvmIrSize,
+    /// Total bytes of a generated documentation directory
+    #[serde(rename = "size:doc_bytes")]
+    DocByteSize,
 }
 
 impl Metric {
@@ -260,6 +263,7 @@ impl Metric {
             Metric::AssemblyFileSize => "size:assembly_file",
             Metric::LlvmBitcodeSize => "size:llvm_bitcode",
             Metric::LlvmIrSize => "size:llvm_ir",
+            Metric::DocByteSize => "size:doc_bytes",
         }
     }