Mini Post: Parsing Environment Variables Safely in Rust

bloom ⋄ posted almost 2 years ago ⋄ last updated almost 2 years ago ⋄ 2 min read

Almost every app I write nowadays uses an environment file, and therefore I always have to read things from the environment and parse them into different forms. Doing this safely with defaults in Rust was a bit tricky for me at first, but I’ve landed on a solution I like.

Optional Numbers#

The reason I was looking for code like this was to try and get a port for a database from the environment with a reasonable fallback. Here’s what I landed on:

1
use std::env;
2
use dotenv::dotenv;
3

4
fn main() {
5
    dotenv().ok();
6
    let count = env::var("COUNT")
7
        // convert to Option<String>
8
        .ok()
9
        // convert to Option<u64> via parse
10
        .and_then(|value| value.parse::<u64>().ok())
11
        // Use 5 as a default if either the value was not found or it could
12
        // not be converted.
13
        .unwrap_or(5)
14

15
    println!("{}", count * 5);
16
}

Default Strings#

Of course, you can always just get a required value like this:

1
std::env;
2
use dotenv::dotenv;
3

4
fn main() {
5
    dotenv().ok();
6
    let test = env::var("TEST")
7
        .unwrap_or(String::from("Test"));
8

9
    println!("{}", test);
10
}

Required Environment Variables#

And of course, you can make them required by just unwraping them, but that will crash your program:

1
std::env;
2
use dotenv::dotenv;
3

4
fn main() {
5
    dotenv().ok();
6
    let test = env::var("TEST").unwrap();
7

8
    println!("{}", test);
9
}

thread 'main' panicked at src\main.rs:6:35:
called `Result::unwrap()` on an `Err` value: NotPresent

You can instead print a helpful message:

1
std::env;
2
use dotenv::dotenv;
3

4
fn main() {
5
    dotenv().ok();
6
    let test = env::var("TEST");
7

8
    match test {
9
        Ok(test) => {
10
            println!("{}", test)
11
        }
12
        Err(_) => {
13
            eprintln!("Missing required environment variable 'TEST'");
14
            return;
15
        }
16
    }
17
}

Missing required environment variable 'TEST'