weld 0.4.0

Weld is a language and runtime for improving the performance of data-intensive applications.
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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104

//! Utilities for dumping code to a file.
//!
//! This module can be used both for writing code for consumption in another program (e.g., writing
//! LLVM files that can then be passed to Clang) or for debugging.

use time;
use uuid;

use crate::conf::DumpCodeConfig;
use crate::error::*;

use std::fs::OpenOptions;
use std::io::Error as IOError;
use std::io::Write;
use std::path::PathBuf;

use uuid::Uuid;

impl From<IOError> for WeldCompileError {
    fn from(err: IOError) -> WeldCompileError {
        WeldCompileError::new(err.to_string())
    }
}

#[derive(Copy, Clone, Debug, Hash, PartialEq, Eq)]
pub enum DumpCodeFormat {
    Weld,
    WeldOpt,
    LLVM,
    LLVMOpt,
    SIR,
    Assembly,
}

impl DumpCodeFormat {
    /// Returns a vector with all formats.
    pub fn all() -> Vec<DumpCodeFormat> {
        use self::DumpCodeFormat::*;
        vec![Weld, WeldOpt, LLVM, LLVMOpt, SIR, Assembly]
    }

    /// Returns a filename suffix for the format.
    pub fn suffix(self) -> String {
        use self::DumpCodeFormat::*;
        match self {
            WeldOpt | LLVMOpt => "-opt",
            _ => "",
        }
        .to_string()
    }

    /// Returns a filename extension for the format.
    pub fn extension(self) -> String {
        use self::DumpCodeFormat::*;
        match self {
            Weld | WeldOpt => "weld",
            LLVM | LLVMOpt => "ll",
            SIR => "sir",
            Assembly => "S",
        }
        .to_string()
    }
}

/// Writes code to a file using the given configuration.
///
/// The format determines the extension used for the dumped file.
pub fn write_code<T: AsRef<str>>(
    code: T,
    format: DumpCodeFormat,
    config: &DumpCodeConfig,
) -> WeldResult<()> {
    // Code dumping is not enabled - return.
    if !config.enabled {
        return Ok(());
    }

    // Format not registered: return.
    if !config.formats.contains(&format) {
        return Ok(());
    }

    let path = &mut PathBuf::new();
    path.push(&config.directory);
    path.push(&format!("{}{}", &config.filename, format.suffix()));
    path.set_extension(format.extension());

    info!("Writing code to {}", path.to_str().unwrap());

    let mut options = OpenOptions::new();
    let mut file = options.write(true).create_new(true).open(path)?;

    file.write_all(code.as_ref().as_bytes())?;
    Ok(())
}

/// Return a timestamp-based filename for `dumpCode`.
///
/// The timestamp has a 2-character identifier attached to prevent naming conflicts.
pub fn unique_filename() -> String {
    let uuid = Uuid::new_v4().to_simple().to_string();
    let suffix = &uuid[0..2];
    format!("code-{}-{}", time::now().to_timespec().sec, suffix)
}