saprkteamDash/pkg/config/setting.go

package config

import (
	"fmt"
	"github.com/spf13/viper"
	"gorm.io/gorm/logger"
)

var App Config

func Init() {
	viper.SetConfigName("config")
	viper.SetConfigType("yaml")
	viper.AddConfigPath("./configs")
	viper.AddConfigPath("./")
	if err := viper.ReadInConfig(); err != nil {
		panic(fmt.Errorf("ReadInConfig err:%v", err))
	}
	if err := viper.Unmarshal(&App); err != nil {
		panic(fmt.Errorf("Unmarshal err:%v", err))
	}
}

func GetLogLevel() logger.LogLevel {
	logMode := App.LogMode
	switch logMode {
	case "info":
		return logger.Info
	case "warn":
		return logger.Warn
	case "error":
		return logger.Error
	}
	return logger.Error
}