diff --git a/internal/statements/query.go b/internal/statements/query.go index 8a9e59e4..674f8712 100644 --- a/internal/statements/query.go +++ b/internal/statements/query.go @@ -233,8 +233,8 @@ func (statement *Statement) writeForUpdate(w *builder.BytesWriter) error { return nil } - if statement.dialect.URI().DBType != schemas.MYSQL { - return errors.New("only support mysql for update") + if statement.dialect.URI().DBType != schemas.MYSQL && statement.dialect.URI().DBType != schemas.POSTGRES { + return errors.New("only support mysql and postgres for update") } _, err := fmt.Fprint(w, " FOR UPDATE") return err diff --git a/tests/tests.go b/tests/tests.go index 220e1c67..e0a760e4 100644 --- a/tests/tests.go +++ b/tests/tests.go @@ -88,7 +88,6 @@ func createEngine(dbType, connStr string) error { } } db.Close() - *ignoreSelectUpdate = true case schemas.MYSQL: db, err := sql.Open(dbType, strings.ReplaceAll(connStr, "xorm_test", "mysql")) if err != nil {