// starting server
port := os.Getenv("APP_PORT")
log.Printf("Starting server at port %v\n", port)
log.Fatal(http.ListenAndServe(fmt.Sprintf(":%v", port), handler))

$ APP_PORT=8081 go run .
2022/02/02 10:10:52 Starting server at port 8081

package util

import (
	"os"
	"strconv"
)

func GetEnv(key string, defaultValue string) string {
	val, ok := os.LookupEnv(key)
	if !ok {
		return defaultValue
	}
	return val
}

func GetEnvInt(key string, defaultValue int) int {
	val, ok := os.LookupEnv(key)
	if !ok {
		return defaultValue
	}
	v, err := strconv.Atoi(val)
	if err != nil {
		return defaultValue
	}
	return v
}

func GetEnvBool(key string, defaultValue bool) bool {
	val, ok := os.LookupEnv(key)
	if !ok {
		return defaultValue
	}
	v, err := strconv.ParseBool(val)
	if err != nil {
		return defaultValue
	}
	return v
}

// starting server
port := util.GetEnvInt("APP_PORT", 8080)
log.Printf("Starting server at port %v\n", port)
log.Fatal(http.ListenAndServe(fmt.Sprintf(":%v", port), handler))

app:
  port: 8080

db:
  driver: 'postgres'
  host: 'john.db.elephantsql.com'
  port: 5432
  username: 'fcricryh'
  password: 'F5a7wATfocTUNww1Dm14AfebtPaysqIn'
  database: 'fcricryh'

package config

import (
	"fmt"
	"strings"

	"github.com/spf13/viper"
)

func LoadConfig() {
	viper.SetConfigName("config")                          // กำหนดชื่อไฟล์ config (without extension)
	viper.SetConfigType("yaml")                            // ระบุประเภทของไฟล์ config
	viper.AddConfigPath(".")                               // ระบุตำแหน่งของไฟล์ config อยู่ที่ working directory

	err := viper.ReadInConfig() // อ่านไฟล์ config
	if err != nil {             // ถ้าอ่านไฟล์ config ไม่ได้ให้ panic ไปเลย
		panic(fmt.Errorf("fatal error config file: %w", err))
	}
}

func main() {
	config.LoadConfig()
	database.ConnectDB()
  // ...
	// starting server
	port := viper.GetInt("app.port")
	log.Printf("Starting server at port %v\n", port)
	log.Fatal(http.ListenAndServe(fmt.Sprintf(":%v", port), handler))
}

func init() {
	viper.SetConfigName("config")                          // กำหนดชื่อไฟล์ config (without extension)
	viper.SetConfigType("yaml")                            // ระบุประเภทของไฟล์ config
	viper.AddConfigPath(".")                               // ระบุตำแหน่งของไฟล์ config อยู่ที่ working directory
	viper.AutomaticEnv()                                   // ให้อ่านค่าจาก env มา replace ในไฟล์ config
	viper.SetEnvKeyReplacer(strings.NewReplacer(".", "_")) // แปลง _ underscore ใน env เป็น . dot notation ใน viper

	err := viper.ReadInConfig() // อ่านไฟล์ config
	if err != nil {             // ถ้าอ่านไฟล์ config ไม่ได้ให้ข้ามไปเพราะให้เอาค่าจาก env มาแทนได้
		fmt.Println("please consider environment variables", err.Error())
	}
}

// APP_PORT=8081 go run .

func LoadConfig() {
	viper.SetConfigName("config")                          // กำหนดชื่อไฟล์ config (without extension)
	viper.SetConfigType("yaml")                            // ระบุประเภทของไฟล์ config
	viper.AddConfigPath(".")                               // ระบุตำแหน่งของไฟล์ config อยู่ที่ working directory
	viper.AutomaticEnv()                                   // ให้อ่านค่าจาก env มา replace ในไฟล์ config
	viper.SetEnvKeyReplacer(strings.NewReplacer(".", "_")) // แปลง _ underscore ใน env เป็น . dot notation ใน viper

	err := viper.ReadInConfig() // อ่านไฟล์ config
	if err != nil {             // ถ้าอ่านไฟล์ config ไม่ได้ให้ข้ามไปเพราะให้เอาค่าจาก env มาแทนได้
		fmt.Println("please consider environment variables", err.Error())
	}

  // กำหนด Default Value
	viper.SetDefault("app.port", 8080)
}

package config

import (
	"fmt"
	"log"
	"strings"

	"github.com/spf13/viper"
)

type configuration struct {
	App appConfig
	Db  dbConfig
}

type appConfig struct {
	Port uint
}

type dbConfig struct {
	Driver   string
	Host     string
	Port     uint
	Username string
	Password string
	Database string
}

var Config *configuration

func LoadConfig() {
	viper.SetConfigName("config")                          // กำหนดชื่อไฟล์ config (without extension)
	viper.SetConfigType("yaml")                            // ระบุประเภทของไฟล์ config
	viper.AddConfigPath(".")                               // ระบุตำแหน่งของไฟล์ config อยู่ที่ working directory
	viper.AutomaticEnv()                                   // ให้อ่านค่าจาก env มา replace ในไฟล์ config
	viper.SetEnvKeyReplacer(strings.NewReplacer(".", "_")) // แปลง _ underscore ใน env เป็น . dot notation ใน viper

	err := viper.ReadInConfig() // อ่านไฟล์ config
	if err != nil {             // ถ้าอ่านไฟล์ config ไม่ได้ให้ข้ามไปเพราะให้เอาค่าจาก env มาแทนได้
		fmt.Println("please consider environment variables", err.Error())
	}

	// กำหนด Default Value
	viper.SetDefault("app.port", 8080)

	// Decode config ด้วย Unmarshling
	Config = &configuration{}
	err = viper.Unmarshal(Config)
	if err != nil {
		log.Fatalf("unable to decode config into struct, %v", err)
	}
}

func LoadConfig() {
	// ...

	// กำหนด Default Value
	viper.SetDefault("app.port", 8080)

	Config = &configuration{
		App: appConfig{
			Port: viper.GetUint("app.port"),
		},
		Db: dbConfig{
			Driver:   viper.GetString("db.driver"),
			Host:     viper.GetString("db.host"),
			Port:     viper.GetUint("db.port"),
			Username: viper.GetString("db.username"),
			Password: viper.GetString("db.password"),
			Database: viper.GetString("db.database"),
		},
  }
  // err = viper.Unmarshal(Config)
	// if err != nil {
	// 	log.Fatalf("unable to decode config into struct, %v", err)
	// }
}

func LoadConfig() {
  // ...

	// ตรวจสอบว่ากำหนดค่ามาครบหรือไม่
	validate := validator.New()

	err = validate.Struct(Config)
	if err != nil {
		log.Fatalf("load config error, %v", err)
	}
}

func main() {
	config.LoadConfig()
	database.ConnectDB()
  // ...
	// starting server
	port := config.Config.App.Port
	log.Printf("Starting server at port %v\n", port)
	log.Fatal(http.ListenAndServe(fmt.Sprintf(":%v", port), handler))
}