DEV Community

Shubham Chadokar
Shubham Chadokar

Posted on • Originally published at schadokar.dev

Use Environment Variable in your next Golang Project

When it comes to creating a production-grade application, using the environment variable in the application is de facto.

This post was originally posted on my personal blog ๐Ÿ–Š๏ธ schadokar.dev.

Why should we use the environment variable?

Suppose you have an application with many features and each feature need to access the Database. You configured all the DB information like DBURL, DBNAME, USERNAME and PASSWORD in each feature.

There are a few major disadvantages to this approach, there can be many.

  • Security Issue:

    • You're entering all the information in the code. Now, all the unauthorized person also have access to the DB.
    • If you're using code versioning tool like git then the details of your DB will go public once you push the code.
  • Code Management:

    • If you are changing a single variable then you have to change in all the features. There is a high possibility that you'll miss one or two. ๐Ÿ˜Œ been there
    • You can categorize the environment variables like PROD, DEV, or TEST. Just prefix the variable with the environment.

In the start, it might look like some extra work, but this will reward you a lot in your project.

โš ๏ธ Just don't forget to include your environment files in the .gitignore.โš ๏ธ


It is time for some action. ๐Ÿ”จ

What are we going to do in this tutorial?

In this tutorial, we will access environment variables in 3 different ways.

You can use according to your requirement.

  • os package
  • godotenv package
  • viper package

Create a Project

Create a project go-env-ways outside the $GOPATH.

Initialize the module

Open the terminal inside the project root directory, and run the below command.

go mod init go-env-ways
Enter fullscreen mode Exit fullscreen mode

This module will keep a record of all the packages and their version used in the project. It is similar to package.json in nodejs.


Let's start with the easiest one, using os package.

os Package

Golang provides os package, an easy way to configure and access the environment variable.

To set the environment variable,

os.Setenv(key, value)
Enter fullscreen mode Exit fullscreen mode

To get the environment variable,

value := os.Getenv(key)
Enter fullscreen mode Exit fullscreen mode

Create a new file main.go inside the project.

package main

import (
  "fmt"
  "os"
)

// use os package to get the env variable which is already set
func envVariable(key string) string {

  // set env variable using os package
  os.Setenv(key, "gopher")

  // return the env variable using os package
  return os.Getenv(key)
}

func main() {
    // os package
  value := envVariable("name")

  fmt.Printf("os package: %s = %s \n", "name", value)
}
Enter fullscreen mode Exit fullscreen mode

Run the below command to check.

go run main.go

// Output
os package: name = gopher
Enter fullscreen mode Exit fullscreen mode

GoDotEnv Package

The easiest way to load the .env file is using godotenv package.

Install

Open the terminal in the project root directory.

go get github.com/joho/godotenv
Enter fullscreen mode Exit fullscreen mode
// Load the .env file in the current directory
godotenv.Load()

// or

godotenv.Load(".env")
Enter fullscreen mode Exit fullscreen mode

Load method can load multiple env files at once. This also supports yaml. For more information check out the documentation.

Create a new .env file in the project root directory.

STRONGEST_AVENGER=Thor
Enter fullscreen mode Exit fullscreen mode

Update the main.go.

package main

import (

    ...
    // Import godotenv
  "github.com/joho/godotenv"
)


// use godot package to load/read the .env file and
// return the value of the key
func goDotEnvVariable(key string) string {

  // load .env file
  err := godotenv.Load(".env")

  if err != nil {
    log.Fatalf("Error loading .env file")
  }

  return os.Getenv(key)
}

func main() {
    // os package
    ... 

  // godotenv package
  dotenv := goDotEnvVariable("STRONGEST_AVENGER")

  fmt.Printf("godotenv : %s = %s \n", "STRONGEST_AVENGER", dotenv)
}
Enter fullscreen mode Exit fullscreen mode

Open the terminal and run the main.go.

go run main.go

// Output
os package: name = gopher

godotenv : STRONGEST_AVENGER = Thor
Enter fullscreen mode Exit fullscreen mode

Just add the code at the end of the os package in the main function.


Viper Package

Viper is one of the most popular packages in the golang community. Many Go projects are built using Viper including Hugo, Docker Notary, Mercury.

Viper is a complete configuration solution for Go applications including 12-Factor apps. It is designed to work within an application and can handle all types of configuration needs and formats. Reading from JSON, TOML, YAML, HCL, envfile and Java properties config files

For more information read the official documentation of viper

Install

Open the terminal in the project root directory.

