add config path setting

skye-z · Mar 18, 2024 · a99f18a · a99f18a
1 parent 8064671
commit a99f18a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/main.go b/main.go
@@ -11,7 +11,7 @@ import (
 var page embed.FS
 
 func main() {
-	util.InitConfig()
+	util.InitConfig(".")
 	router := service.InitRouter(page)
 	service.RunRouter(router)
 }
diff --git a/util/config.go b/util/config.go
@@ -14,10 +14,10 @@ import (
 	"github.com/spf13/viper"
 )
 
-func InitConfig() {
+func InitConfig(path string) {
 	viper.SetConfigName("config")
 	viper.SetConfigType("ini")
-	viper.AddConfigPath(".")
+	viper.AddConfigPath(path)
 	err := viper.ReadInConfig()
 	if err != nil {
 		if _, ok := err.(viper.ConfigFileNotFoundError); ok {