XinYeYouKu/config/config.go

package config

import (
	"github.com/BurntSushi/toml"
	"github.com/qit-team/snow-core/config"
	"os"
)

const (
	ProdEnv  = "production" //线上环境
	BetaEnv  = "beta"       //beta环境
	DevEnv   = "develop"    //开发环境
	LocalEnv = "local"      //本地环境
)

var srvConf *Config

// ------------------------配置文件解析
type Config struct {
	ServiceName             string             `toml:"ServiceName"`
	Env                     string             `toml:"Env"`
	Debug                   bool               `toml:"Debug"`
	PrometheusCollectEnable bool               `toml:"PrometheusCollectEnable"`
	SkyWalkingOapServer     string             `toml:"SkyWalkingOapServer"`
	Log                     config.LogConfig   `toml:"Log"`
	Redis                   config.RedisConfig `toml:"Redis"`
	Mns                     config.MnsConfig   `toml:"AliMns"`
	Db                      config.DbConfig    `toml:"Db"`
	Api                     config.ApiConfig   `toml:"Api"`
	Admin                   config.ApiConfig   `toml:"Admin"`
	Nacos                   Nacos              `toml:"Nacas"`
	Rpc                     Rpc                `toml:"Rpc"`
	AppKey                  string             `toml:"AppKey"`
	KafkaUrl                []string           `toml:"KafkaUrl"`
	AliSms                  AliSms             `toml:"AliSms"`
	AdminGate               string             `toml:"AdminGate"`
	PluginDir               string             `toml:"PLUGIN_DIR"`
	ImgDir                  string             `toml:"IMG_DIR"`
	OtherDir                string             `toml:"OTHER_DIR"`
	AliOss                  AliOss             `toml:"AliOss"`
	TemplateUrl             string             `toml:"TemplateUrl"`
	Message                 Message            `toml:"Message"`
	OpenApi                 OpenApi            `toml:"OpenApi"`
	OpenApiMarketConfig     MarketConfig       `toml:"MarketConfig"`
	RocketMq                []string           `toml:"RocketMq"`
	Sm2                     Sm2                `toml:"Sm2"`
}

type Sm2 struct {
	PublicKey  string
	PrivateKey string
}

type OpenApi struct {
	MerchantId string
	SecretKey  string
	IsProd     bool
	NotifyUrl  string
	TimeOut    int
}

type Message struct {
	ClientKey    string
	ClientSecret string
	ServerIndex  string
	TempIndex    string
	Url          string
}

type AliOss struct {
	AccessKey       string
	AccessKeySecret string
	EndPoint        string
	BucKet          string
	Domain          string
	Dir             string
}

type AliSms struct {
	Url             string
	AccessKeyId     string
	AccessKeySecret string
	SignName        string
	TemplateSmsCode string
	Nums            int
}

type Rpc struct {
	Port    string
	Servers []string
}
type Nacos struct {
	Url         string
	Port        int64
	ServiceName string
	GroupName   string
	ClusterName string
	ServicePort int64
}

type MarketConfig struct {
	AppId     string `json:"app_id"`   //APP ID
	Sign      string `json:"sign"`     //签名
	ReqCode   string `json:"req_code"` //固定值：voucher.create
	MemId     string `json:"mem_id"`   //商户号
	PosId     string `json:"pos_id"`   //商户方平台号
	Host      string `json:"-"`
	SecretKey string
}

func newConfig() *Config {
	return new(Config)
}

// ------------------------ 加载配置 ------------------------//
func Load(path string) (*Config, error) {
	_, err := os.Stat(path)
	if err != nil {
		return nil, err
	}

	conf := newConfig()
	if _, err := toml.DecodeFile(path, conf); err != nil {
		return nil, err
	}
	srvConf = conf
	return conf, nil
}

// 当前配置
func GetConf() *Config {
	return srvConf
}

// 是否调试模式
func IsDebug() bool {
	return srvConf.Debug
}

// 当前环境，默认本地开发
func GetEnv() string {
	if srvConf.Env == "" {
		return LocalEnv
	}
	return srvConf.Env
}

// 是否当前环境
func IsEnvEqual(env string) bool {
	return GetEnv() == env
}