mirror of https://github.com/go-gorm/gorm.git
Compare commits
3 Commits
17bcbdd914
...
753796c998
Author | SHA1 | Date |
---|---|---|
Cr. | 753796c998 | |
Jinzhu | 6bfccf8afa | |
a631807682 | 68fccb87b0 |
|
@ -380,14 +380,68 @@ func (db *DB) Scopes(funcs ...func(*DB) *DB) (tx *DB) {
|
|||
}
|
||||
|
||||
func (db *DB) executeScopes() (tx *DB) {
|
||||
if len(db.Statement.scopes) == 0 {
|
||||
return db
|
||||
}
|
||||
|
||||
scopes := db.Statement.scopes
|
||||
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 {
|
||||
db = scope(db)
|
||||
txs = append(txs, scope(cleanDB))
|
||||
}
|
||||
|
||||
db.Statement.Clauses = originClause
|
||||
db.mergeClauses(txs)
|
||||
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
|
||||
//
|
||||
// // 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,
|
||||
// overriding the soft deletion behavior.
|
||||
// Example:
|
||||
//
|
||||
// var users []User
|
||||
// db.Unscoped().Find(&users)
|
||||
// // Retrieves all users, including deleted ones.
|
||||
|
|
2
go.mod
2
go.mod
|
@ -5,5 +5,5 @@ go 1.18
|
|||
require (
|
||||
github.com/jinzhu/inflection v1.0.0
|
||||
github.com/jinzhu/now v1.1.5
|
||||
golang.org/x/text v0.14.0
|
||||
golang.org/x/text v0.20.0
|
||||
)
|
||||
|
|
4
go.sum
4
go.sum
|
@ -2,5 +2,5 @@ github.com/jinzhu/inflection v1.0.0 h1:K317FqzuhWc8YvSVlFMCCUb36O/S9MCKRDI7QkRKD
|
|||
github.com/jinzhu/inflection v1.0.0/go.mod h1:h+uFLlag+Qp1Va5pdKtLDYj+kHp5pxUVkryuEj+Srlc=
|
||||
github.com/jinzhu/now v1.1.5 h1:/o9tlHleP7gOFmsnYNz3RGnqzefHA47wQpKrrdTIwXQ=
|
||||
github.com/jinzhu/now v1.1.5/go.mod h1:d3SSVoowX0Lcu0IBviAWJpolVfI5UJVZZ7cO71lE/z8=
|
||||
golang.org/x/text v0.14.0 h1:ScX5w1eTa3QqT8oi6+ziP7dTV1S2+ALU0bI+0zXKWiQ=
|
||||
golang.org/x/text v0.14.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU=
|
||||
golang.org/x/text v0.20.0 h1:gK/Kv2otX8gz+wn7Rmb3vT96ZwuoxnQlY+HlJVj7Qug=
|
||||
golang.org/x/text v0.20.0/go.mod h1:D4IsuqiFMhST5bX19pQ9ikHC2GsaKyk/oF+pn3ducp4=
|
||||
|
|
1
gorm.go
1
gorm.go
|
@ -183,7 +183,6 @@ func Open(dialector Dialector, opts ...Option) (db *DB, err error) {
|
|||
|
||||
if config.Dialector != nil {
|
||||
err = config.Dialector.Initialize(db)
|
||||
|
||||
if err != nil {
|
||||
if db, _ := db.DB(); db != nil {
|
||||
_ = db.Close()
|
||||
|
|
|
@ -325,7 +325,7 @@ func (stmt *Statement) BuildCondition(query interface{}, args ...interface{}) []
|
|||
case clause.Expression:
|
||||
conds = append(conds, v)
|
||||
case *DB:
|
||||
v.executeScopes()
|
||||
v = v.executeScopes()
|
||||
|
||||
if cs, ok := v.Statement.Clauses["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)
|
||||
}
|
||||
}
|
||||
|
||||
conds = append(conds, clause.And(where.Exprs...))
|
||||
} else if cs.Expression != nil {
|
||||
conds = append(conds, cs.Expression)
|
||||
|
|
|
@ -8,7 +8,7 @@ require (
|
|||
github.com/lib/pq v1.10.9
|
||||
github.com/stretchr/testify v1.9.0
|
||||
gorm.io/driver/mysql v1.5.7
|
||||
gorm.io/driver/postgres v1.5.9
|
||||
gorm.io/driver/postgres v1.5.10
|
||||
gorm.io/driver/sqlite v1.5.6
|
||||
gorm.io/driver/sqlserver v1.5.4
|
||||
gorm.io/gorm v1.25.12
|
||||
|
@ -25,12 +25,12 @@ require (
|
|||
github.com/jackc/pgx/v5 v5.7.1 // indirect
|
||||
github.com/jinzhu/inflection v1.0.0 // indirect
|
||||
github.com/kr/text v0.2.0 // indirect
|
||||
github.com/mattn/go-sqlite3 v1.14.23 // indirect
|
||||
github.com/mattn/go-sqlite3 v1.14.24 // indirect
|
||||
github.com/microsoft/go-mssqldb v1.7.2 // indirect
|
||||
github.com/pmezard/go-difflib v1.0.0 // indirect
|
||||
github.com/rogpeppe/go-internal v1.12.0 // indirect
|
||||
golang.org/x/crypto v0.27.0 // indirect
|
||||
golang.org/x/text v0.18.0 // indirect
|
||||
golang.org/x/crypto v0.29.0 // indirect
|
||||
golang.org/x/text v0.20.0 // indirect
|
||||
gopkg.in/yaml.v3 v3.0.1 // indirect
|
||||
)
|
||||
|
||||
|
|
|
@ -90,7 +90,31 @@ func TestComplexScopes(t *testing.T) {
|
|||
).Find(&Language{})
|
||||
},
|
||||
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",
|
||||
queryFn: func(tx *gorm.DB) *gorm.DB {
|
||||
return tx.Where("z = 0").Scopes(
|
||||
|
|
|
@ -16,22 +16,23 @@ then
|
|||
fi
|
||||
|
||||
# SqlServer for Mac M1
|
||||
if [[ -z $GITHUB_ACTION ]]; then
|
||||
if [ -d tests ]
|
||||
then
|
||||
if [[ -z $GITHUB_ACTION && -d tests ]]; then
|
||||
cd tests
|
||||
if [[ $(uname -a) == *" arm64" ]]; then
|
||||
MSSQL_IMAGE=mcr.microsoft.com/azure-sql-edge docker compose up -d || true
|
||||
MSSQL_IMAGE=mcr.microsoft.com/azure-sql-edge docker compose up -d --wait
|
||||
go install github.com/microsoft/go-sqlcmd/cmd/sqlcmd@latest || true
|
||||
SQLCMDPASSWORD=LoremIpsum86 sqlcmd -U sa -S localhost:9930 -Q "IF DB_ID('gorm') IS NULL CREATE DATABASE gorm" > /dev/null || true
|
||||
SQLCMDPASSWORD=LoremIpsum86 sqlcmd -U sa -S localhost:9930 -Q "IF SUSER_ID (N'gorm') IS NULL CREATE LOGIN gorm WITH PASSWORD = 'LoremIpsum86';" > /dev/null || true
|
||||
SQLCMDPASSWORD=LoremIpsum86 sqlcmd -U sa -S localhost:9930 -Q "IF USER_ID (N'gorm') IS NULL CREATE USER gorm FROM LOGIN gorm; ALTER SERVER ROLE sysadmin ADD MEMBER [gorm];" > /dev/null || true
|
||||
for query in \
|
||||
"IF DB_ID('gorm') IS NULL CREATE DATABASE gorm" \
|
||||
"IF SUSER_ID (N'gorm') IS NULL CREATE LOGIN gorm WITH PASSWORD = 'LoremIpsum86';" \
|
||||
"IF USER_ID (N'gorm') IS NULL CREATE USER gorm FROM LOGIN gorm; ALTER SERVER ROLE sysadmin ADD MEMBER [gorm];"
|
||||
do
|
||||
SQLCMDPASSWORD=LoremIpsum86 sqlcmd -U sa -S localhost:9930 -Q "$query" > /dev/null || true
|
||||
done
|
||||
else
|
||||
MSSQL_IMAGE=mcr.microsoft.com/mssql/server docker compose up -d
|
||||
MSSQL_IMAGE=mcr.microsoft.com/mssql/server docker compose up -d --wait
|
||||
fi
|
||||
cd ..
|
||||
fi
|
||||
fi
|
||||
|
||||
|
||||
for dialect in "${dialects[@]}" ; do
|
||||
|
|
Loading…
Reference in New Issue