forked from mirror/gorm
Revert "Revert "Don't override the from clauses, close #4129" close #4139"
This reverts commit d6c23586ae
.
This commit is contained in:
parent
d6c23586ae
commit
a948c84607
|
@ -104,6 +104,11 @@ func BuildQuerySQL(db *gorm.DB) {
|
||||||
}
|
}
|
||||||
|
|
||||||
joins := []clause.Join{}
|
joins := []clause.Join{}
|
||||||
|
|
||||||
|
if fromClause, ok := db.Statement.Clauses["FROM"].Expression.(clause.From); ok {
|
||||||
|
joins = fromClause.Joins
|
||||||
|
}
|
||||||
|
|
||||||
for _, join := range db.Statement.Joins {
|
for _, join := range db.Statement.Joins {
|
||||||
if db.Statement.Schema == nil {
|
if db.Statement.Schema == nil {
|
||||||
joins = append(joins, clause.Join{
|
joins = append(joins, clause.Join{
|
||||||
|
@ -154,6 +159,7 @@ func BuildQuerySQL(db *gorm.DB) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
db.Statement.Joins = nil
|
||||||
db.Statement.AddClause(clause.From{Joins: joins})
|
db.Statement.AddClause(clause.From{Joins: joins})
|
||||||
} else {
|
} else {
|
||||||
db.Statement.AddClauseIfNotExists(clause.From{})
|
db.Statement.AddClauseIfNotExists(clause.From{})
|
||||||
|
|
|
@ -6,6 +6,7 @@ import (
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
"gorm.io/gorm"
|
"gorm.io/gorm"
|
||||||
|
"gorm.io/gorm/clause"
|
||||||
. "gorm.io/gorm/utils/tests"
|
. "gorm.io/gorm/utils/tests"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -242,3 +243,47 @@ func TestCombineStringConditions(t *testing.T) {
|
||||||
t.Fatalf("invalid sql generated, got %v", sql)
|
t.Fatalf("invalid sql generated, got %v", sql)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestFromWithJoins(t *testing.T) {
|
||||||
|
var result User
|
||||||
|
|
||||||
|
newDB := DB.Session(&gorm.Session{NewDB: true, DryRun: true}).Table("users")
|
||||||
|
|
||||||
|
newDB.Clauses(
|
||||||
|
clause.From{
|
||||||
|
Tables: []clause.Table{{Name: "users"}},
|
||||||
|
Joins: []clause.Join{
|
||||||
|
{
|
||||||
|
Table: clause.Table{Name: "companies", Raw: false},
|
||||||
|
ON: clause.Where{
|
||||||
|
Exprs: []clause.Expression{
|
||||||
|
clause.Eq{
|
||||||
|
Column: clause.Column{
|
||||||
|
Table: "users",
|
||||||
|
Name: "company_id",
|
||||||
|
},
|
||||||
|
Value: clause.Column{
|
||||||
|
Table: "companies",
|
||||||
|
Name: "id",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
)
|
||||||
|
|
||||||
|
newDB.Joins("inner join rgs on rgs.id = user.id")
|
||||||
|
|
||||||
|
stmt := newDB.First(&result).Statement
|
||||||
|
str := stmt.SQL.String()
|
||||||
|
|
||||||
|
if !strings.Contains(str, "rgs.id = user.id") {
|
||||||
|
t.Errorf("The second join condition is over written instead of combining")
|
||||||
|
}
|
||||||
|
|
||||||
|
if !strings.Contains(str, "`users`.`company_id` = `companies`.`id`") && !strings.Contains(str, "\"users\".\"company_id\" = \"companies\".\"id\"") {
|
||||||
|
t.Errorf("The first join condition is over written instead of combining")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
Loading…
Reference in New Issue