mirror of
https://github.com/go-admin-team/go-admin.git
synced 2026-10-09 15:38:32 +00:00
41 lines
908 B
Go
41 lines
908 B
Go
package config
|
|
|
|
import "github.com/spf13/viper"
|
|
|
|
type Application struct {
|
|
ReadTimeout int
|
|
WriterTimeout int
|
|
Host string
|
|
Port string
|
|
Name string
|
|
JwtSecret string
|
|
Mode string
|
|
DemoMsg string
|
|
EnableDP bool
|
|
}
|
|
|
|
func InitApplication(cfg *viper.Viper) *Application {
|
|
return &Application{
|
|
ReadTimeout: cfg.GetInt("readTimeout"),
|
|
WriterTimeout: cfg.GetInt("writerTimeout"),
|
|
Host: cfg.GetString("host"),
|
|
Port: portDefault(cfg),
|
|
Name: cfg.GetString("name"),
|
|
JwtSecret: cfg.GetString("jwtSecret"),
|
|
Mode: cfg.GetString("mode"),
|
|
DemoMsg: cfg.GetString("demoMsg"),
|
|
EnableDP: cfg.GetBool("enabledp"),
|
|
}
|
|
}
|
|
|
|
var ApplicationConfig = new(Application)
|
|
|
|
func portDefault(cfg *viper.Viper) string {
|
|
if cfg.GetString("port") == "" {
|
|
return "8000"
|
|
} else {
|
|
return cfg.GetString("port")
|
|
}
|
|
}
|
|
|