diff --git a/g/os/gfile/gfile.go b/g/os/gfile/gfile.go index 224b40bd0..47bbdb680 100644 --- a/g/os/gfile/gfile.go +++ b/g/os/gfile/gfile.go @@ -454,6 +454,9 @@ func MainPkgPath() string { if gstr.Contains(file, "/gf/g/") { continue } + if Ext(file) != ".go" { + continue + } path = file for path != "/" && gstr.Contains(path, "/") { files, _ := ScanDir(path, "*.go") diff --git a/g/os/gfile/gfile_test.go b/g/os/gfile/gfile_test.go index b7b9317b8..c0315f9f5 100644 --- a/g/os/gfile/gfile_test.go +++ b/g/os/gfile/gfile_test.go @@ -653,8 +653,8 @@ func TestMkdir(t *testing.T) { func TestStat(t *testing.T) { gtest.Case(t, func() { var ( - tpath1 string = "/testfile_t1.txt" - tpath2 string = "./testfile_t1_no.txt" + tpath1 = "/testfile_t1.txt" + tpath2 = "./testfile_t1_no.txt" err error fileiofo os.FileInfo ) @@ -675,12 +675,7 @@ func TestStat(t *testing.T) { func TestMainPkgPath(t *testing.T) { gtest.Case(t, func() { - var ( - reads string - ) - - reads = gfile.MainPkgPath() + reads := gfile.MainPkgPath() gtest.Assert(reads, "") - }) }