GO/app/config/app.go

package config

import (
	"fmt"
	"gopkg.in/ini.v1"
	"log"
	"os"
	"path/filepath"
)

var Config = &App{}

const ConfigProd = "prod"

type App struct {
	Page       Page       `ini:"page"`
	Redis      Redis      `ini:"redis"`
	Mysql      Mysql      `ini:"mysql"`
	Server     Server     `ini:"server"`
	RunModel   string     `ini:"runModel"`
	Debug      bool       `ini:"debug"`
	Upload     Upload     `ini:"upload"`
	Wxapp      Wxapp      `ini:"wxapp"`
	Wxopen     Wxopen     `ini:"wxopen"`
	Che300     Che300     `ini:"che300"`
	Alipay     Alipay     `ini:"alipay"`
	Wxpay      Wxpay      `ini:"wxpay"`
	Jpush      Jpush      `ini:"jpush"`
	AliSms     AliSms     `ini:"aliSms"`
	Setting    Setting    `ini:"setting"`
	AliOcr     AliOcr     `ini:"aliOcr"`
	TencentEss TencentEss `ini:"tencentEss"`
	AppleLogin AppleLogin `ini:"appleLogin"`
	OpenSearch OpenSearch `ini:"openSearch"`
}
type IniConfig struct {
	ConfigPath       string
	ConfigName       string
	RunModelName     string
	RunModel         string
	RunModelErrAllow bool // 允许不使用拓展配置文件
}

func InitConfig(config *IniConfig) {
	if config.ConfigName == "" {
		config.ConfigName = "app.ini"
	}
	// 读取配置文件
	load, err := ini.Load(filepath.Join(config.ConfigPath, config.ConfigName))
	if err != nil {
		log.Fatal("配置文件读取错误，err:", err)
		return
	}
	// 环境配置
	if config.RunModel == "" {
		config.RunModel = os.Getenv("RunModel")
		if config.RunModel == "" {
			config.RunModel = load.Section("").Key("defaultModel").String()
			if config.RunModel == "" {
				config.RunModel = "test"
			}
		}
	}
	if config.RunModelName == "" {
		config.RunModelName = fmt.Sprintf("app.%s.ini", config.RunModel)
	}
	if err := load.Append(filepath.Join(config.ConfigPath, config.RunModelName)); err != nil {
		if config.RunModelErrAllow {
			log.Println("环境配置文件读取错误，err:", err)
		} else {
			log.Fatal("环境配置文件读取错误，err:", err)
		}
	}
	// 映射配置
	if err := load.MapTo(Config); err != nil {
		log.Fatal("配置文件映射错误，err:", err)
		return
	}
	if Config.RunModel == "" {
		Config.RunModel = config.RunModel
	}
}

// IsProd @Title 判断是否正式环境
func IsProd() bool {
	return Config.RunModel == ConfigProd
}