mirror of https://github.com/go-gorm/gorm.git
Fix tests doesn't follow https://gorm.io/docs/method_chaining.html convention
This commit is contained in:
parent
87decced23
commit
940358e0dd
|
@ -367,33 +367,12 @@ func (db *DB) Scopes(funcs ...func(*DB) *DB) (tx *DB) {
|
||||||
}
|
}
|
||||||
|
|
||||||
func (db *DB) executeScopes() (tx *DB) {
|
func (db *DB) executeScopes() (tx *DB) {
|
||||||
tx = db.getInstance()
|
|
||||||
scopes := db.Statement.scopes
|
scopes := db.Statement.scopes
|
||||||
if len(scopes) == 0 {
|
db.Statement.scopes = nil
|
||||||
return tx
|
|
||||||
}
|
|
||||||
tx.Statement.scopes = nil
|
|
||||||
|
|
||||||
conditions := make([]clause.Interface, 0, 4)
|
|
||||||
if cs, ok := tx.Statement.Clauses["WHERE"]; ok && cs.Expression != nil {
|
|
||||||
conditions = append(conditions, cs.Expression.(clause.Interface))
|
|
||||||
cs.Expression = nil
|
|
||||||
tx.Statement.Clauses["WHERE"] = cs
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, scope := range scopes {
|
for _, scope := range scopes {
|
||||||
tx = scope(tx)
|
db = scope(db)
|
||||||
if cs, ok := tx.Statement.Clauses["WHERE"]; ok && cs.Expression != nil {
|
|
||||||
conditions = append(conditions, cs.Expression.(clause.Interface))
|
|
||||||
cs.Expression = nil
|
|
||||||
tx.Statement.Clauses["WHERE"] = cs
|
|
||||||
}
|
}
|
||||||
}
|
return db
|
||||||
|
|
||||||
for _, condition := range conditions {
|
|
||||||
tx.Statement.AddClause(condition)
|
|
||||||
}
|
|
||||||
return tx
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Preload preload associations with given conditions
|
// Preload preload associations with given conditions
|
||||||
|
|
|
@ -21,6 +21,12 @@ func (where Where) Name() string {
|
||||||
|
|
||||||
// Build build where clause
|
// Build build where clause
|
||||||
func (where Where) Build(builder Builder) {
|
func (where Where) Build(builder Builder) {
|
||||||
|
if len(where.Exprs) == 1 {
|
||||||
|
if andCondition, ok := where.Exprs[0].(AndConditions); ok {
|
||||||
|
where.Exprs = andCondition.Exprs
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// Switch position if the first query expression is a single Or condition
|
// Switch position if the first query expression is a single Or condition
|
||||||
for idx, expr := range where.Exprs {
|
for idx, expr := range where.Exprs {
|
||||||
if v, ok := expr.(OrConditions); !ok || len(v.Exprs) > 1 {
|
if v, ok := expr.(OrConditions); !ok || len(v.Exprs) > 1 {
|
||||||
|
@ -147,6 +153,11 @@ func Not(exprs ...Expression) Expression {
|
||||||
if len(exprs) == 0 {
|
if len(exprs) == 0 {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
if len(exprs) == 1 {
|
||||||
|
if andCondition, ok := exprs[0].(AndConditions); ok {
|
||||||
|
exprs = andCondition.Exprs
|
||||||
|
}
|
||||||
|
}
|
||||||
return NotConditions{Exprs: exprs}
|
return NotConditions{Exprs: exprs}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -63,7 +63,7 @@ func TestWhere(t *testing.T) {
|
||||||
[]clause.Interface{clause.Select{}, clause.From{}, clause.Where{
|
[]clause.Interface{clause.Select{}, clause.From{}, clause.Where{
|
||||||
Exprs: []clause.Expression{clause.And(clause.Eq{Column: "age", Value: 18}, clause.Or(clause.Neq{Column: "name", Value: "jinzhu"}))},
|
Exprs: []clause.Expression{clause.And(clause.Eq{Column: "age", Value: 18}, clause.Or(clause.Neq{Column: "name", Value: "jinzhu"}))},
|
||||||
}},
|
}},
|
||||||
"SELECT * FROM `users` WHERE (`age` = ? OR `name` <> ?)",
|
"SELECT * FROM `users` WHERE `age` = ? OR `name` <> ?",
|
||||||
[]interface{}{18, "jinzhu"},
|
[]interface{}{18, "jinzhu"},
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
|
@ -94,7 +94,7 @@ func TestWhere(t *testing.T) {
|
||||||
clause.And(clause.Expr{SQL: "`score` <= ?", Vars: []interface{}{100}, WithoutParentheses: false})),
|
clause.And(clause.Expr{SQL: "`score` <= ?", Vars: []interface{}{100}, WithoutParentheses: false})),
|
||||||
},
|
},
|
||||||
}},
|
}},
|
||||||
"SELECT * FROM `users` WHERE (`users`.`id` <> ? AND `score` <= ?)",
|
"SELECT * FROM `users` WHERE `users`.`id` <> ? AND `score` <= ?",
|
||||||
[]interface{}{"1", 100},
|
[]interface{}{"1", 100},
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
|
|
|
@ -376,8 +376,12 @@ func (db *DB) FirstOrCreate(dest interface{}, conds ...interface{}) (tx *DB) {
|
||||||
} else if len(db.Statement.assigns) > 0 {
|
} else if len(db.Statement.assigns) > 0 {
|
||||||
exprs := tx.Statement.BuildCondition(db.Statement.assigns[0], db.Statement.assigns[1:]...)
|
exprs := tx.Statement.BuildCondition(db.Statement.assigns[0], db.Statement.assigns[1:]...)
|
||||||
assigns := map[string]interface{}{}
|
assigns := map[string]interface{}{}
|
||||||
for _, expr := range exprs {
|
for i := 0; i < len(exprs); i++ {
|
||||||
if eq, ok := expr.(clause.Eq); ok {
|
expr := exprs[i]
|
||||||
|
|
||||||
|
if eq, ok := expr.(clause.AndConditions); ok {
|
||||||
|
exprs = append(exprs, eq.Exprs...)
|
||||||
|
} else if eq, ok := expr.(clause.Eq); ok {
|
||||||
switch column := eq.Column.(type) {
|
switch column := eq.Column.(type) {
|
||||||
case string:
|
case string:
|
||||||
assigns[column] = eq.Value
|
assigns[column] = eq.Value
|
||||||
|
|
16
statement.go
16
statement.go
|
@ -326,7 +326,7 @@ func (stmt *Statement) BuildCondition(query interface{}, args ...interface{}) []
|
||||||
case *DB:
|
case *DB:
|
||||||
v.executeScopes()
|
v.executeScopes()
|
||||||
|
|
||||||
if cs, ok := v.Statement.Clauses["WHERE"]; ok && cs.Expression != nil {
|
if cs, ok := v.Statement.Clauses["WHERE"]; ok {
|
||||||
if where, ok := cs.Expression.(clause.Where); ok {
|
if where, ok := cs.Expression.(clause.Where); ok {
|
||||||
if len(where.Exprs) == 1 {
|
if len(where.Exprs) == 1 {
|
||||||
if orConds, ok := where.Exprs[0].(clause.OrConditions); ok {
|
if orConds, ok := where.Exprs[0].(clause.OrConditions); ok {
|
||||||
|
@ -334,13 +334,9 @@ func (stmt *Statement) BuildCondition(query interface{}, args ...interface{}) []
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
conds = append(conds, clause.And(where.Exprs...))
|
conds = append(conds, clause.And(where.Exprs...))
|
||||||
} else {
|
} else if cs.Expression != nil {
|
||||||
conds = append(conds, cs.Expression)
|
conds = append(conds, cs.Expression)
|
||||||
}
|
}
|
||||||
if v.Statement == stmt {
|
|
||||||
cs.Expression = nil
|
|
||||||
stmt.Statement.Clauses["WHERE"] = cs
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
case map[interface{}]interface{}:
|
case map[interface{}]interface{}:
|
||||||
for i, j := range v {
|
for i, j := range v {
|
||||||
|
@ -451,8 +447,9 @@ func (stmt *Statement) BuildCondition(query interface{}, args ...interface{}) []
|
||||||
|
|
||||||
if len(values) > 0 {
|
if len(values) > 0 {
|
||||||
conds = append(conds, clause.IN{Column: clause.PrimaryColumn, Values: values})
|
conds = append(conds, clause.IN{Column: clause.PrimaryColumn, Values: values})
|
||||||
|
return []clause.Expression{clause.And(conds...)}
|
||||||
}
|
}
|
||||||
return conds
|
return nil
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -461,7 +458,10 @@ func (stmt *Statement) BuildCondition(query interface{}, args ...interface{}) []
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return conds
|
if len(conds) > 0 {
|
||||||
|
return []clause.Expression{clause.And(conds...)}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// Build build sql with clauses names
|
// Build build sql with clauses names
|
||||||
|
|
10
tests/go.mod
10
tests/go.mod
|
@ -3,7 +3,7 @@ module gorm.io/gorm/tests
|
||||||
go 1.18
|
go 1.18
|
||||||
|
|
||||||
require (
|
require (
|
||||||
github.com/google/uuid v1.4.0
|
github.com/google/uuid v1.5.0
|
||||||
github.com/jinzhu/now v1.1.5
|
github.com/jinzhu/now v1.1.5
|
||||||
github.com/lib/pq v1.10.9
|
github.com/lib/pq v1.10.9
|
||||||
gorm.io/driver/mysql v1.5.2
|
gorm.io/driver/mysql v1.5.2
|
||||||
|
@ -18,12 +18,12 @@ require (
|
||||||
github.com/golang-sql/civil v0.0.0-20220223132316-b832511892a9 // indirect
|
github.com/golang-sql/civil v0.0.0-20220223132316-b832511892a9 // indirect
|
||||||
github.com/golang-sql/sqlexp v0.1.0 // indirect
|
github.com/golang-sql/sqlexp v0.1.0 // indirect
|
||||||
github.com/jackc/pgpassfile v1.0.0 // indirect
|
github.com/jackc/pgpassfile v1.0.0 // indirect
|
||||||
github.com/jackc/pgservicefile v0.0.0-20221227161230-091c0ba34f0a // indirect
|
github.com/jackc/pgservicefile v0.0.0-20231201235250-de7065d80cb9 // indirect
|
||||||
github.com/jackc/pgx/v5 v5.5.0 // indirect
|
github.com/jackc/pgx/v5 v5.5.1 // indirect
|
||||||
github.com/jinzhu/inflection v1.0.0 // indirect
|
github.com/jinzhu/inflection v1.0.0 // indirect
|
||||||
github.com/mattn/go-sqlite3 v1.14.18 // indirect
|
github.com/mattn/go-sqlite3 v1.14.19 // indirect
|
||||||
github.com/microsoft/go-mssqldb v1.6.0 // indirect
|
github.com/microsoft/go-mssqldb v1.6.0 // indirect
|
||||||
golang.org/x/crypto v0.15.0 // indirect
|
golang.org/x/crypto v0.18.0 // indirect
|
||||||
golang.org/x/text v0.14.0 // indirect
|
golang.org/x/text v0.14.0 // indirect
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
|
@ -1118,12 +1118,12 @@ func TestSearchWithStruct(t *testing.T) {
|
||||||
}
|
}
|
||||||
|
|
||||||
result = dryRunDB.Where(User{Name: "jinzhu", Age: 18}).Find(&User{})
|
result = dryRunDB.Where(User{Name: "jinzhu", Age: 18}).Find(&User{})
|
||||||
if !regexp.MustCompile(`WHERE .users.\..name. = .{1,3} AND .users.\..age. = .{1,3} AND .users.\..deleted_at. IS NULL`).MatchString(result.Statement.SQL.String()) {
|
if !regexp.MustCompile(`WHERE \(.users.\..name. = .{1,3} AND .users.\..age. = .{1,3}\) AND .users.\..deleted_at. IS NULL`).MatchString(result.Statement.SQL.String()) {
|
||||||
t.Errorf("invalid query SQL, got %v", result.Statement.SQL.String())
|
t.Errorf("invalid query SQL, got %v", result.Statement.SQL.String())
|
||||||
}
|
}
|
||||||
|
|
||||||
result = dryRunDB.Where(User{Name: "jinzhu"}, "name", "Age").Find(&User{})
|
result = dryRunDB.Where(User{Name: "jinzhu"}, "name", "Age").Find(&User{})
|
||||||
if !regexp.MustCompile(`WHERE .users.\..name. = .{1,3} AND .users.\..age. = .{1,3} AND .users.\..deleted_at. IS NULL`).MatchString(result.Statement.SQL.String()) {
|
if !regexp.MustCompile(`WHERE \(.users.\..name. = .{1,3} AND .users.\..age. = .{1,3}\) AND .users.\..deleted_at. IS NULL`).MatchString(result.Statement.SQL.String()) {
|
||||||
t.Errorf("invalid query SQL, got %v", result.Statement.SQL.String())
|
t.Errorf("invalid query SQL, got %v", result.Statement.SQL.String())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -84,7 +84,9 @@ func TestComplexScopes(t *testing.T) {
|
||||||
queryFn: func(tx *gorm.DB) *gorm.DB {
|
queryFn: func(tx *gorm.DB) *gorm.DB {
|
||||||
return tx.Scopes(
|
return tx.Scopes(
|
||||||
func(d *gorm.DB) *gorm.DB { return d.Where("a = 1") },
|
func(d *gorm.DB) *gorm.DB { return d.Where("a = 1") },
|
||||||
func(d *gorm.DB) *gorm.DB { return d.Where(d.Or("b = 2").Or("c = 3")) },
|
func(d *gorm.DB) *gorm.DB {
|
||||||
|
return d.Where(DB.Or("b = 2").Or("c = 3"))
|
||||||
|
},
|
||||||
).Find(&Language{})
|
).Find(&Language{})
|
||||||
},
|
},
|
||||||
expected: `SELECT * FROM "languages" WHERE a = 1 AND (b = 2 OR c = 3)`,
|
expected: `SELECT * FROM "languages" WHERE a = 1 AND (b = 2 OR c = 3)`,
|
||||||
|
@ -93,7 +95,9 @@ func TestComplexScopes(t *testing.T) {
|
||||||
queryFn: func(tx *gorm.DB) *gorm.DB {
|
queryFn: func(tx *gorm.DB) *gorm.DB {
|
||||||
return tx.Where("z = 0").Scopes(
|
return tx.Where("z = 0").Scopes(
|
||||||
func(d *gorm.DB) *gorm.DB { return d.Where("a = 1") },
|
func(d *gorm.DB) *gorm.DB { return d.Where("a = 1") },
|
||||||
func(d *gorm.DB) *gorm.DB { return d.Or(d.Where("b = 2").Or("c = 3")) },
|
func(d *gorm.DB) *gorm.DB {
|
||||||
|
return d.Or(DB.Where("b = 2").Or("c = 3"))
|
||||||
|
},
|
||||||
).Find(&Language{})
|
).Find(&Language{})
|
||||||
},
|
},
|
||||||
expected: `SELECT * FROM "languages" WHERE z = 0 AND a = 1 OR (b = 2 OR c = 3)`,
|
expected: `SELECT * FROM "languages" WHERE z = 0 AND a = 1 OR (b = 2 OR c = 3)`,
|
||||||
|
@ -104,7 +108,7 @@ func TestComplexScopes(t *testing.T) {
|
||||||
func(d *gorm.DB) *gorm.DB { return d.Model(&Language{}) },
|
func(d *gorm.DB) *gorm.DB { return d.Model(&Language{}) },
|
||||||
func(d *gorm.DB) *gorm.DB {
|
func(d *gorm.DB) *gorm.DB {
|
||||||
return d.
|
return d.
|
||||||
Or(d.Scopes(
|
Or(DB.Scopes(
|
||||||
func(d *gorm.DB) *gorm.DB { return d.Where("a = 1") },
|
func(d *gorm.DB) *gorm.DB { return d.Where("a = 1") },
|
||||||
func(d *gorm.DB) *gorm.DB { return d.Where("b = 2") },
|
func(d *gorm.DB) *gorm.DB { return d.Where("b = 2") },
|
||||||
)).
|
)).
|
||||||
|
|
|
@ -388,7 +388,7 @@ func TestToSQL(t *testing.T) {
|
||||||
sql = DB.ToSQL(func(tx *gorm.DB) *gorm.DB {
|
sql = DB.ToSQL(func(tx *gorm.DB) *gorm.DB {
|
||||||
return tx.Model(&User{}).Where(&User{Name: "foo", Age: 20}).Limit(10).Offset(5).Order("name ASC").First(&User{})
|
return tx.Model(&User{}).Where(&User{Name: "foo", Age: 20}).Limit(10).Offset(5).Order("name ASC").First(&User{})
|
||||||
})
|
})
|
||||||
assertEqualSQL(t, `SELECT * FROM "users" WHERE "users"."name" = 'foo' AND "users"."age" = 20 AND "users"."deleted_at" IS NULL ORDER BY name ASC,"users"."id" LIMIT 1 OFFSET 5`, sql)
|
assertEqualSQL(t, `SELECT * FROM "users" WHERE ("users"."name" = 'foo' AND "users"."age" = 20) AND "users"."deleted_at" IS NULL ORDER BY name ASC,"users"."id" LIMIT 1 OFFSET 5`, sql)
|
||||||
|
|
||||||
// last and unscoped
|
// last and unscoped
|
||||||
sql = DB.ToSQL(func(tx *gorm.DB) *gorm.DB {
|
sql = DB.ToSQL(func(tx *gorm.DB) *gorm.DB {
|
||||||
|
|
Loading…
Reference in New Issue