adds config

Author: fighting-dreamer

application.yml

@@ -0,0 +1,3 @@
+app_port: 8080
+app_name: "brew_machine"
+log_level: "info"

application_test.yml

@@ -0,0 +1,3 @@
+app_port: 8080
+app_name: "brew_machine"
+log_level: "info"

config/config.go

@@ -0,0 +1,51 @@
+package config
+
+import (
+	"github.com/spf13/viper"
+)
+
+type Config struct {
+	appPort  int
+	appName  string
+	logLevel string
+}
+
+var appConfiguration *Config
+
+func LoadTestConfig() {
+	load("application_test")
+}
+
+func Load() {
+	load("application")
+}
+
+func load(configFileName string) {
+	viper.AutomaticEnv()
+	viper.SetConfigName(configFileName)
+	viper.SetConfigType("yaml")
+	viper.AddConfigPath("./")
+	viper.AddConfigPath("../")
+	viper.AddConfigPath("../../")
+	viper.ReadInConfig()
+
+	appConfiguration = &Config{
+		appPort:  getIntOrPanic("app_port"),
+		appName:  getStringOrPanic("app_name"),
+		logLevel: getStringOrPanic("log_level"),
+	}
+
+	return
+}
+
+func AppPort() int {
+	return appConfiguration.appPort
+}
+
+func AppName() string {
+	return appConfiguration.appName
+}
+
+func LogLevel() string {
+	return appConfiguration.logLevel
+}

config/utils.go

@@ -0,0 +1,49 @@
+package config
+
+import (
+	"errors"
+	"fmt"
+	"os"
+	"strconv"
+
+	"github.com/spf13/viper"
+)
+
+func getIntOrPanic(key string) int {
+	intValue, err := strconv.Atoi(getConfigStringValue(key))
+	if err != nil {
+		panicForkey(key, err)
+	}
+	return intValue
+}
+
+func getBoolOrPanic(key string) bool {
+	boolValue, err := strconv.ParseBool(getConfigStringValue(key))
+	if err != nil {
+		panicForkey(key, err)
+	}
+	return boolValue
+}
+
+func getStringOrPanic(key string) string {
+	value := getConfigStringValue(key)
+	if value == "" {
+		panicForkey(key, errors.New("config is not set"))
+	}
+	return value
+}
+
+func getConfigStringValue(key string) string {
+	if !viper.IsSet(key) && os.Getenv(key) == "" {
+		fmt.Printf("config %s is not set\n", key)
+	}
+	value := os.Getenv(key)
+	if value == "" {
+		value = viper.GetString(key)
+	}
+	return value
+}
+
+func panicForkey(key string, err error) {
+	panic(fmt.Sprintf("Error %v occured while reading config %s", err.Error(), key))
+}

go.mod

@@ -1,3 +1,7 @@
 module nipun.io/brew_machine
 
 go 1.16
+
+require (
+	github.com/spf13/viper v1.8.1
+)