bingyi/gf
1
0
Fork 0
mirror of https://gitee.com/johng/gf synced 2026-06-06 02:25:47 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix issue #1190

Browse Source
This commit is contained in:
jianchenma
2021-03-19 11:35:12 +08:00
parent 48d840a1b8
commit 36963c05a2
1 changed files with 18 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
os/gfile/gfile_source.go
View File

@ -8,6 +8,7 @@ package gfile
import (
"github.com/gogf/gf/text/gstr"
"os"
"runtime"
"strings"
@ -44,6 +45,7 @@ func MainPkgPath() string {
if path != "" {
return path
}
var lastFile string
for i := 1; i < 10000; i++ {
if pc, file, _, ok := runtime.Caller(i); ok {
if goRootForFilter != "" && len(file) >= len(goRootForFilter) && file[0:len(goRootForFilter)] == goRootForFilter {
@ -61,6 +63,7 @@ func MainPkgPath() string {
if Ext(file) != ".go" {
continue
}
lastFile = file
if gregex.IsMatchString(`package\s+main`, GetContents(file)) {
mainPkgPath.Set(Dir(file))
return Dir(file)
@ -69,5 +72,20 @@ func MainPkgPath() string {
break
}
}
// If it still cannot find the path of the package main,
// it recursively searches the directory and its parents directory of the last go file.
// It's usually necessary for uint testing cases of business project.
if lastFile != "" {
for path = Dir(lastFile); len(path) > 1 && Exists(path) && path[len(path)-1] != os.PathSeparator; {
files, _ := ScanDir(path, "*.go")
for _, v := range files {
if gregex.IsMatchString(`package\s+main`, GetContents(v)) {
mainPkgPath.Set(path)
return path
}
}
path = Dir(path)
}
}
return ""
}