lucky_bot/magic_admin/base/config/app_env.go

package config

import (
	"fmt"
	"github.com/spf13/viper"
)

type EnvConfig struct {
	Mysql   Mysql     `mapstructure:"mysql" json:"mysql" yaml:"mysql"`
	BizList []BizInfo `mapstructure:"biz-list" json:"biz-list" yaml:"biz-list"`
	JWT     JWT       `mapstructure:"jwt" json:"jwt" yaml:"jwt"`
	Zap     *Zap      `mapstructure:"zap" json:"zap" yaml:"zap"`
	Redis   *Redis    `mapstructure:"redis" json:"redis" yaml:"redis"`
	File    *File     `mapstructure:"file" json:"file" yaml:"file"`
	StsOss  *OssSts   `mapstructure:"sts-oss" json:"sts-oss" yaml:"sts-oss"`
}

var envConfig *EnvConfig

func EnvConf() *EnvConfig {
	if envConfig == nil {
		appEnvConfigInit()
	}
	return envConfig
}

// 环境配置初始化

func appEnvConfigInit() {
	configFile := GetConfigFileNameByMod(AppConf().Mod)
	vp := viper.New()
	vp.SetConfigFile(configFile)
	vp.SetConfigType("json")
	setEnvDefaultConf(vp)
	handVpToFile(vp)
	if err := vp.Unmarshal(&envConfig); err != nil {
		panic(err)
	}
}

func setEnvDefaultConf(v *viper.Viper) {
	v.SetDefault("file", &File{
		OssType:   OssTypeLocal,
		Path:      fmt.Sprintf("http://127.0.0.1:%d", AdminHostPort),
		ProxyPath: "/admin/api/static",
		StorePath: "./static",
		OriginConf: map[string]interface{}{
			"endpoint":          "oss-ap-southeast-1.aliyuncs.com",
			"access-key-id":     "LTAI5tHwrKhv2sd927cXYoRw",
			"access-key-secret": "9TezWQUvvXGvucUdXXFMNaIGjwmusa",
			"bucket-name":       "oss-sgp-prd-fomo-1",
			"bucket-url":        "https://oss-sgp-prd-fomo-1.oss-ap-southeast-1.aliyuncs.com",
			"base-path":         "test",
		},
	})
	v.SetDefault("mysql", Mysql{
		GeneralDB: GeneralDB{
			Prefix:       "",
			Port:         "3306",
			Config:       `charset=utf8mb4&parseTime=True&loc=Local`,
			Dbname:       AdminDbName,
			Username:     "root",
			Password:     "123456",
			Path:         "127.0.0.1",
			Engine:       "",
			LogMode:      "error",
			MaxIdleConns: 10,
			MaxOpenConns: 10,
			Singular:     false,
			LogZap:       false,
		}})
	v.SetDefault("biz-list", []BizInfo{
		{
			AliasName:  BizAliasName,
			ProxyUrl:   fmt.Sprintf("http://127.0.0.1:%d", BizHostPort),
			ProxyAlias: BizAliasName,
			GeneralDB: GeneralDB{
				Prefix:       "",
				Port:         "3306",
				Config:       `charset=utf8mb4&parseTime=True&loc=Local`,
				Dbname:       BizDbName,
				Username:     "root",
				Password:     "123456",
				Path:         "127.0.0.1",
				Engine:       "",
				LogMode:      "error",
				MaxIdleConns: 10,
				MaxOpenConns: 10,
				Singular:     false,
				LogZap:       false,
			},
		},
	})

	v.SetDefault("zap", &Zap{
		TagName:       "app",
		Level:         "debug",
		Prefix:        "AMS_",
		Format:        "_json",
		Director:      "logs",
		EncodeLevel:   "LowercaseLevelEncoder",
		StacktraceKey: "error",
		MaxAge:        3,
		RotationSize:  100,
		RotationCount: 3,
		ShowLine:      true,
		LogInConsole:  true,
		LogOutputFile: true,
	})

	v.SetDefault("redis", &Redis{
		Addr:           "127.0.0.1:6379",
		Password:       "",
		Db:             0,
		EnabledTls:     false,
		EnabledCluster: false,
		Cluster:        []string{"127.0.0.1:6379"},
	})

	v.SetDefault("jwt", &JWT{
		SigningKey:  fmt.Sprintf("%s.!@#$1234", BizAliasName),
		ExpiresTime: 24,
		Issuer:      "appTemp",
	})
	v.SetDefault("sts-oss", &OssSts{
		AccessKeyId:        "LTAI5tHwrKhv2sd927cXYoRw",
		AccessKeySecret:    "9TezWQUvvXGvucUdXXFMNaIGjwmusa",
		BucketName:         "oss-sgp-prd-fomo-1",
		BucketUrl:          "oss-sgp-prd-fomo-1.oss-ap-southeast-1.aliyuncs.com",
		BasePath:           "test",
		StsEndpoint:        "sts.ap-southeast-1.aliyuncs.com",
		StsDurationSeconds: 1200,
		StsRoleSessionName: "fomo-sts",
		StsRoleArn:         "acs:ram::5796137199632408:role/fomo-sts",
		StsRegion:          "oss-ap-southeast-1",
	})
}