improve gfile.MainPkgPath

yanllllk · Mar 19, 2021 · 4aaf09f · 4aaf09f
1 parent 2f3df76
commit 4aaf09f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/os/gfile/gfile_source.go b/os/gfile/gfile_source.go
@@ -64,7 +64,7 @@ func MainPkgPath() string {
 					continue
 				}
 			}
-			if gregex.IsMatchString(`package\s+main`, GetContents(file)) {
+			if gregex.IsMatchString(`package\s+main\s+{`, GetContents(file)) {
 				mainPkgPath.Set(Dir(file))
 				return Dir(file)
 			}
@@ -79,7 +79,7 @@ func MainPkgPath() string {
 		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)) {
+				if gregex.IsMatchString(`package\s+main\s+{`, GetContents(v)) {
 					mainPkgPath.Set(path)
 					return path
 				}