Flat directory/module structure

src/export/mod.rs

@@ -0,0 +1,89 @@
+mod asciidoc;
+mod csv;
+mod json;
+mod markdown;
+
+use self::asciidoc::AsciidocExporter;
+use self::csv::CsvExporter;
+use self::json::JsonExporter;
+use self::markdown::MarkdownExporter;
+
+use std::fs::{File, OpenOptions};
+use std::io::{Result, Write};
+
+use crate::benchmark_result::BenchmarkResult;
+use crate::units::Unit;
+
+/// The desired form of exporter to use for a given file.
+#[derive(Clone)]
+pub enum ExportType {
+    /// Asciidoc Table
+    Asciidoc,
+
+    /// CSV (comma separated values) format
+    Csv,
+
+    /// JSON format
+    Json,
+
+    /// Markdown table
+    Markdown,
+}
+
+/// Interface for different exporters.
+trait Exporter {
+    /// Export the given entries in the serialized form.
+    fn serialize(&self, results: &[BenchmarkResult], unit: Option<Unit>) -> Result<Vec<u8>>;
+}
+
+struct ExporterWithFilename {
+    exporter: Box<dyn Exporter>,
+    filename: String,
+}
+
+/// Handles the management of multiple file exporters.
+pub struct ExportManager {
+    exporters: Vec<ExporterWithFilename>,
+}
+
+impl ExportManager {
+    /// Create a new ExportManager
+    pub fn new() -> ExportManager {
+        ExportManager {
+            exporters: Vec::new(),
+        }
+    }
+
+    /// Add an additional exporter to the ExportManager
+    pub fn add_exporter(&mut self, export_type: ExportType, filename: &str) -> Result<()> {
+        let _ = File::create(filename)?;
+
+        let exporter: Box<dyn Exporter> = match export_type {
+            ExportType::Asciidoc => Box::new(AsciidocExporter::default()),
+            ExportType::Csv => Box::new(CsvExporter::default()),
+            ExportType::Json => Box::new(JsonExporter::default()),
+            ExportType::Markdown => Box::new(MarkdownExporter::default()),
+        };
+        self.exporters.push(ExporterWithFilename {
+            exporter,
+            filename: filename.to_string(),
+        });
+
+        Ok(())
+    }
+
+    /// Write the given results to all Exporters contained within this manager
+    pub fn write_results(&self, results: &[BenchmarkResult], unit: Option<Unit>) -> Result<()> {
+        for e in &self.exporters {
+            let file_content = e.exporter.serialize(results, unit)?;
+            write_to_file(&e.filename, &file_content)?;
+        }
+        Ok(())
+    }
+}
+
+/// Write the given content to a file with the specified name
+fn write_to_file(filename: &str, content: &[u8]) -> Result<()> {
+    let mut file = OpenOptions::new().write(true).open(filename)?;
+    file.write_all(content)
+}