tablet/gorm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(finisher_api): compose sqls and vars statement

Browse Source
This commit is contained in:
Anton Kichaev 2023-08-31 20:58:33 +07:00
parent e57e5d8884
commit 28331bf198
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
finisher_api.go
View File

@ -36,6 +36,9 @@ func (db *DB) CreateInBatches(value interface{}, batchSize int) (tx *DB) {
// the reflection length judgment of the optimized value // the reflection length judgment of the optimized value
reflectLen := reflectValue.Len() reflectLen := reflectValue.Len()
statementSql := strings.Builder{}
statementVars := make([]interface{}, 0, reflectLen)
callFc := func(tx *DB) error { callFc := func(tx *DB) error {
for i := 0; i < reflectLen; i += batchSize { for i := 0; i < reflectLen; i += batchSize {
ends := i + batchSize ends := i + batchSize
@ -49,6 +52,11 @@ func (db *DB) CreateInBatches(value interface{}, batchSize int) (tx *DB) {
if subtx.Error != nil { if subtx.Error != nil {
return subtx.Error return subtx.Error
} }
statementSql.WriteString(subtx.Statement.SQL.String())
statementSql.WriteString(";")
statementVars = append(statementVars, subtx.Statement.Vars...)
rowsAffected += subtx.RowsAffected rowsAffected += subtx.RowsAffected
} }
return nil return nil
@ -59,6 +67,8 @@ func (db *DB) CreateInBatches(value interface{}, batchSize int) (tx *DB) {
} else { } else {
tx.AddError(tx.Transaction(callFc)) tx.AddError(tx.Transaction(callFc))
} }
tx.Statement.SQL = statementSql
tx.Statement.Vars = statementVars
tx.RowsAffected = rowsAffected tx.RowsAffected = rowsAffected
default: default: