Only enable simple protocal for pgx

jinzhu · jinzhu · commit 096c357296a6 · 2022-10-07T19:59:21.000+08:00
diff --git a/migrator.go b/migrator.go
@@ -527,8 +527,9 @@ func (m Migrator) GetRows(currentSchema interface{}, table interface{}) (*sql.Ro
 	}
 
 	return m.DB.Session(&gorm.Session{}).Table(name).Limit(1).Scopes(func(d *gorm.DB) *gorm.DB {
+		dialector, _ := m.Dialector.(Dialector)
 		// use simple protocol
-		if !m.DB.PrepareStmt && len(d.Statement.Vars) != 0 {
+		if !m.DB.PrepareStmt && (dialector.Config != nil && (dialector.Config.DriverName == "" || dialector.Config.DriverName == "pgx")) {
 			d.Statement.Vars = append(d.Statement.Vars, pgx.QuerySimpleProtocol(true))
 		}
 		return d

Original file line number	Diff line number	Diff line change
`@@ -527,8 +527,9 @@ func (m Migrator) GetRows(currentSchema interface{}, table interface{}) (*sql.Ro`
`527`	`527`	`}`
`528`	`528`
`529`	`529`	`return m.DB.Session(&gorm.Session{}).Table(name).Limit(1).Scopes(func(d gorm.DB) gorm.DB {`
	`530`	`+ dialector, _ := m.Dialector.(Dialector)`
`530`	`531`	`// use simple protocol`
`531`		`- if !m.DB.PrepareStmt && len(d.Statement.Vars) != 0 {`
	`532`	`+ if !m.DB.PrepareStmt && (dialector.Config != nil && (dialector.Config.DriverName == "" \|\| dialector.Config.DriverName == "pgx")) {`
`532`	`533`	`d.Statement.Vars = append(d.Statement.Vars, pgx.QuerySimpleProtocol(true))`
`533`	`534`	`}`
`534`	`535`	`return d`