やってみる

アウトプットすべく己を導くためのブログ。その試行錯誤すらたれ流す。

Rustで環境変数を取得する

Rustで環境変数を取得する

 std::env::var()で。

成果物

参考

前回からの続き

  1. Rustでコマンドライン引数を受け取る
  2. Rustのファイル読込
  3. Rustでリファクタリング(モジュール性とエラー処理の向上)
  4. Rustでテスト駆動開発

大文字小文字を区別しない

 まずは失敗するテストから書く。

#[cfg(test)]
mod test {
    use super::*;

    #[test]
    fn case_sensitive() {
        let query = "duct";
// Rust
// 安全かつ高速で生産的
// 三つを選んで
// ガムテープ
        let contents = "\
Rust:
safe, fast, productive.
Pick three.
Duct tape.";

        assert_eq!(
            vec!["safe, fast, productive."],
            search(query, contents)
        );
    }

    #[test]
    fn case_insensitive() {
        let query = "rUsT";
// (最後の行のみ)
// 私を信じて
        let contents = "\
Rust:
safe, fast, productive.
Pick three.
Trust me.";

        assert_eq!(
            vec!["Rust:", "Trust me."],
            search_case_insensitive(query, contents)
        );
    }
}

 実装。

pub fn search_case_insensitive<'a>(query: &str, contents: &'a str) -> Vec<&'a str> {
    let query = query.to_lowercase();
    let mut results = Vec::new();
    for line in contents.lines() {
        if line.to_lowercase().contains(&query) { results.push(line); }
    }
    results
}

 str.to_lowercase()がポイント。

区別する/しない

pub struct Config {
    pub query: String,
    pub filename: String,
    pub case_sensitive: bool,
}
impl Config {
    pub fn new(args: &[String]) -> Result<Config, &'static str> {
        if args.len() < 3 { return Err("引数不足。2つ必要です。第一引数に検索文字列、第二引数に検索対象ファイルパス。"); }
        let query = args[1].clone();
        let filename = args[2].clone();
        let case_sensitive = std::end::var("CASE_SENSITIVE").is_err(); // 環境変数の取得
        Ok(Config { query, filename, case_sensitive })
    }
}
pub fn run(config: Config) -> Result<(), Box<Error>> {
    let mut f = File::open(config.filename)?;
    let mut contents = String::new();
    f.read_to_string(&mut contents)?;
    let result = if config.case_sensitive { search(&config.query, &contents); }
                 else { search_case_insensitive(&config.query, &contents); };
    for line in result { println!("{}", line); }
    Ok(())
}

std::env::var()

 指定した環境変数があればResultOkを返す。なければErr

Result.is_err()

 Errならtrue, Okならfalse

実行

 toを含む。(大文字小文字を区別する)

$ cargo run to poem.txt
...
query: to
filename: poem.txt
Are you nobody, too?
How dreary to be somebody!

 toを含む。(大文字小文字を区別しない)

$ CASE_SENSITIVE=1 cargo run to poem.txt
...
query: to
filename: poem.txt
Are you nobody, too?
How dreary to be somebody!
To tell your name the livelong day
To an admiring bog!

 Toも出てきた。成功。

 ところでドキュメント$ CASE_INSENSITIVE=1 cargo run to poem.txtという間違った名前の環境変数で指定しているので期待通りにならない。誤字と思われる。

 というか、文言の意味とコード間で整合性がとれていない。sensitiveは敏感の意。insensitiveはその対義語で鈍感の意。たぶんsensitiveだと大文字小文字の区別して、insensitiveだと区別しないのが正しい挙動と思われる。つまり以下のようにするのが正しいのでは?

pub struct Config {
    pub case_sensitive: bool,
}
impl Config {
    pub fn new(args: &[String]) -> Result<Config, &'static str> {
        let case_sensitive = std::end::var("CASE_SENSITIVE").is_ok(); // is_errでなく
    }
}
pub fn run(config: Config) -> Result<(), Box<Error>> {
    let result = if config.case_sensitive { search(&config.query, &contents); }
                 else { search_case_insensitive(&config.query, &contents); };
}

 実行結果は以下。

$ cargo run to poem.txt    Finished dev [unoptimized + debuginfo] target(s) in 0.11s
     Running `/tmp/work/Rust.Minigrep.Env.20190701132148/src/2/minigrep/target/debug/minigrep to poem.txt`
query: to
filename: poem.txt
Are you nobody, too?
How dreary to be somebody!
To tell your name the livelong day
To an admiring bog!

 CASE_SENSITIVE環境変数を指定すると以下。(大文字小文字を区別する)

$ CASE_SENSITIVE=1 cargo run to poem.txt   Compiling minigrep v0.1.0 (/tmp/work/Rust.Minigrep.Env.20190701132148/src/2/minigrep)
    Finished dev [unoptimized + debuginfo] target(s) in 4.84s
     Running `/tmp/work/Rust.Minigrep.Env.20190701132148/src/2/minigrep/target/debug/minigrep to poem.txt`
query: to
filename: poem.txt
Are you nobody, too?
How dreary to be somebody!

 環境変数を指定しないと大文字小文字を区別せずに出力する。たぶんこのほうがいい。

対象環境

$ uname -a
Linux raspberrypi 4.19.42-v7+ #1219 SMP Tue May 14 21:20:58 BST 2019 armv7l GNU/Linux

前回まで