fix issue #2734 (#2740)

gogf · Jul 5, 2023 · c90e931 · c90e931
1 parent ba2a7e4
commit c90e931
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/cmd/gf/internal/cmd/genservice/genservice.go b/cmd/gf/internal/cmd/genservice/genservice.go
@@ -324,13 +324,21 @@ func (c CGenService) checkAndUpdateMain(srcFolder string) (err error) {
 		mainFilePath     = gfile.Join(gfile.Dir(gfile.Dir(gfile.Dir(logicFilePath))), "main.go")
 		mainFileContent  = gfile.GetContents(mainFilePath)
 	)
+	// No main content found.
+	if mainFileContent == "" {
+		return nil
+	}
 	if gstr.Contains(mainFileContent, importStr) {
 		return nil
 	}
 	match, err := gregex.MatchString(`import \(([\s\S]+?)\)`, mainFileContent)
 	if err != nil {
 		return err
 	}
+	// No match.
+	if len(match) < 2 {
+		return nil
+	}
 	lines := garray.NewStrArrayFrom(gstr.Split(match[1], "\n"))
 	for i, line := range lines.Slice() {
 		line = gstr.Trim(line)