perf: modify log level for query field comment to avoid printing excessive useless logs (#146)

iTanken · web-flow · commit 6825faf33710 · 2025-01-12T20:04:21.000+08:00
diff --git a/migrator.go b/migrator.go
@@ -8,6 +8,7 @@ import (
 
 	"gorm.io/gorm"
 	"gorm.io/gorm/clause"
+	"gorm.io/gorm/logger"
 	"gorm.io/gorm/migrator"
 	"gorm.io/gorm/schema"
 )
@@ -273,9 +274,8 @@ func (m Migrator) RenameColumn(value interface{}, oldName, newName string) error
 }
 
 func (m Migrator) GetColumnComment(stmt *gorm.Statement, fieldDBName string) (description string) {
-	queryTx := m.DB
+	queryTx := m.DB.Session(&gorm.Session{Logger: m.DB.Logger.LogMode(logger.Warn)})
 	if m.DB.DryRun {
-		queryTx = m.DB.Session(&gorm.Session{})
 		queryTx.DryRun = false
 	}
 	var comment sql.NullString

Original file line number	Diff line number	Diff line change
`@@ -8,6 +8,7 @@ import (`
`8`	`8`
`9`	`9`	`"gorm.io/gorm"`
`10`	`10`	`"gorm.io/gorm/clause"`
	`11`	`+ "gorm.io/gorm/logger"`
`11`	`12`	`"gorm.io/gorm/migrator"`
`12`	`13`	`"gorm.io/gorm/schema"`
`13`	`14`	`)`
`@@ -273,9 +274,8 @@ func (m Migrator) RenameColumn(value interface{}, oldName, newName string) error`
`273`	`274`	`}`
`274`	`275`
`275`	`276`	`func (m Migrator) GetColumnComment(stmt *gorm.Statement, fieldDBName string) (description string) {`
`276`		`- queryTx := m.DB`
	`277`	`+ queryTx := m.DB.Session(&gorm.Session{Logger: m.DB.Logger.LogMode(logger.Warn)})`
`277`	`278`	`if m.DB.DryRun {`
`278`		`- queryTx = m.DB.Session(&gorm.Session{})`
`279`	`279`	`queryTx.DryRun = false`
`280`	`280`	`}`
`281`	`281`	`var comment sql.NullString`