feat(config): Add config file flag

config/config.go

@@ -2,9 +2,10 @@ package config
 
 import (
 	"fmt"
-	"github.com/spf13/viper"
 	"os"
 	"strconv"
+
+	"github.com/spf13/viper"
 )
 
 var Share *AppConf
@@ -57,8 +58,8 @@ type Logger struct {
 	MaxBackups int    `mapstructure:"max_backups"`
 }
 
-func Init() error {
-	viper.SetConfigFile("./conf/config.yaml")
+func Init(f string) error {
+	viper.SetConfigFile(f)
 	err := viper.ReadInConfig()
 	if err != nil {
 		return err

main.go

@@ -1,17 +1,27 @@
 package main
 
 import (
+	"flag"
+
 	"apihut-server/config"
 	"apihut-server/dao/mysql"
 	"apihut-server/dao/redis"
 	"apihut-server/logger"
 	"apihut-server/routers"
 )
 
+var configFile string
+
+func init() {
+	flag.StringVar(&configFile, "f", "./conf/config.yaml", "Config file")
+}
+
 func main() {
+	flag.Parse()
+
 	var err error
 	// 初始化配置
-	err = config.Init()
+	err = config.Init(configFile)
 	if err != nil {
 		panic(err)
 	}