126 lines
2.5 KiB
Go
126 lines
2.5 KiB
Go
package config
|
|
|
|
import (
|
|
"errors"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/caarlos0/env/v6"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/joho/godotenv"
|
|
"github.com/yaoapp/kun/exception"
|
|
"github.com/yaoapp/kun/log"
|
|
)
|
|
|
|
// Conf 配置参数
|
|
var Conf Config
|
|
|
|
// LogOutput 日志输出
|
|
var LogOutput *os.File // 日志文件
|
|
|
|
func init() {
|
|
filename, _ := filepath.Abs(filepath.Join(".", ".env"))
|
|
if _, err := os.Stat(filename); errors.Is(err, os.ErrNotExist) {
|
|
Conf = Load()
|
|
return
|
|
}
|
|
Conf = LoadFrom(filename)
|
|
if Conf.Mode == "production" {
|
|
Production()
|
|
} else if Conf.Mode == "development" {
|
|
Development()
|
|
}
|
|
}
|
|
|
|
// LoadFrom 从配置项中加载
|
|
func LoadFrom(envfile string) Config {
|
|
|
|
file, err := filepath.Abs(envfile)
|
|
if err != nil {
|
|
log.Warn("Can't load env file. %s", err.Error())
|
|
}
|
|
err = godotenv.Overload(file)
|
|
if err != nil {
|
|
log.Warn("Can't load env file. %s", err.Error())
|
|
}
|
|
|
|
return Load()
|
|
}
|
|
|
|
// Load 加载配置
|
|
func Load() Config {
|
|
cfg := Config{}
|
|
if err := env.Parse(&cfg); err != nil {
|
|
exception.New("Can't read config %s", 500, err.Error()).Throw()
|
|
}
|
|
cfg.Root, _ = filepath.Abs(cfg.Root)
|
|
return cfg
|
|
}
|
|
|
|
// Production 设定为生产环境
|
|
func Production() {
|
|
Conf.Mode = "production"
|
|
log.SetLevel(log.InfoLevel)
|
|
log.SetFormatter(log.TEXT)
|
|
if Conf.LogMode == "JSON" {
|
|
log.SetFormatter(log.JSON)
|
|
}
|
|
gin.SetMode(gin.ReleaseMode)
|
|
ReloadLog()
|
|
}
|
|
|
|
// Development 设定为开发环境
|
|
func Development() {
|
|
Conf.Mode = "development"
|
|
log.SetLevel(log.TraceLevel)
|
|
log.SetFormatter(log.TEXT)
|
|
if Conf.LogMode == "JSON" {
|
|
log.SetFormatter(log.JSON)
|
|
}
|
|
gin.SetMode(gin.DebugMode)
|
|
ReloadLog()
|
|
}
|
|
|
|
// ReloadLog 重新打开日志
|
|
func ReloadLog() {
|
|
CloseLog()
|
|
OpenLog()
|
|
}
|
|
|
|
// OpenLog 打开日志
|
|
func OpenLog() {
|
|
if Conf.Log != "" {
|
|
logfile, err := filepath.Abs(Conf.Log)
|
|
if err != nil {
|
|
log.With(log.F{"file": logfile}).Error(err.Error())
|
|
return
|
|
}
|
|
|
|
logpath := filepath.Dir(logfile)
|
|
if _, err := os.Stat(logpath); os.IsNotExist(err) {
|
|
if err := os.MkdirAll(logpath, os.ModePerm); err != nil {
|
|
log.With(log.F{"file": logfile}).Error(err.Error())
|
|
return
|
|
}
|
|
}
|
|
LogOutput, err = os.OpenFile(logfile, os.O_WRONLY|os.O_CREATE|os.O_APPEND, 0644)
|
|
if err != nil {
|
|
log.With(log.F{"file": logfile}).Error(err.Error())
|
|
return
|
|
}
|
|
|
|
log.SetOutput(LogOutput)
|
|
gin.DefaultWriter = LogOutput
|
|
}
|
|
}
|
|
|
|
// CloseLog 关闭日志
|
|
func CloseLog() {
|
|
if LogOutput != nil {
|
|
err := LogOutput.Close()
|
|
if err != nil {
|
|
log.Error(err.Error())
|
|
return
|
|
}
|
|
}
|
|
}
|