101 lines
2.0 KiB
Go
101 lines
2.0 KiB
Go
package configs
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/xinliangnote/go-gin-api/pkg/env"
|
|
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
var config = new(Config)
|
|
|
|
type Config struct {
|
|
MySQL struct {
|
|
Read struct {
|
|
Addr string `toml:"addr"`
|
|
User string `toml:"user"`
|
|
Pass string `toml:"pass"`
|
|
Name string `toml:"name"`
|
|
} `toml:"read"`
|
|
Write struct {
|
|
Addr string `toml:"addr"`
|
|
User string `toml:"user"`
|
|
Pass string `toml:"pass"`
|
|
Name string `toml:"name"`
|
|
} `toml:"write"`
|
|
Base struct {
|
|
MaxOpenConn int `toml:"maxOpenConn"`
|
|
MaxIdleConn int `toml:"maxIdleConn"`
|
|
ConnMaxLifeTime time.Duration `toml:"connMaxLifeTime"`
|
|
} `toml:"base"`
|
|
} `toml:"mysql"`
|
|
|
|
Redis struct {
|
|
Addr string `toml:"addr"`
|
|
Pass string `toml:"pass"`
|
|
Db int `toml:"db"`
|
|
MaxRetries int `toml:"maxRetries"`
|
|
PoolSize int `toml:"poolSize"`
|
|
MinIdleConns int `toml:"minIdleConns"`
|
|
} `toml:"redis"`
|
|
|
|
Mail struct {
|
|
Host string `toml:"host"`
|
|
Port int `toml:"port"`
|
|
User string `toml:"user"`
|
|
Pass string `toml:"pass"`
|
|
To string `toml:"to"`
|
|
} `toml:"mail"`
|
|
|
|
JWT struct {
|
|
Secret string `toml:"secret"`
|
|
ExpireDuration time.Duration `toml:"expireDuration"`
|
|
} `toml:"jwt"`
|
|
|
|
Aes struct {
|
|
Key string `toml:"key"`
|
|
Iv string `toml:"iv"`
|
|
} `toml:"aes"`
|
|
|
|
Rsa struct {
|
|
Private string `toml:"private"`
|
|
Public string `toml:"public"`
|
|
} `toml:"rsa"`
|
|
|
|
Cmd struct {
|
|
GenTables string `toml:"genTables"`
|
|
} `toml:"cmd"`
|
|
}
|
|
|
|
func init() {
|
|
viper.SetConfigName(env.Active().Value() + "_configs")
|
|
viper.SetConfigType("toml")
|
|
viper.AddConfigPath("./configs")
|
|
|
|
if err := viper.ReadInConfig(); err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
if err := viper.Unmarshal(config); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func Get() Config {
|
|
return *config
|
|
}
|
|
|
|
func ProjectName() string {
|
|
return "go-gin-api"
|
|
}
|
|
|
|
func ProjectPort() string {
|
|
return ":9999"
|
|
}
|
|
|
|
func ProjectLogFile() string {
|
|
return fmt.Sprintf("./logs/%s-access.log", ProjectName())
|
|
}
|