feat: diy generator logger

generator.go

@@ -57,6 +57,8 @@ func NewGenerator(cfg Config) *Generator {
 		Config: cfg,
 		Data:   make(map[string]*genInfo),
 		models: make(map[string]*generate.QueryStructMeta),
+
+		logger: log.Default(),
 	}
 }
 
@@ -84,12 +86,23 @@ func (i *genInfo) methodInGenInfo(m *generate.InterfaceMethod) bool {
 	return false
 }
 
+// Logger  gen logger interface
+type Logger interface {
+	Println(v ...any)
+}
+
 // Generator code generator
 type Generator struct {
 	Config
-
 	Data   map[string]*genInfo                  //gen query data
 	models map[string]*generate.QueryStructMeta //gen model data
+
+	logger Logger
+}
+
+// SetLogger  set gen logger
+func (g *Generator) SetLogger(logger Logger) {
+	g.logger = logger
 }
 
 // UseDB set db connection
@@ -278,7 +291,7 @@ func (g *Generator) Execute() {
 func (g *Generator) info(logInfos ...string) {
 	for _, l := range logInfos {
 		g.db.Logger.Info(context.Background(), l)
-		log.Println(l)
+		g.logger.Println(l)
 	}
 }