From 4de17ddfe011c8ed07222f17193c098d431c8ae9 Mon Sep 17 00:00:00 2001
From: david_du <david.du@yuansuan.com>
Date: Sun, 15 Jun 2025 11:43:43 +0800
Subject: [PATCH] fix: return early when error occurs

---
 generator.go | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/generator.go b/generator.go
index b90bbc79..f20548c1 100644
--- a/generator.go
+++ b/generator.go
@@ -94,8 +94,8 @@ type Logger interface {
 // Generator code generator
 type Generator struct {
 	Config
-	Data   map[string]*genInfo                  //gen query data
-	models map[string]*generate.QueryStructMeta //gen model data
+	Data   map[string]*genInfo                  // gen query data
+	models map[string]*generate.QueryStructMeta // gen model data
 
 	logger Logger
 }
@@ -315,12 +315,14 @@ func (g *Generator) generateQueryFile() (err error) {
 			err := g.generateSingleQueryFile(info)
 			if err != nil {
 				errChan <- err
+				return
 			}
 
 			if g.WithUnitTest {
 				err = g.generateQueryUnitTestFile(info)
 				if err != nil { // do not panic
 					g.db.Logger.Error(context.Background(), "generate unit test fail: %s", err)
+					return
 				}
 			}
 		}(info)