fea: 测试单元寻找配置目录

go-home-admin · Jun 1, 2023 · 4c98e7e · 4c98e7e
1 parent 983233d
commit 4c98e7e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/bootstrap/providers/config_provider.go b/bootstrap/providers/config_provider.go
@@ -56,7 +56,9 @@ func (c *ConfigProvider) initFile() {
 		parDir := ""
 		for i := 0; i <= 100; i++ {
 			checkDir := pwd + parDir
-			if _, err := os.Stat(checkDir + "/go.mod"); err == nil {
+			_, err1 := os.Stat(checkDir + "/go.mod")
+			_, err2 := os.Stat(checkDir + "/.env")
+			if err1 == nil || err2 == nil {
 				dirs, _ := filepath.Abs(checkDir + "/" + defaultDir)
 				DirEntry, err = os.ReadDir(dirs)
 				if err != nil {