Merge pull request #20 from go-gorm/lzq

Update template
go-gorm · Aug 6, 2021 · 7646027 · 7646027
2 parents 1a23f3b + 18cef56
commit 7646027
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 33 deletions.
diff --git a/generator.go b/generator.go
@@ -229,15 +229,22 @@ func (g *Generator) generatedBaseStruct() (err error) {
 		pkg = check.ModelPkg
 	}
 	outPath = fmt.Sprint(filepath.Dir(outPath), "/", pkg, "/")
-	if _, err := os.Stat(outPath); err != nil {
-		if err := os.Mkdir(outPath, os.ModePerm); err != nil {
-			log.Fatalf("mkdir failed: %s", err)
-		}
-	}
+
+	_, err = os.Stat(outPath)
+	created := err == nil
+
 	for _, data := range g.Data {
 		if data.BaseStruct == nil || !data.BaseStruct.GenBaseStruct {
 			continue
 		}
+
+		if !created {
+			if err := os.Mkdir(outPath, os.ModePerm); err != nil {
+				log.Fatalf("mkdir failed: %s", err)
+			}
+			created = true
+		}
+
 		var buf bytes.Buffer
 		err = render(tmpl.ModelTemplate, &buf, data.BaseStruct)
 		if err != nil {

diff --git a/internal/template/tmpl.go b/internal/template/tmpl.go