mirror of https://github.com/go-gorm/gorm.git
Merge 68fccb87b0
into 49bbaa637f
This commit is contained in:
commit
17bcbdd914
|
@ -380,14 +380,68 @@ func (db *DB) Scopes(funcs ...func(*DB) *DB) (tx *DB) {
|
||||||
}
|
}
|
||||||
|
|
||||||
func (db *DB) executeScopes() (tx *DB) {
|
func (db *DB) executeScopes() (tx *DB) {
|
||||||
|
if len(db.Statement.scopes) == 0 {
|
||||||
|
return db
|
||||||
|
}
|
||||||
|
|
||||||
scopes := db.Statement.scopes
|
scopes := db.Statement.scopes
|
||||||
db.Statement.scopes = nil
|
db.Statement.scopes = nil
|
||||||
|
originClause := db.Statement.Clauses
|
||||||
|
|
||||||
|
// use clean db in scope
|
||||||
|
cleanDB := db.Session(&Session{})
|
||||||
|
cleanDB.Statement.Clauses = map[string]clause.Clause{}
|
||||||
|
|
||||||
|
txs := make([]*DB, 0, len(scopes))
|
||||||
for _, scope := range scopes {
|
for _, scope := range scopes {
|
||||||
db = scope(db)
|
txs = append(txs, scope(cleanDB))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
db.Statement.Clauses = originClause
|
||||||
|
db.mergeClauses(txs)
|
||||||
return db
|
return db
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (db *DB) mergeClauses(txs []*DB) {
|
||||||
|
if len(txs) == 0 {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tx := range txs {
|
||||||
|
stmt := tx.Statement
|
||||||
|
if stmt != nil {
|
||||||
|
stmtClause := stmt.Clauses
|
||||||
|
// merge clauses
|
||||||
|
if cs, ok := stmtClause["WHERE"]; ok {
|
||||||
|
if where, ok := cs.Expression.(clause.Where); ok {
|
||||||
|
db.Statement.AddClause(where)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// cover other expr
|
||||||
|
if stmt.TableExpr != nil {
|
||||||
|
db.Statement.TableExpr = stmt.TableExpr
|
||||||
|
}
|
||||||
|
|
||||||
|
if stmt.Table != "" {
|
||||||
|
db.Statement.Table = stmt.Table
|
||||||
|
}
|
||||||
|
|
||||||
|
if stmt.Model != nil {
|
||||||
|
db.Statement.Model = stmt.Model
|
||||||
|
}
|
||||||
|
|
||||||
|
if stmt.Selects != nil {
|
||||||
|
db.Statement.Selects = stmt.Selects
|
||||||
|
}
|
||||||
|
|
||||||
|
if stmt.Omits != nil {
|
||||||
|
db.Statement.Omits = stmt.Omits
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// Preload preload associations with given conditions
|
// Preload preload associations with given conditions
|
||||||
//
|
//
|
||||||
// // get all users, and preload all non-cancelled orders
|
// // get all users, and preload all non-cancelled orders
|
||||||
|
@ -448,6 +502,7 @@ func (db *DB) Assign(attrs ...interface{}) (tx *DB) {
|
||||||
// Unscoped allows queries to include records marked as deleted,
|
// Unscoped allows queries to include records marked as deleted,
|
||||||
// overriding the soft deletion behavior.
|
// overriding the soft deletion behavior.
|
||||||
// Example:
|
// Example:
|
||||||
|
//
|
||||||
// var users []User
|
// var users []User
|
||||||
// db.Unscoped().Find(&users)
|
// db.Unscoped().Find(&users)
|
||||||
// // Retrieves all users, including deleted ones.
|
// // Retrieves all users, including deleted ones.
|
||||||
|
|
|
@ -325,7 +325,7 @@ func (stmt *Statement) BuildCondition(query interface{}, args ...interface{}) []
|
||||||
case clause.Expression:
|
case clause.Expression:
|
||||||
conds = append(conds, v)
|
conds = append(conds, v)
|
||||||
case *DB:
|
case *DB:
|
||||||
v.executeScopes()
|
v = v.executeScopes()
|
||||||
|
|
||||||
if cs, ok := v.Statement.Clauses["WHERE"]; ok {
|
if cs, ok := v.Statement.Clauses["WHERE"]; ok {
|
||||||
if where, ok := cs.Expression.(clause.Where); ok {
|
if where, ok := cs.Expression.(clause.Where); ok {
|
||||||
|
@ -334,6 +334,7 @@ func (stmt *Statement) BuildCondition(query interface{}, args ...interface{}) []
|
||||||
where.Exprs[0] = clause.AndConditions(orConds)
|
where.Exprs[0] = clause.AndConditions(orConds)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
conds = append(conds, clause.And(where.Exprs...))
|
conds = append(conds, clause.And(where.Exprs...))
|
||||||
} else if cs.Expression != nil {
|
} else if cs.Expression != nil {
|
||||||
conds = append(conds, cs.Expression)
|
conds = append(conds, cs.Expression)
|
||||||
|
|
|
@ -90,7 +90,31 @@ func TestComplexScopes(t *testing.T) {
|
||||||
).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)`,
|
||||||
}, {
|
},
|
||||||
|
{
|
||||||
|
name: "group_cond",
|
||||||
|
queryFn: func(tx *gorm.DB) *gorm.DB {
|
||||||
|
return tx.Scopes(
|
||||||
|
func(d *gorm.DB) *gorm.DB { return d.Table("languages1") },
|
||||||
|
func(d *gorm.DB) *gorm.DB { return d.Table("languages2") },
|
||||||
|
func(d *gorm.DB) *gorm.DB {
|
||||||
|
return d.Where(
|
||||||
|
d.Where("a = 1").Or("b = 2"),
|
||||||
|
)
|
||||||
|
},
|
||||||
|
func(d *gorm.DB) *gorm.DB {
|
||||||
|
return d.Select("f1, f2")
|
||||||
|
},
|
||||||
|
func(d *gorm.DB) *gorm.DB {
|
||||||
|
return d.Where(
|
||||||
|
d.Where("c = 3"),
|
||||||
|
)
|
||||||
|
},
|
||||||
|
).Find(&Language{})
|
||||||
|
},
|
||||||
|
expected: `SELECT f1, f2 FROM "languages2" WHERE (a = 1 OR b = 2) AND c = 3`,
|
||||||
|
},
|
||||||
|
{
|
||||||
name: "depth_1_pre_cond",
|
name: "depth_1_pre_cond",
|
||||||
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(
|
||||||
|
|
Loading…
Reference in New Issue