Don't override the from clauses, close #4129
This commit is contained in:
		
							parent
							
								
									90476fea7a
								
							
						
					
					
						commit
						664755270d
					
				| @ -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{ | ||||||
|  | |||||||
| @ -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…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user
	 Jinzhu
						Jinzhu