mirror of
https://github.com/ZSCNetSupportDept/scheduler.git
synced 2025-10-28 12:35:03 +08:00
86 lines
1.8 KiB
Go
86 lines
1.8 KiB
Go
package config
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/spf13/viper"
|
|
"os"
|
|
)
|
|
|
|
func Load() {
|
|
// where to read config
|
|
viper.SetConfigName("config")
|
|
viper.SetConfigType("yaml")
|
|
viper.AddConfigPath(".")
|
|
|
|
readconfig()
|
|
|
|
check()
|
|
|
|
debugprint()
|
|
|
|
if len(os.Args) != 1 {
|
|
handleArguments()
|
|
}
|
|
|
|
}
|
|
|
|
func check() {
|
|
// 暂时只支持SQLite
|
|
if DB.Type != "SQLite" {
|
|
fmt.Println("sorry,we support SQLite only so far(At config/config.go : check())")
|
|
os.Exit(1)
|
|
}
|
|
|
|
}
|
|
func debugprint() {
|
|
|
|
fmt.Printf("ListenPort=%v\n", ListenPort)
|
|
fmt.Printf("database type:%s\n", DB.Type)
|
|
fmt.Printf("database path : %s\n", DB.Path)
|
|
fmt.Printf("database port:%d\n", DB.Port)
|
|
fmt.Printf("database user:%s\n", DB.User)
|
|
fmt.Printf("database passowrd:%s\n", DB.Password)
|
|
fmt.Printf("database name:%s\n", DB.Name)
|
|
fmt.Printf("session:%s\n", Session)
|
|
fmt.Printf("semester:%d\n", Semester)
|
|
fmt.Printf("start time:%s\n", StartTime)
|
|
fmt.Printf("week:%d\n", Week)
|
|
fmt.Printf("File=%v\n", File)
|
|
|
|
}
|
|
|
|
func handleArguments() {
|
|
if len(os.Args) > 2 {
|
|
fmt.Println("Please enter only 1 argument")
|
|
os.Exit(1)
|
|
}
|
|
switch os.Args[1] {
|
|
case "newsemester":
|
|
if DB.Type == "SQLite" {
|
|
sqliteNewSemester()
|
|
}
|
|
default:
|
|
panic("invalid argument")
|
|
}
|
|
|
|
}
|
|
|
|
func readconfig() {
|
|
if err := viper.ReadInConfig(); err != nil {
|
|
fmt.Printf("Error reading config file: %v\n", err)
|
|
os.Exit(1)
|
|
}
|
|
ListenPort = viper.GetInt("ListenPort")
|
|
DB.Type = viper.GetString("DB.Type")
|
|
DB.Path = viper.GetString("DB.Path")
|
|
DB.Port = viper.GetInt("DB.Port")
|
|
DB.User = viper.GetString("DB.User")
|
|
DB.Password = viper.GetString("DB.Password")
|
|
DB.Name = viper.GetString("DB.Name")
|
|
Session = viper.GetString("Session")
|
|
Semester = viper.GetInt("Semester")
|
|
StartTime = viper.GetString("StartTime")
|
|
Week = viper.GetInt("Week")
|
|
File = viper.GetString("File")
|
|
}
|