mirror of
https://github.com/Rayzggz/server_torii.git
synced 2025-06-21 06:27:35 +08:00
feat: Default Main config
This commit is contained in:
@ -27,6 +27,20 @@ type MainConfig struct {
|
|||||||
|
|
||||||
// LoadMainConfig Read the configuration file and return the configuration object
|
// LoadMainConfig Read the configuration file and return the configuration object
|
||||||
func LoadMainConfig(basePath string) (*MainConfig, error) {
|
func LoadMainConfig(basePath string) (*MainConfig, error) {
|
||||||
|
|
||||||
|
defaultCfg := MainConfig{
|
||||||
|
Port: "25555",
|
||||||
|
WebPath: "/torii",
|
||||||
|
RulePath: "/www/server_torii/config/rules",
|
||||||
|
ErrorPage: "/www/server_torii/config/error_page",
|
||||||
|
LogPath: "/www/server_torii/log/",
|
||||||
|
NodeName: "Server Torii",
|
||||||
|
ConnectingHostHeaders: []string{"Torii-Real-Host"},
|
||||||
|
ConnectingIPHeaders: []string{"Torii-Real-IP"},
|
||||||
|
ConnectingURIHeaders: []string{"Torii-Original-URI"},
|
||||||
|
ConnectingCaptchaStatusHeaders: []string{"Torii-Captcha-Status"},
|
||||||
|
}
|
||||||
|
|
||||||
exePath, err := os.Executable()
|
exePath, err := os.Executable()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
@ -38,12 +52,12 @@ func LoadMainConfig(basePath string) (*MainConfig, error) {
|
|||||||
|
|
||||||
data, err := os.ReadFile(configPath)
|
data, err := os.ReadFile(configPath)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return &defaultCfg, err
|
||||||
}
|
}
|
||||||
|
|
||||||
var cfg MainConfig
|
var cfg MainConfig
|
||||||
if err := yaml.Unmarshal(data, &cfg); err != nil {
|
if err := yaml.Unmarshal(data, &cfg); err != nil {
|
||||||
return nil, err
|
return &defaultCfg, err
|
||||||
}
|
}
|
||||||
|
|
||||||
return &cfg, nil
|
return &cfg, nil
|
||||||
|
2
main.go
2
main.go
@ -27,7 +27,7 @@ func main() {
|
|||||||
// Load MainConfig
|
// Load MainConfig
|
||||||
cfg, err := config.LoadMainConfig(basePath)
|
cfg, err := config.LoadMainConfig(basePath)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatalf("Load config failed: %v", err)
|
log.Printf("[ERROR] Load config failed: %v. Using default config.", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Load rules
|
// Load rules
|
||||||
|
Reference in New Issue
Block a user