知识分享之Golang——和我一起搭建一个快速开发框架（一）

go get github.com/spf13/viper

package begin

import (
    "bufio"
    "flag"
    "fmt"
    "github.com/spf13/viper"
    "os"
    "strings"
)

// Start 主配置启动函数
func Start() error {
    sysConfigDir := flag.String("d", "./conf/", "配置一个启动配置文件存放目录，默认：./conf/")
    flag.Parse()
    sysConfigName := flag.String("n", "sys", "配置启动配置文件名称，默认：sys")
    flag.Parse()
    sysConfigType := flag.String("t", "toml", "配置启动配置文件类型，默认：toml")
    flag.Parse()
    // 加载启动配置
    configLoad(*sysConfigDir, *sysConfigName, *sysConfigType)
    return nil
}

func configLoad(sysConfigDir, sysConfigName, sysConfigType string) {
    // 设定读取名称
    viper.SetConfigName(sysConfigName)
    // 设定配置文件类型
    viper.SetConfigType(sysConfigType)
    // 设定扫描目录
    viper.AddConfigPath(sysConfigDir)
    // 读取配置文件
    if err := viper.ReadInConfig(); err != nil {
        if _, ok := err.(viper.ConfigFileNotFoundError); ok {
            reader := bufio.NewReader(os.Stdin)
            fmt.Println("没有找到配置文件，是否自动创建默认配置文件并启动?y/n")
            // Config file not found; ignore error if desired
            for ok {
                fmt.Print("-> ")
                isContinue, err := reader.ReadString('\n')
                if err != nil {
                    panic(fmt.Errorf("发生致命错误: %w \n", err))
                    break
                } else {
                    // 转换为小写并去除前后空格
                    isContinue = strings.TrimSpace(strings.ToLower(isContinue))
                    // 判断输入是否正确
                    if isContinue == "y" || isContinue == "yes" || isContinue == "0" {
                        // 正常进行
                        //TODO 调用自动化创建初始配置文件函数
                        break
                    } else if isContinue == "n" || isContinue == "no" || isContinue == "1" {
                        // 结束
                        break
                    }
                }
            }
        } else {
            // Config file was found but another error was produced
            panic(fmt.Errorf("配置文件致命错误: %w \n", err))
        }
    }
}

func main() {
    begin.Start()
}