go get github.com/spf13/viper
Enter fullscreen mode Exit fullscreen mode

To set the config file and path

viper.SetConfigFile(".env")
Enter fullscreen mode Exit fullscreen mode

To read the config file

viper.ReadInConfig()
Enter fullscreen mode Exit fullscreen mode

To get the value from the config file using key

viper.Get(key)
Enter fullscreen mode Exit fullscreen mode

Update the main.go.

import (
  "fmt"
  "log"
  "os"

  "github.com/joho/godotenv"
  "github.com/spf13/viper"
)

// use viper package to read .env file
// return the value of the key
func viperEnvVariable(key string) string {

  // SetConfigFile explicitly defines the path, name and extension of the config file.
  // Viper will use this and not check any of the config paths.
  // .env - It will search for the .env file in the current directory
  viper.SetConfigFile(".env")

  // Find and read the config file
  err := viper.ReadInConfig()

  if err != nil {
    log.Fatalf("Error while reading config file %s", err)
  }

  // viper.Get() returns an empty interface{}
  // to get the underlying type of the key,
  // we have to do the type assertion, we know the underlying value is string
  // if we type assert to other type it will throw an error
  value, ok := viper.Get(key).(string)

  // If the type is a string then ok will be true
  // ok will make sure the program not break
  if !ok {
    log.Fatalf("Invalid type assertion")
  }

  return value
}

func main() {

    // os package  
    ...

  // godotenv package
  ...

  // viper package read .env
  viperenv := viperEnvVariable("STRONGEST_AVENGER")

  fmt.Printf("viper : %s = %s \n", "STRONGEST_AVENGER", viperenv)
}
Enter fullscreen mode Exit fullscreen mode

Open the terminal and run the main.go.

go run main.go

// Output
os package: name = gopher

godotenv : STRONGEST_AVENGER = Thor

viper : STRONGEST_AVENGER = Thor
Enter fullscreen mode Exit fullscreen mode

Viper is not limited to .env files.

It supports:

  • setting defaults
  • reading from JSON, TOML, YAML, HCL, envfile and Java properties config files
  • live watching and re-reading of config files (optional)
  • reading from environment variables
  • reading from remote config systems (etcd or Consul), and watching changes
  • reading from command line flags
  • reading from buffer
  • setting explicit values

Viper can be thought of as a registry for all of your applications configuration needs.

Let's experiment: ๐Ÿ’ฃ

Create a new config.yaml file in the project root directory.

I_AM_INEVITABLE: "I am Iron Man"
Enter fullscreen mode Exit fullscreen mode

To set the config filename

viper.SetConfigName("config")
Enter fullscreen mode Exit fullscreen mode

To set the config file path

// Look in the current working directory
viper.AddConfigPath(".")
Enter fullscreen mode Exit fullscreen mode

To read the config file

viper.ReadInConfig()
Enter fullscreen mode Exit fullscreen mode

Update the main.go

// use viper package to load/read the config file or .env file and
// return the value of the key
func viperConfigVariable(key string) string {

  // name of config file (without extension)
  viper.SetConfigName("config")
  // look for config in the working directory
  viper.AddConfigPath(".")

  // Find and read the config file
  err := viper.ReadInConfig()

  if err != nil {
    log.Fatalf("Error while reading config file %s", err)
  }

  // viper.Get() returns an empty interface{}
  // to get the underlying type of the key,
  // we have to do the type assertion, we know the underlying value is string
  // if we type assert to other type it will throw an error
  value, ok := viper.Get(key).(string)

  // If the type is a string then ok will be true
  // ok will make sure the program not break
  if !ok {
    log.Fatalf("Invalid type assertion")
  }

  return value
}

func main() {

  // os package
  ...

  // godotenv package
  ...

  // viper package read .env
  ...

  // viper package read config file
  viperconfig := viperConfigVariable("I_AM_INEVITABLE")  

  fmt.Printf("viper config : %s = %s \n", "I_AM_INEVITABLE", viperconfig)  
}
Enter fullscreen mode Exit fullscreen mode

Open the terminal and run the main.go

go run main.go

// Output
os package: name = gopher

godotenv : STRONGEST_AVENGER = Thor

viper : STRONGEST_AVENGER = Thor

viper config : I_AM_INEVITABLE = I am Iron Man
Enter fullscreen mode Exit fullscreen mode

Conclusion

That's it, now you can explore more of their secrets. If you find something worth sharing don't hesitate.

The complete code is available in the github.


Top comments (0)