util.go

package util

import (
	"os"

	"github.com/sirupsen/logrus"
	"github.com/spf13/viper"
)

const (
	// DefaultDir is default config file search dir
	DefaultDir = "/etc/paas/"
	// DefaultFileName is default config file name
	DefaultFileName = "redis"
	// DefaultEnvPrefixKey is default prefix of environment variable
	DefaultEnvPrefixKey = ""
	// EnvPrefixKey is prefix of environment variable
	EnvPrefixKey = "ENV_PREFIX"
	// ConfigDirKey is key to get file search path from environment variable
	ConfigDirKey = "CONFIG_DIR"
	// ConfigNameKey is key to get file name from environment variable
	ConfigNameKey = "CONFIG_NAME"
)

//LoadParamsFromEnv will use env params to create viper.Viper
func LoadParamsFromEnv() *viper.Viper {
	v := viper.New()
	prefix := os.Getenv(EnvPrefixKey)
	if prefix == "" {
		prefix = DefaultEnvPrefixKey
		logrus.Warnf("ENV_PREFIX not exist in env Use default env prefix: %s", DefaultEnvPrefixKey)
	} else {
		logrus.Warnf("Use EnvPrefixKey: %s", prefix)
	}
	v.SetEnvPrefix(prefix)
	v.AutomaticEnv()
	return v
}

//LoadParamsFromVolume  wile use volume params create viper.Viper
func LoadParamsFromVolume() (*viper.Viper, error) {
	v := viper.New()
	configDir := os.Getenv(ConfigDirKey)
	fileName := os.Getenv(ConfigNameKey)

	//Use default DIR
	if configDir == "" {
		configDir = DefaultDir
		logrus.Warnf("ConfigDirKey not exist in env Use default dir %s", DefaultDir)
	} else {
		logrus.Infof("Use Config_Dir: %s", configDir)
	}

	////Use default config file name
	if fileName == "" {
		fileName = DefaultFileName
		logrus.Warnf("ConfigNameKey not exist in env Use default name %s", DefaultFileName)
	} else {
		logrus.Infof("Use CONFIG_NAME: %s", fileName)
	}

	v.SetConfigName(fileName)
	v.AddConfigPath(configDir)

	return v, v.ReadInConfig()
}

func LoadMixedParams() (*viper.Viper, error) {
	v := viper.New()
	configDir := os.Getenv(ConfigDirKey)
	fileName := os.Getenv(ConfigNameKey)
	prefix := os.Getenv(EnvPrefixKey)

	if prefix == "" {
		prefix = DefaultEnvPrefixKey
		logrus.Warnf("ENV_PREFIX not exist in env Use default env prefix: %s", DefaultEnvPrefixKey)
	} else {
		logrus.Warnf("Use EnvPrefixKey: %s", prefix)
	}

	//Use default DIR
	if configDir == "" {
		configDir = DefaultDir
		logrus.Warnf("ConfigDirKey not exist in env Use default dir %s", DefaultDir)
	} else {
		logrus.Infof("Use Config_Dir: %s", configDir)
	}
	//Use default config file name
	if fileName == "" {
		fileName = DefaultFileName
		logrus.Warnf("ConfigNameKey not exist in env Use default name %s", DefaultFileName)
	} else {
		logrus.Infof("Use CONFIG_NAME: %s", fileName)
	}
	v.SetConfigName(fileName)
	v.AddConfigPath(configDir)
	v.SetEnvPrefix(prefix)
	v.AutomaticEnv()
	return v, v.ReadInConfig()
}