mirror of
https://github.com/ZSCNetSupportDept/scheduler.git
synced 2025-10-28 12:35:03 +08:00
49 lines
802 B
Go
49 lines
802 B
Go
package config
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/spf13/viper"
|
|
"os"
|
|
)
|
|
|
|
func Load() {
|
|
|
|
viper.SetConfigName("config")
|
|
viper.SetConfigType("yaml")
|
|
viper.AddConfigPath(".")
|
|
|
|
if err := viper.ReadInConfig(); err != nil {
|
|
fmt.Printf("Error reading config file: %v\n", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
ListenPort = viper.GetInt("ListenPort")
|
|
File = viper.GetString("File")
|
|
DB.Path = viper.GetString("DB.Path")
|
|
//DB.Port = viper.GetString("DB.Port")
|
|
DB.Type = viper.GetString("DB.Type")
|
|
|
|
err := check()
|
|
if err != nil {
|
|
fmt.Println("check your config!")
|
|
os.Exit(1)
|
|
}
|
|
|
|
debugprint()
|
|
|
|
}
|
|
|
|
func check() error {
|
|
|
|
return nil
|
|
|
|
}
|
|
func debugprint() {
|
|
|
|
fmt.Printf("ListenPort=%v\n", ListenPort)
|
|
fmt.Printf("File=%v\n", File)
|
|
fmt.Printf("database path : %s\n", DB.Path)
|
|
fmt.Printf("database type:%s\n", DB.Type)
|
|
|
|
}
|