添加 config load

config.yaml

@@ -0,0 +1,8 @@
+# This is a example file for the system
+ListenPort: 25005
+File: "member.csv"
+DB:
+  Path: "scheduler.db"
+  Port: "" # not need if use SQLite
+  User: "" # not need if use SQLite
+  Password: "" # not need if use SQLite

config/config.go

@@ -0,0 +1,46 @@
+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")
+
+	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", ListenPort)
+	fmt.Printf("File=%v", File)
+	fmt.Printf("database path : %s", DB.Path)
+
+}

config/model.go

@@ -0,0 +1,16 @@
+package config
+
+var (
+	ListenPort int
+	File       string
+	DB         database
+)
+
+type database struct {
+	Path string
+	//Port     string
+	//User     string
+	//Password string
+
+	// enable if you want use an instance other than SQLite
+}

main.go

@@ -1 +1,10 @@
 package main
+
+import (
+	"zsxyww.com/scheduler/config"
+)
+
+func main() {
+
+	config.Load()
+}